xref: /linux/net/ipv4/tcp.c (revision 6e5e1de616bf5f3df1769abc9292191dfad9110a)
12874c5fdSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
21da177e4SLinus Torvalds /*
31da177e4SLinus Torvalds  * INET		An implementation of the TCP/IP protocol suite for the LINUX
41da177e4SLinus Torvalds  *		operating system.  INET is implemented using the  BSD Socket
51da177e4SLinus Torvalds  *		interface as the means of communication with the user level.
61da177e4SLinus Torvalds  *
71da177e4SLinus Torvalds  *		Implementation of the Transmission Control Protocol(TCP).
81da177e4SLinus Torvalds  *
902c30a84SJesper Juhl  * Authors:	Ross Biro
101da177e4SLinus Torvalds  *		Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
111da177e4SLinus Torvalds  *		Mark Evans, <evansmp@uhura.aston.ac.uk>
121da177e4SLinus Torvalds  *		Corey Minyard <wf-rch!minyard@relay.EU.net>
131da177e4SLinus Torvalds  *		Florian La Roche, <flla@stud.uni-sb.de>
141da177e4SLinus Torvalds  *		Charles Hedrick, <hedrick@klinzhai.rutgers.edu>
151da177e4SLinus Torvalds  *		Linus Torvalds, <torvalds@cs.helsinki.fi>
161da177e4SLinus Torvalds  *		Alan Cox, <gw4pts@gw4pts.ampr.org>
171da177e4SLinus Torvalds  *		Matthew Dillon, <dillon@apollo.west.oic.com>
181da177e4SLinus Torvalds  *		Arnt Gulbrandsen, <agulbra@nvg.unit.no>
191da177e4SLinus Torvalds  *		Jorge Cwik, <jorge@laser.satlink.net>
201da177e4SLinus Torvalds  *
211da177e4SLinus Torvalds  * Fixes:
221da177e4SLinus Torvalds  *		Alan Cox	:	Numerous verify_area() calls
231da177e4SLinus Torvalds  *		Alan Cox	:	Set the ACK bit on a reset
241da177e4SLinus Torvalds  *		Alan Cox	:	Stopped it crashing if it closed while
251da177e4SLinus Torvalds  *					sk->inuse=1 and was trying to connect
261da177e4SLinus Torvalds  *					(tcp_err()).
271da177e4SLinus Torvalds  *		Alan Cox	:	All icmp error handling was broken
281da177e4SLinus Torvalds  *					pointers passed where wrong and the
291da177e4SLinus Torvalds  *					socket was looked up backwards. Nobody
301da177e4SLinus Torvalds  *					tested any icmp error code obviously.
311da177e4SLinus Torvalds  *		Alan Cox	:	tcp_err() now handled properly. It
321da177e4SLinus Torvalds  *					wakes people on errors. poll
331da177e4SLinus Torvalds  *					behaves and the icmp error race
341da177e4SLinus Torvalds  *					has gone by moving it into sock.c
351da177e4SLinus Torvalds  *		Alan Cox	:	tcp_send_reset() fixed to work for
361da177e4SLinus Torvalds  *					everything not just packets for
371da177e4SLinus Torvalds  *					unknown sockets.
381da177e4SLinus Torvalds  *		Alan Cox	:	tcp option processing.
391da177e4SLinus Torvalds  *		Alan Cox	:	Reset tweaked (still not 100%) [Had
401da177e4SLinus Torvalds  *					syn rule wrong]
411da177e4SLinus Torvalds  *		Herp Rosmanith  :	More reset fixes
421da177e4SLinus Torvalds  *		Alan Cox	:	No longer acks invalid rst frames.
431da177e4SLinus Torvalds  *					Acking any kind of RST is right out.
441da177e4SLinus Torvalds  *		Alan Cox	:	Sets an ignore me flag on an rst
451da177e4SLinus Torvalds  *					receive otherwise odd bits of prattle
461da177e4SLinus Torvalds  *					escape still
471da177e4SLinus Torvalds  *		Alan Cox	:	Fixed another acking RST frame bug.
481da177e4SLinus Torvalds  *					Should stop LAN workplace lockups.
491da177e4SLinus Torvalds  *		Alan Cox	: 	Some tidyups using the new skb list
501da177e4SLinus Torvalds  *					facilities
511da177e4SLinus Torvalds  *		Alan Cox	:	sk->keepopen now seems to work
521da177e4SLinus Torvalds  *		Alan Cox	:	Pulls options out correctly on accepts
531da177e4SLinus Torvalds  *		Alan Cox	:	Fixed assorted sk->rqueue->next errors
541da177e4SLinus Torvalds  *		Alan Cox	:	PSH doesn't end a TCP read. Switched a
551da177e4SLinus Torvalds  *					bit to skb ops.
561da177e4SLinus Torvalds  *		Alan Cox	:	Tidied tcp_data to avoid a potential
571da177e4SLinus Torvalds  *					nasty.
581da177e4SLinus Torvalds  *		Alan Cox	:	Added some better commenting, as the
591da177e4SLinus Torvalds  *					tcp is hard to follow
601da177e4SLinus Torvalds  *		Alan Cox	:	Removed incorrect check for 20 * psh
611da177e4SLinus Torvalds  *	Michael O'Reilly	:	ack < copied bug fix.
621da177e4SLinus Torvalds  *	Johannes Stille		:	Misc tcp fixes (not all in yet).
631da177e4SLinus Torvalds  *		Alan Cox	:	FIN with no memory -> CRASH
641da177e4SLinus Torvalds  *		Alan Cox	:	Added socket option proto entries.
651da177e4SLinus Torvalds  *					Also added awareness of them to accept.
661da177e4SLinus Torvalds  *		Alan Cox	:	Added TCP options (SOL_TCP)
671da177e4SLinus Torvalds  *		Alan Cox	:	Switched wakeup calls to callbacks,
681da177e4SLinus Torvalds  *					so the kernel can layer network
691da177e4SLinus Torvalds  *					sockets.
701da177e4SLinus Torvalds  *		Alan Cox	:	Use ip_tos/ip_ttl settings.
711da177e4SLinus Torvalds  *		Alan Cox	:	Handle FIN (more) properly (we hope).
721da177e4SLinus Torvalds  *		Alan Cox	:	RST frames sent on unsynchronised
731da177e4SLinus Torvalds  *					state ack error.
741da177e4SLinus Torvalds  *		Alan Cox	:	Put in missing check for SYN bit.
751da177e4SLinus Torvalds  *		Alan Cox	:	Added tcp_select_window() aka NET2E
761da177e4SLinus Torvalds  *					window non shrink trick.
771da177e4SLinus Torvalds  *		Alan Cox	:	Added a couple of small NET2E timer
781da177e4SLinus Torvalds  *					fixes
791da177e4SLinus Torvalds  *		Charles Hedrick :	TCP fixes
801da177e4SLinus Torvalds  *		Toomas Tamm	:	TCP window fixes
811da177e4SLinus Torvalds  *		Alan Cox	:	Small URG fix to rlogin ^C ack fight
821da177e4SLinus Torvalds  *		Charles Hedrick	:	Rewrote most of it to actually work
831da177e4SLinus Torvalds  *		Linus		:	Rewrote tcp_read() and URG handling
841da177e4SLinus Torvalds  *					completely
851da177e4SLinus Torvalds  *		Gerhard Koerting:	Fixed some missing timer handling
861da177e4SLinus Torvalds  *		Matthew Dillon  :	Reworked TCP machine states as per RFC
871da177e4SLinus Torvalds  *		Gerhard Koerting:	PC/TCP workarounds
881da177e4SLinus Torvalds  *		Adam Caldwell	:	Assorted timer/timing errors
891da177e4SLinus Torvalds  *		Matthew Dillon	:	Fixed another RST bug
901da177e4SLinus Torvalds  *		Alan Cox	:	Move to kernel side addressing changes.
911da177e4SLinus Torvalds  *		Alan Cox	:	Beginning work on TCP fastpathing
921da177e4SLinus Torvalds  *					(not yet usable)
931da177e4SLinus Torvalds  *		Arnt Gulbrandsen:	Turbocharged tcp_check() routine.
941da177e4SLinus Torvalds  *		Alan Cox	:	TCP fast path debugging
951da177e4SLinus Torvalds  *		Alan Cox	:	Window clamping
961da177e4SLinus Torvalds  *		Michael Riepe	:	Bug in tcp_check()
971da177e4SLinus Torvalds  *		Matt Dillon	:	More TCP improvements and RST bug fixes
981da177e4SLinus Torvalds  *		Matt Dillon	:	Yet more small nasties remove from the
991da177e4SLinus Torvalds  *					TCP code (Be very nice to this man if
1001da177e4SLinus Torvalds  *					tcp finally works 100%) 8)
1011da177e4SLinus Torvalds  *		Alan Cox	:	BSD accept semantics.
1021da177e4SLinus Torvalds  *		Alan Cox	:	Reset on closedown bug.
1031da177e4SLinus Torvalds  *	Peter De Schrijver	:	ENOTCONN check missing in tcp_sendto().
1041da177e4SLinus Torvalds  *		Michael Pall	:	Handle poll() after URG properly in
1051da177e4SLinus Torvalds  *					all cases.
1061da177e4SLinus Torvalds  *		Michael Pall	:	Undo the last fix in tcp_read_urg()
1071da177e4SLinus Torvalds  *					(multi URG PUSH broke rlogin).
1081da177e4SLinus Torvalds  *		Michael Pall	:	Fix the multi URG PUSH problem in
1091da177e4SLinus Torvalds  *					tcp_readable(), poll() after URG
1101da177e4SLinus Torvalds  *					works now.
1111da177e4SLinus Torvalds  *		Michael Pall	:	recv(...,MSG_OOB) never blocks in the
1121da177e4SLinus Torvalds  *					BSD api.
1131da177e4SLinus Torvalds  *		Alan Cox	:	Changed the semantics of sk->socket to
1141da177e4SLinus Torvalds  *					fix a race and a signal problem with
1151da177e4SLinus Torvalds  *					accept() and async I/O.
1161da177e4SLinus Torvalds  *		Alan Cox	:	Relaxed the rules on tcp_sendto().
1171da177e4SLinus Torvalds  *		Yury Shevchuk	:	Really fixed accept() blocking problem.
1181da177e4SLinus Torvalds  *		Craig I. Hagan  :	Allow for BSD compatible TIME_WAIT for
1191da177e4SLinus Torvalds  *					clients/servers which listen in on
1201da177e4SLinus Torvalds  *					fixed ports.
1211da177e4SLinus Torvalds  *		Alan Cox	:	Cleaned the above up and shrank it to
1221da177e4SLinus Torvalds  *					a sensible code size.
1231da177e4SLinus Torvalds  *		Alan Cox	:	Self connect lockup fix.
1241da177e4SLinus Torvalds  *		Alan Cox	:	No connect to multicast.
1251da177e4SLinus Torvalds  *		Ross Biro	:	Close unaccepted children on master
1261da177e4SLinus Torvalds  *					socket close.
1271da177e4SLinus Torvalds  *		Alan Cox	:	Reset tracing code.
1281da177e4SLinus Torvalds  *		Alan Cox	:	Spurious resets on shutdown.
1291da177e4SLinus Torvalds  *		Alan Cox	:	Giant 15 minute/60 second timer error
1301da177e4SLinus Torvalds  *		Alan Cox	:	Small whoops in polling before an
1311da177e4SLinus Torvalds  *					accept.
1321da177e4SLinus Torvalds  *		Alan Cox	:	Kept the state trace facility since
1331da177e4SLinus Torvalds  *					it's handy for debugging.
1341da177e4SLinus Torvalds  *		Alan Cox	:	More reset handler fixes.
1351da177e4SLinus Torvalds  *		Alan Cox	:	Started rewriting the code based on
1361da177e4SLinus Torvalds  *					the RFC's for other useful protocol
1371da177e4SLinus Torvalds  *					references see: Comer, KA9Q NOS, and
1381da177e4SLinus Torvalds  *					for a reference on the difference
1391da177e4SLinus Torvalds  *					between specifications and how BSD
1401da177e4SLinus Torvalds  *					works see the 4.4lite source.
1411da177e4SLinus Torvalds  *		A.N.Kuznetsov	:	Don't time wait on completion of tidy
1421da177e4SLinus Torvalds  *					close.
1431da177e4SLinus Torvalds  *		Linus Torvalds	:	Fin/Shutdown & copied_seq changes.
1441da177e4SLinus Torvalds  *		Linus Torvalds	:	Fixed BSD port reuse to work first syn
1451da177e4SLinus Torvalds  *		Alan Cox	:	Reimplemented timers as per the RFC
1461da177e4SLinus Torvalds  *					and using multiple timers for sanity.
1471da177e4SLinus Torvalds  *		Alan Cox	:	Small bug fixes, and a lot of new
1481da177e4SLinus Torvalds  *					comments.
1491da177e4SLinus Torvalds  *		Alan Cox	:	Fixed dual reader crash by locking
1501da177e4SLinus Torvalds  *					the buffers (much like datagram.c)
1511da177e4SLinus Torvalds  *		Alan Cox	:	Fixed stuck sockets in probe. A probe
1521da177e4SLinus Torvalds  *					now gets fed up of retrying without
1531da177e4SLinus Torvalds  *					(even a no space) answer.
1541da177e4SLinus Torvalds  *		Alan Cox	:	Extracted closing code better
1551da177e4SLinus Torvalds  *		Alan Cox	:	Fixed the closing state machine to
1561da177e4SLinus Torvalds  *					resemble the RFC.
1571da177e4SLinus Torvalds  *		Alan Cox	:	More 'per spec' fixes.
1581da177e4SLinus Torvalds  *		Jorge Cwik	:	Even faster checksumming.
1591da177e4SLinus Torvalds  *		Alan Cox	:	tcp_data() doesn't ack illegal PSH
1601da177e4SLinus Torvalds  *					only frames. At least one pc tcp stack
1611da177e4SLinus Torvalds  *					generates them.
1621da177e4SLinus Torvalds  *		Alan Cox	:	Cache last socket.
1631da177e4SLinus Torvalds  *		Alan Cox	:	Per route irtt.
1641da177e4SLinus Torvalds  *		Matt Day	:	poll()->select() match BSD precisely on error
1651da177e4SLinus Torvalds  *		Alan Cox	:	New buffers
1661da177e4SLinus Torvalds  *		Marc Tamsky	:	Various sk->prot->retransmits and
1671da177e4SLinus Torvalds  *					sk->retransmits misupdating fixed.
1681da177e4SLinus Torvalds  *					Fixed tcp_write_timeout: stuck close,
1691da177e4SLinus Torvalds  *					and TCP syn retries gets used now.
1701da177e4SLinus Torvalds  *		Mark Yarvis	:	In tcp_read_wakeup(), don't send an
1711da177e4SLinus Torvalds  *					ack if state is TCP_CLOSED.
1721da177e4SLinus Torvalds  *		Alan Cox	:	Look up device on a retransmit - routes may
1731da177e4SLinus Torvalds  *					change. Doesn't yet cope with MSS shrink right
1741da177e4SLinus Torvalds  *					but it's a start!
1751da177e4SLinus Torvalds  *		Marc Tamsky	:	Closing in closing fixes.
1761da177e4SLinus Torvalds  *		Mike Shaver	:	RFC1122 verifications.
1771da177e4SLinus Torvalds  *		Alan Cox	:	rcv_saddr errors.
1781da177e4SLinus Torvalds  *		Alan Cox	:	Block double connect().
1791da177e4SLinus Torvalds  *		Alan Cox	:	Small hooks for enSKIP.
1801da177e4SLinus Torvalds  *		Alexey Kuznetsov:	Path MTU discovery.
1811da177e4SLinus Torvalds  *		Alan Cox	:	Support soft errors.
1821da177e4SLinus Torvalds  *		Alan Cox	:	Fix MTU discovery pathological case
1831da177e4SLinus Torvalds  *					when the remote claims no mtu!
1841da177e4SLinus Torvalds  *		Marc Tamsky	:	TCP_CLOSE fix.
1851da177e4SLinus Torvalds  *		Colin (G3TNE)	:	Send a reset on syn ack replies in
1861da177e4SLinus Torvalds  *					window but wrong (fixes NT lpd problems)
1871da177e4SLinus Torvalds  *		Pedro Roque	:	Better TCP window handling, delayed ack.
1881da177e4SLinus Torvalds  *		Joerg Reuter	:	No modification of locked buffers in
1891da177e4SLinus Torvalds  *					tcp_do_retransmit()
1901da177e4SLinus Torvalds  *		Eric Schenk	:	Changed receiver side silly window
1911da177e4SLinus Torvalds  *					avoidance algorithm to BSD style
1921da177e4SLinus Torvalds  *					algorithm. This doubles throughput
1931da177e4SLinus Torvalds  *					against machines running Solaris,
1941da177e4SLinus Torvalds  *					and seems to result in general
1951da177e4SLinus Torvalds  *					improvement.
1961da177e4SLinus Torvalds  *	Stefan Magdalinski	:	adjusted tcp_readable() to fix FIONREAD
1971da177e4SLinus Torvalds  *	Willy Konynenberg	:	Transparent proxying support.
1981da177e4SLinus Torvalds  *	Mike McLagan		:	Routing by source
1991da177e4SLinus Torvalds  *		Keith Owens	:	Do proper merging with partial SKB's in
2001da177e4SLinus Torvalds  *					tcp_do_sendmsg to avoid burstiness.
2011da177e4SLinus Torvalds  *		Eric Schenk	:	Fix fast close down bug with
2021da177e4SLinus Torvalds  *					shutdown() followed by close().
2031da177e4SLinus Torvalds  *		Andi Kleen 	:	Make poll agree with SIGIO
2041da177e4SLinus Torvalds  *	Salvatore Sanfilippo	:	Support SO_LINGER with linger == 1 and
2051da177e4SLinus Torvalds  *					lingertime == 0 (RFC 793 ABORT Call)
2061da177e4SLinus Torvalds  *	Hirokazu Takahashi	:	Use copy_from_user() instead of
2071da177e4SLinus Torvalds  *					csum_and_copy_from_user() if possible.
2081da177e4SLinus Torvalds  *
2091da177e4SLinus Torvalds  * Description of States:
2101da177e4SLinus Torvalds  *
2111da177e4SLinus Torvalds  *	TCP_SYN_SENT		sent a connection request, waiting for ack
2121da177e4SLinus Torvalds  *
2131da177e4SLinus Torvalds  *	TCP_SYN_RECV		received a connection request, sent ack,
2141da177e4SLinus Torvalds  *				waiting for final ack in three-way handshake.
2151da177e4SLinus Torvalds  *
2161da177e4SLinus Torvalds  *	TCP_ESTABLISHED		connection established
2171da177e4SLinus Torvalds  *
2181da177e4SLinus Torvalds  *	TCP_FIN_WAIT1		our side has shutdown, waiting to complete
2191da177e4SLinus Torvalds  *				transmission of remaining buffered data
2201da177e4SLinus Torvalds  *
2211da177e4SLinus Torvalds  *	TCP_FIN_WAIT2		all buffered data sent, waiting for remote
2221da177e4SLinus Torvalds  *				to shutdown
2231da177e4SLinus Torvalds  *
2241da177e4SLinus Torvalds  *	TCP_CLOSING		both sides have shutdown but we still have
2251da177e4SLinus Torvalds  *				data we have to finish sending
2261da177e4SLinus Torvalds  *
2271da177e4SLinus Torvalds  *	TCP_TIME_WAIT		timeout to catch resent junk before entering
2281da177e4SLinus Torvalds  *				closed, can only be entered from FIN_WAIT2
2291da177e4SLinus Torvalds  *				or CLOSING.  Required because the other end
2301da177e4SLinus Torvalds  *				may not have gotten our last ACK causing it
2311da177e4SLinus Torvalds  *				to retransmit the data packet (which we ignore)
2321da177e4SLinus Torvalds  *
2331da177e4SLinus Torvalds  *	TCP_CLOSE_WAIT		remote side has shutdown and is waiting for
2341da177e4SLinus Torvalds  *				us to finish writing our data and to shutdown
2351da177e4SLinus Torvalds  *				(we have to close() to move on to LAST_ACK)
2361da177e4SLinus Torvalds  *
2371da177e4SLinus Torvalds  *	TCP_LAST_ACK		out side has shutdown after remote has
2381da177e4SLinus Torvalds  *				shutdown.  There may still be data in our
2391da177e4SLinus Torvalds  *				buffer that we have to finish sending
2401da177e4SLinus Torvalds  *
2411da177e4SLinus Torvalds  *	TCP_CLOSE		socket is finished
2421da177e4SLinus Torvalds  */
2431da177e4SLinus Torvalds 
244afd46503SJoe Perches #define pr_fmt(fmt) "TCP: " fmt
245afd46503SJoe Perches 
246cf80e0e4SHerbert Xu #include <crypto/hash.h>
247172589ccSIlpo Järvinen #include <linux/kernel.h>
2481da177e4SLinus Torvalds #include <linux/module.h>
2491da177e4SLinus Torvalds #include <linux/types.h>
2501da177e4SLinus Torvalds #include <linux/fcntl.h>
2511da177e4SLinus Torvalds #include <linux/poll.h>
2526e9250f5SEric Dumazet #include <linux/inet_diag.h>
2531da177e4SLinus Torvalds #include <linux/init.h>
2541da177e4SLinus Torvalds #include <linux/fs.h>
2559c55e01cSJens Axboe #include <linux/skbuff.h>
25681b23b4aSAndrew Morton #include <linux/scatterlist.h>
2579c55e01cSJens Axboe #include <linux/splice.h>
2589c55e01cSJens Axboe #include <linux/net.h>
2599c55e01cSJens Axboe #include <linux/socket.h>
2601da177e4SLinus Torvalds #include <linux/random.h>
26157c8a661SMike Rapoport #include <linux/memblock.h>
26257413ebcSMiquel van Smoorenburg #include <linux/highmem.h>
263b8059eadSDavid S. Miller #include <linux/cache.h>
264f4c50d99SHerbert Xu #include <linux/err.h>
265da5c78c8SWilliam Allen Simpson #include <linux/time.h>
2665a0e3ad6STejun Heo #include <linux/slab.h>
26798aaa913SMike Maloney #include <linux/errqueue.h>
26860e2a778SUrsula Braun #include <linux/static_key.h>
26997a19cafSYonghong Song #include <linux/btf.h>
2701da177e4SLinus Torvalds 
2711da177e4SLinus Torvalds #include <net/icmp.h>
272cf60af03SYuchung Cheng #include <net/inet_common.h>
2731da177e4SLinus Torvalds #include <net/tcp.h>
274f870fa0bSMat Martineau #include <net/mptcp.h>
2751da177e4SLinus Torvalds #include <net/xfrm.h>
2761da177e4SLinus Torvalds #include <net/ip.h>
2779c55e01cSJens Axboe #include <net/sock.h>
2781da177e4SLinus Torvalds 
2797c0f6ba6SLinus Torvalds #include <linux/uaccess.h>
2801da177e4SLinus Torvalds #include <asm/ioctls.h>
281076bb0c8SEliezer Tamir #include <net/busy_poll.h>
2821da177e4SLinus Torvalds 
283925bba24SArjun Roy /* Track pending CMSGs. */
284925bba24SArjun Roy enum {
285925bba24SArjun Roy 	TCP_CMSG_INQ = 1,
286925bba24SArjun Roy 	TCP_CMSG_TS = 2
287925bba24SArjun Roy };
288925bba24SArjun Roy 
28919757cebSEric Dumazet DEFINE_PER_CPU(unsigned int, tcp_orphan_count);
29019757cebSEric Dumazet EXPORT_PER_CPU_SYMBOL_GPL(tcp_orphan_count);
2910a5578cfSArnaldo Carvalho de Melo 
292a4fe34bfSEric W. Biederman long sysctl_tcp_mem[3] __read_mostly;
293a4fe34bfSEric W. Biederman EXPORT_SYMBOL(sysctl_tcp_mem);
2941da177e4SLinus Torvalds 
29591b6d325SEric Dumazet atomic_long_t tcp_memory_allocated ____cacheline_aligned_in_smp;	/* Current allocated memory. */
2961da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_memory_allocated);
2970defbb0aSEric Dumazet DEFINE_PER_CPU(int, tcp_memory_per_cpu_fw_alloc);
2980defbb0aSEric Dumazet EXPORT_PER_CPU_SYMBOL_GPL(tcp_memory_per_cpu_fw_alloc);
2991748376bSEric Dumazet 
30060e2a778SUrsula Braun #if IS_ENABLED(CONFIG_SMC)
30160e2a778SUrsula Braun DEFINE_STATIC_KEY_FALSE(tcp_have_smc);
30260e2a778SUrsula Braun EXPORT_SYMBOL(tcp_have_smc);
30360e2a778SUrsula Braun #endif
30460e2a778SUrsula Braun 
3051748376bSEric Dumazet /*
3061748376bSEric Dumazet  * Current number of TCP sockets.
3071748376bSEric Dumazet  */
30891b6d325SEric Dumazet struct percpu_counter tcp_sockets_allocated ____cacheline_aligned_in_smp;
3091da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_sockets_allocated);
3101da177e4SLinus Torvalds 
3111da177e4SLinus Torvalds /*
3129c55e01cSJens Axboe  * TCP splice context
3139c55e01cSJens Axboe  */
3149c55e01cSJens Axboe struct tcp_splice_state {
3159c55e01cSJens Axboe 	struct pipe_inode_info *pipe;
3169c55e01cSJens Axboe 	size_t len;
3179c55e01cSJens Axboe 	unsigned int flags;
3189c55e01cSJens Axboe };
3199c55e01cSJens Axboe 
3209c55e01cSJens Axboe /*
3211da177e4SLinus Torvalds  * Pressure flag: try to collapse.
3221da177e4SLinus Torvalds  * Technical note: it is used by multiple contexts non atomically.
3233ab224beSHideo Aoki  * All the __sk_mem_schedule() is of this nature: accounting
3241da177e4SLinus Torvalds  * is strict, actions are advisory and have some latency.
3251da177e4SLinus Torvalds  */
32606044751SEric Dumazet unsigned long tcp_memory_pressure __read_mostly;
32706044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_memory_pressure);
3281da177e4SLinus Torvalds 
3295c52ba17SPavel Emelyanov void tcp_enter_memory_pressure(struct sock *sk)
3301da177e4SLinus Torvalds {
33106044751SEric Dumazet 	unsigned long val;
33206044751SEric Dumazet 
3331f142c17SEric Dumazet 	if (READ_ONCE(tcp_memory_pressure))
33406044751SEric Dumazet 		return;
33506044751SEric Dumazet 	val = jiffies;
33606044751SEric Dumazet 
33706044751SEric Dumazet 	if (!val)
33806044751SEric Dumazet 		val--;
33906044751SEric Dumazet 	if (!cmpxchg(&tcp_memory_pressure, 0, val))
3404e673444SPavel Emelyanov 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURES);
3411da177e4SLinus Torvalds }
34206044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_enter_memory_pressure);
34306044751SEric Dumazet 
34406044751SEric Dumazet void tcp_leave_memory_pressure(struct sock *sk)
34506044751SEric Dumazet {
34606044751SEric Dumazet 	unsigned long val;
34706044751SEric Dumazet 
3481f142c17SEric Dumazet 	if (!READ_ONCE(tcp_memory_pressure))
34906044751SEric Dumazet 		return;
35006044751SEric Dumazet 	val = xchg(&tcp_memory_pressure, 0);
35106044751SEric Dumazet 	if (val)
35206044751SEric Dumazet 		NET_ADD_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURESCHRONO,
35306044751SEric Dumazet 			      jiffies_to_msecs(jiffies - val));
3541da177e4SLinus Torvalds }
35506044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_leave_memory_pressure);
3561da177e4SLinus Torvalds 
357b103cf34SJulian Anastasov /* Convert seconds to retransmits based on initial and max timeout */
358b103cf34SJulian Anastasov static u8 secs_to_retrans(int seconds, int timeout, int rto_max)
359b103cf34SJulian Anastasov {
360b103cf34SJulian Anastasov 	u8 res = 0;
361b103cf34SJulian Anastasov 
362b103cf34SJulian Anastasov 	if (seconds > 0) {
363b103cf34SJulian Anastasov 		int period = timeout;
364b103cf34SJulian Anastasov 
365b103cf34SJulian Anastasov 		res = 1;
366b103cf34SJulian Anastasov 		while (seconds > period && res < 255) {
367b103cf34SJulian Anastasov 			res++;
368b103cf34SJulian Anastasov 			timeout <<= 1;
369b103cf34SJulian Anastasov 			if (timeout > rto_max)
370b103cf34SJulian Anastasov 				timeout = rto_max;
371b103cf34SJulian Anastasov 			period += timeout;
372b103cf34SJulian Anastasov 		}
373b103cf34SJulian Anastasov 	}
374b103cf34SJulian Anastasov 	return res;
375b103cf34SJulian Anastasov }
376b103cf34SJulian Anastasov 
377b103cf34SJulian Anastasov /* Convert retransmits to seconds based on initial and max timeout */
378b103cf34SJulian Anastasov static int retrans_to_secs(u8 retrans, int timeout, int rto_max)
379b103cf34SJulian Anastasov {
380b103cf34SJulian Anastasov 	int period = 0;
381b103cf34SJulian Anastasov 
382b103cf34SJulian Anastasov 	if (retrans > 0) {
383b103cf34SJulian Anastasov 		period = timeout;
384b103cf34SJulian Anastasov 		while (--retrans) {
385b103cf34SJulian Anastasov 			timeout <<= 1;
386b103cf34SJulian Anastasov 			if (timeout > rto_max)
387b103cf34SJulian Anastasov 				timeout = rto_max;
388b103cf34SJulian Anastasov 			period += timeout;
389b103cf34SJulian Anastasov 		}
390b103cf34SJulian Anastasov 	}
391b103cf34SJulian Anastasov 	return period;
392b103cf34SJulian Anastasov }
393b103cf34SJulian Anastasov 
3940263598cSWei Wang static u64 tcp_compute_delivery_rate(const struct tcp_sock *tp)
3950263598cSWei Wang {
3960263598cSWei Wang 	u32 rate = READ_ONCE(tp->rate_delivered);
3970263598cSWei Wang 	u32 intv = READ_ONCE(tp->rate_interval_us);
3980263598cSWei Wang 	u64 rate64 = 0;
3990263598cSWei Wang 
4000263598cSWei Wang 	if (rate && intv) {
4010263598cSWei Wang 		rate64 = (u64)rate * tp->mss_cache * USEC_PER_SEC;
4020263598cSWei Wang 		do_div(rate64, intv);
4030263598cSWei Wang 	}
4040263598cSWei Wang 	return rate64;
4050263598cSWei Wang }
4060263598cSWei Wang 
407900f65d3SNeal Cardwell /* Address-family independent initialization for a tcp_sock.
408900f65d3SNeal Cardwell  *
409900f65d3SNeal Cardwell  * NOTE: A lot of things set to zero explicitly by call to
410900f65d3SNeal Cardwell  *       sk_alloc() so need not be done here.
411900f65d3SNeal Cardwell  */
412900f65d3SNeal Cardwell void tcp_init_sock(struct sock *sk)
413900f65d3SNeal Cardwell {
414900f65d3SNeal Cardwell 	struct inet_connection_sock *icsk = inet_csk(sk);
415900f65d3SNeal Cardwell 	struct tcp_sock *tp = tcp_sk(sk);
416900f65d3SNeal Cardwell 
4179f5afeaeSYaogong Wang 	tp->out_of_order_queue = RB_ROOT;
41875c119afSEric Dumazet 	sk->tcp_rtx_queue = RB_ROOT;
419900f65d3SNeal Cardwell 	tcp_init_xmit_timers(sk);
42046d3ceabSEric Dumazet 	INIT_LIST_HEAD(&tp->tsq_node);
421e2080072SEric Dumazet 	INIT_LIST_HEAD(&tp->tsorted_sent_queue);
422900f65d3SNeal Cardwell 
423900f65d3SNeal Cardwell 	icsk->icsk_rto = TCP_TIMEOUT_INIT;
424ca584ba0SMartin KaFai Lau 	icsk->icsk_rto_min = TCP_RTO_MIN;
4252b8ee4f0SMartin KaFai Lau 	icsk->icsk_delack_max = TCP_DELACK_MAX;
426740b0f18SEric Dumazet 	tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT);
427ac9517fcSEric Dumazet 	minmax_reset(&tp->rtt_min, tcp_jiffies32, ~0U);
428900f65d3SNeal Cardwell 
429900f65d3SNeal Cardwell 	/* So many TCP implementations out there (incorrectly) count the
430900f65d3SNeal Cardwell 	 * initial SYN frame in their delayed-ACK and congestion control
431900f65d3SNeal Cardwell 	 * algorithms that we must have the following bandaid to talk
432900f65d3SNeal Cardwell 	 * efficiently to them.  -DaveM
433900f65d3SNeal Cardwell 	 */
43440570375SEric Dumazet 	tcp_snd_cwnd_set(tp, TCP_INIT_CWND);
435900f65d3SNeal Cardwell 
436d7722e85SSoheil Hassas Yeganeh 	/* There's a bubble in the pipe until at least the first ACK. */
437d7722e85SSoheil Hassas Yeganeh 	tp->app_limited = ~0U;
438300b655dSDavid Morley 	tp->rate_app_limited = 1;
439d7722e85SSoheil Hassas Yeganeh 
440900f65d3SNeal Cardwell 	/* See draft-stevens-tcpca-spec-01 for discussion of the
441900f65d3SNeal Cardwell 	 * initialization of these values.
442900f65d3SNeal Cardwell 	 */
443900f65d3SNeal Cardwell 	tp->snd_ssthresh = TCP_INFINITE_SSTHRESH;
444900f65d3SNeal Cardwell 	tp->snd_cwnd_clamp = ~0;
445900f65d3SNeal Cardwell 	tp->mss_cache = TCP_MSS_DEFAULT;
446900f65d3SNeal Cardwell 
44746778cd1SKuniyuki Iwashima 	tp->reordering = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_reordering);
44855d8694fSFlorian Westphal 	tcp_assign_congestion_control(sk);
449900f65d3SNeal Cardwell 
450ceaa1fefSAndrey Vagin 	tp->tsoffset = 0;
4511f255691SPriyaranjan Jha 	tp->rack.reo_wnd_steps = 1;
452ceaa1fefSAndrey Vagin 
453900f65d3SNeal Cardwell 	sk->sk_write_space = sk_stream_write_space;
454900f65d3SNeal Cardwell 	sock_set_flag(sk, SOCK_USE_WRITE_QUEUE);
455900f65d3SNeal Cardwell 
456900f65d3SNeal Cardwell 	icsk->icsk_sync_mss = tcp_sync_mss;
457900f65d3SNeal Cardwell 
45802739545SKuniyuki Iwashima 	WRITE_ONCE(sk->sk_sndbuf, READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_wmem[1]));
45902739545SKuniyuki Iwashima 	WRITE_ONCE(sk->sk_rcvbuf, READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_rmem[1]));
460900f65d3SNeal Cardwell 
461e993ffe3SPavel Begunkov 	set_bit(SOCK_SUPPORT_ZC, &sk->sk_socket->flags);
462900f65d3SNeal Cardwell 	sk_sockets_allocated_inc(sk);
463900f65d3SNeal Cardwell }
464900f65d3SNeal Cardwell EXPORT_SYMBOL(tcp_init_sock);
465900f65d3SNeal Cardwell 
4664e8cc228SEric Dumazet static void tcp_tx_timestamp(struct sock *sk, u16 tsflags)
4674ed2d765SWillem de Bruijn {
4684e8cc228SEric Dumazet 	struct sk_buff *skb = tcp_write_queue_tail(sk);
4694e8cc228SEric Dumazet 
470ad02c4f5SSoheil Hassas Yeganeh 	if (tsflags && skb) {
4714ed2d765SWillem de Bruijn 		struct skb_shared_info *shinfo = skb_shinfo(skb);
4726b084928SSoheil Hassas Yeganeh 		struct tcp_skb_cb *tcb = TCP_SKB_CB(skb);
4734ed2d765SWillem de Bruijn 
474c14ac945SSoheil Hassas Yeganeh 		sock_tx_timestamp(sk, tsflags, &shinfo->tx_flags);
4750a2cf20cSSoheil Hassas Yeganeh 		if (tsflags & SOF_TIMESTAMPING_TX_ACK)
4760a2cf20cSSoheil Hassas Yeganeh 			tcb->txstamp_ack = 1;
4770a2cf20cSSoheil Hassas Yeganeh 		if (tsflags & SOF_TIMESTAMPING_TX_RECORD_MASK)
4784ed2d765SWillem de Bruijn 			shinfo->tskey = TCP_SKB_CB(skb)->seq + skb->len - 1;
4794ed2d765SWillem de Bruijn 	}
480f066e2b0SWillem de Bruijn }
4814ed2d765SWillem de Bruijn 
48205dc72abSEric Dumazet static bool tcp_stream_is_readable(struct sock *sk, int target)
4838934ce2fSJohn Fastabend {
48405dc72abSEric Dumazet 	if (tcp_epollin_ready(sk, target))
48505dc72abSEric Dumazet 		return true;
4867b50ecfcSCong Wang 	return sk_is_readable(sk);
4878934ce2fSJohn Fastabend }
4888934ce2fSJohn Fastabend 
4891da177e4SLinus Torvalds /*
490a11e1d43SLinus Torvalds  *	Wait for a TCP event.
491a11e1d43SLinus Torvalds  *
492a11e1d43SLinus Torvalds  *	Note that we don't need to lock the socket, as the upper poll layers
493a11e1d43SLinus Torvalds  *	take care of normal races (between the test and the event) and we don't
494a11e1d43SLinus Torvalds  *	go look at any of the socket buffers directly.
4951da177e4SLinus Torvalds  */
496a11e1d43SLinus Torvalds __poll_t tcp_poll(struct file *file, struct socket *sock, poll_table *wait)
4971da177e4SLinus Torvalds {
498a11e1d43SLinus Torvalds 	__poll_t mask;
4991da177e4SLinus Torvalds 	struct sock *sk = sock->sk;
500cf533ea5SEric Dumazet 	const struct tcp_sock *tp = tcp_sk(sk);
501e14cadfdSEric Dumazet 	u8 shutdown;
50200fd38d9SEric Dumazet 	int state;
5031da177e4SLinus Torvalds 
50489ab066dSKarsten Graul 	sock_poll_wait(file, sock, wait);
505a11e1d43SLinus Torvalds 
506986ffdfdSYafang Shao 	state = inet_sk_state_load(sk);
50700fd38d9SEric Dumazet 	if (state == TCP_LISTEN)
508dc40c7bcSArnaldo Carvalho de Melo 		return inet_csk_listen_poll(sk);
5091da177e4SLinus Torvalds 
510a11e1d43SLinus Torvalds 	/* Socket is not locked. We are protected from async events
511a11e1d43SLinus Torvalds 	 * by poll logic and correct handling of state changes
512a11e1d43SLinus Torvalds 	 * made by other threads is impossible in any case.
513a11e1d43SLinus Torvalds 	 */
514a11e1d43SLinus Torvalds 
515a11e1d43SLinus Torvalds 	mask = 0;
516a11e1d43SLinus Torvalds 
5171da177e4SLinus Torvalds 	/*
518a9a08845SLinus Torvalds 	 * EPOLLHUP is certainly not done right. But poll() doesn't
5191da177e4SLinus Torvalds 	 * have a notion of HUP in just one direction, and for a
5201da177e4SLinus Torvalds 	 * socket the read side is more interesting.
5211da177e4SLinus Torvalds 	 *
522a9a08845SLinus Torvalds 	 * Some poll() documentation says that EPOLLHUP is incompatible
523a9a08845SLinus Torvalds 	 * with the EPOLLOUT/POLLWR flags, so somebody should check this
5241da177e4SLinus Torvalds 	 * all. But careful, it tends to be safer to return too many
5251da177e4SLinus Torvalds 	 * bits than too few, and you can easily break real applications
5261da177e4SLinus Torvalds 	 * if you don't tell them that something has hung up!
5271da177e4SLinus Torvalds 	 *
5281da177e4SLinus Torvalds 	 * Check-me.
5291da177e4SLinus Torvalds 	 *
530a9a08845SLinus Torvalds 	 * Check number 1. EPOLLHUP is _UNMASKABLE_ event (see UNIX98 and
5311da177e4SLinus Torvalds 	 * our fs/select.c). It means that after we received EOF,
5321da177e4SLinus Torvalds 	 * poll always returns immediately, making impossible poll() on write()
533a9a08845SLinus Torvalds 	 * in state CLOSE_WAIT. One solution is evident --- to set EPOLLHUP
5341da177e4SLinus Torvalds 	 * if and only if shutdown has been made in both directions.
5351da177e4SLinus Torvalds 	 * Actually, it is interesting to look how Solaris and DUX
536a9a08845SLinus Torvalds 	 * solve this dilemma. I would prefer, if EPOLLHUP were maskable,
5371da177e4SLinus Torvalds 	 * then we could set it on SND_SHUTDOWN. BTW examples given
5381da177e4SLinus Torvalds 	 * in Stevens' books assume exactly this behaviour, it explains
539a9a08845SLinus Torvalds 	 * why EPOLLHUP is incompatible with EPOLLOUT.	--ANK
5401da177e4SLinus Torvalds 	 *
5411da177e4SLinus Torvalds 	 * NOTE. Check for TCP_CLOSE is added. The goal is to prevent
5421da177e4SLinus Torvalds 	 * blocking on fresh not-connected or disconnected socket. --ANK
5431da177e4SLinus Torvalds 	 */
544e14cadfdSEric Dumazet 	shutdown = READ_ONCE(sk->sk_shutdown);
545e14cadfdSEric Dumazet 	if (shutdown == SHUTDOWN_MASK || state == TCP_CLOSE)
546a9a08845SLinus Torvalds 		mask |= EPOLLHUP;
547e14cadfdSEric Dumazet 	if (shutdown & RCV_SHUTDOWN)
548a9a08845SLinus Torvalds 		mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
5491da177e4SLinus Torvalds 
5508336886fSJerry Chu 	/* Connected or passive Fast Open socket? */
55100fd38d9SEric Dumazet 	if (state != TCP_SYN_SENT &&
552d983ea6fSEric Dumazet 	    (state != TCP_SYN_RECV || rcu_access_pointer(tp->fastopen_rsk))) {
553c7004482SDavid S. Miller 		int target = sock_rcvlowat(sk, 0, INT_MAX);
5547b6a893aSEric Dumazet 		u16 urg_data = READ_ONCE(tp->urg_data);
555c7004482SDavid S. Miller 
556b96c51bdSEric Dumazet 		if (unlikely(urg_data) &&
5577b6a893aSEric Dumazet 		    READ_ONCE(tp->urg_seq) == READ_ONCE(tp->copied_seq) &&
5587b6a893aSEric Dumazet 		    !sock_flag(sk, SOCK_URGINLINE))
559b634f875SAlexandra Kossovsky 			target++;
560c7004482SDavid S. Miller 
56105dc72abSEric Dumazet 		if (tcp_stream_is_readable(sk, target))
562a9a08845SLinus Torvalds 			mask |= EPOLLIN | EPOLLRDNORM;
5631da177e4SLinus Torvalds 
564e14cadfdSEric Dumazet 		if (!(shutdown & SEND_SHUTDOWN)) {
5658ba3c9d1SSoheil Hassas Yeganeh 			if (__sk_stream_is_writeable(sk, 1)) {
566a9a08845SLinus Torvalds 				mask |= EPOLLOUT | EPOLLWRNORM;
5671da177e4SLinus Torvalds 			} else {  /* send SIGIO later */
5689cd3e072SEric Dumazet 				sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
5691da177e4SLinus Torvalds 				set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
5701da177e4SLinus Torvalds 
5711da177e4SLinus Torvalds 				/* Race breaker. If space is freed after
5721da177e4SLinus Torvalds 				 * wspace test but before the flags are set,
5733c715127Sjbaron@akamai.com 				 * IO signal will be lost. Memory barrier
5743c715127Sjbaron@akamai.com 				 * pairs with the input side.
5751da177e4SLinus Torvalds 				 */
5763c715127Sjbaron@akamai.com 				smp_mb__after_atomic();
5778ba3c9d1SSoheil Hassas Yeganeh 				if (__sk_stream_is_writeable(sk, 1))
578a9a08845SLinus Torvalds 					mask |= EPOLLOUT | EPOLLWRNORM;
5791da177e4SLinus Torvalds 			}
580d84ba638SKOSAKI Motohiro 		} else
581a9a08845SLinus Torvalds 			mask |= EPOLLOUT | EPOLLWRNORM;
5821da177e4SLinus Torvalds 
5837b6a893aSEric Dumazet 		if (urg_data & TCP_URG_VALID)
584a9a08845SLinus Torvalds 			mask |= EPOLLPRI;
585d68be71eSDavide Caratti 	} else if (state == TCP_SYN_SENT && inet_sk(sk)->defer_connect) {
58619f6d3f3SWei Wang 		/* Active TCP fastopen socket with defer_connect
587a9a08845SLinus Torvalds 		 * Return EPOLLOUT so application can call write()
58819f6d3f3SWei Wang 		 * in order for kernel to generate SYN+data
58919f6d3f3SWei Wang 		 */
590a9a08845SLinus Torvalds 		mask |= EPOLLOUT | EPOLLWRNORM;
5911da177e4SLinus Torvalds 	}
592a4d25803STom Marshall 	/* This barrier is coupled with smp_wmb() in tcp_reset() */
593a4d25803STom Marshall 	smp_rmb();
594e13ec3daSEric Dumazet 	if (READ_ONCE(sk->sk_err) ||
595e13ec3daSEric Dumazet 	    !skb_queue_empty_lockless(&sk->sk_error_queue))
596a9a08845SLinus Torvalds 		mask |= EPOLLERR;
597a4d25803STom Marshall 
5981da177e4SLinus Torvalds 	return mask;
5991da177e4SLinus Torvalds }
600a11e1d43SLinus Torvalds EXPORT_SYMBOL(tcp_poll);
6011da177e4SLinus Torvalds 
602e1d001faSBreno Leitao int tcp_ioctl(struct sock *sk, int cmd, int *karg)
6031da177e4SLinus Torvalds {
6041da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
6051da177e4SLinus Torvalds 	int answ;
6060e71c55cSEric Dumazet 	bool slow;
6071da177e4SLinus Torvalds 
6081da177e4SLinus Torvalds 	switch (cmd) {
6091da177e4SLinus Torvalds 	case SIOCINQ:
6101da177e4SLinus Torvalds 		if (sk->sk_state == TCP_LISTEN)
6111da177e4SLinus Torvalds 			return -EINVAL;
6121da177e4SLinus Torvalds 
6130e71c55cSEric Dumazet 		slow = lock_sock_fast(sk);
614473bd239STom Herbert 		answ = tcp_inq(sk);
6150e71c55cSEric Dumazet 		unlock_sock_fast(sk, slow);
6161da177e4SLinus Torvalds 		break;
6171da177e4SLinus Torvalds 	case SIOCATMARK:
6187b6a893aSEric Dumazet 		answ = READ_ONCE(tp->urg_data) &&
619d9b55bf7SEric Dumazet 		       READ_ONCE(tp->urg_seq) == READ_ONCE(tp->copied_seq);
6201da177e4SLinus Torvalds 		break;
6211da177e4SLinus Torvalds 	case SIOCOUTQ:
6221da177e4SLinus Torvalds 		if (sk->sk_state == TCP_LISTEN)
6231da177e4SLinus Torvalds 			return -EINVAL;
6241da177e4SLinus Torvalds 
6251da177e4SLinus Torvalds 		if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
6261da177e4SLinus Torvalds 			answ = 0;
6271da177e4SLinus Torvalds 		else
6280f317464SEric Dumazet 			answ = READ_ONCE(tp->write_seq) - tp->snd_una;
6291da177e4SLinus Torvalds 		break;
6302f4e1b39SMario Schuknecht 	case SIOCOUTQNSD:
6312f4e1b39SMario Schuknecht 		if (sk->sk_state == TCP_LISTEN)
6322f4e1b39SMario Schuknecht 			return -EINVAL;
6332f4e1b39SMario Schuknecht 
6342f4e1b39SMario Schuknecht 		if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
6352f4e1b39SMario Schuknecht 			answ = 0;
6362f4e1b39SMario Schuknecht 		else
637e0d694d6SEric Dumazet 			answ = READ_ONCE(tp->write_seq) -
638e0d694d6SEric Dumazet 			       READ_ONCE(tp->snd_nxt);
6392f4e1b39SMario Schuknecht 		break;
6401da177e4SLinus Torvalds 	default:
6411da177e4SLinus Torvalds 		return -ENOIOCTLCMD;
6423ff50b79SStephen Hemminger 	}
6431da177e4SLinus Torvalds 
644e1d001faSBreno Leitao 	*karg = answ;
645e1d001faSBreno Leitao 	return 0;
6461da177e4SLinus Torvalds }
6474bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_ioctl);
6481da177e4SLinus Torvalds 
64904d8825cSPaolo Abeni void tcp_mark_push(struct tcp_sock *tp, struct sk_buff *skb)
6501da177e4SLinus Torvalds {
6514de075e0SEric Dumazet 	TCP_SKB_CB(skb)->tcp_flags |= TCPHDR_PSH;
6521da177e4SLinus Torvalds 	tp->pushed_seq = tp->write_seq;
6531da177e4SLinus Torvalds }
6541da177e4SLinus Torvalds 
655a2a385d6SEric Dumazet static inline bool forced_push(const struct tcp_sock *tp)
6561da177e4SLinus Torvalds {
6571da177e4SLinus Torvalds 	return after(tp->write_seq, tp->pushed_seq + (tp->max_window >> 1));
6581da177e4SLinus Torvalds }
6591da177e4SLinus Torvalds 
66004d8825cSPaolo Abeni void tcp_skb_entail(struct sock *sk, struct sk_buff *skb)
6611da177e4SLinus Torvalds {
6629e412ba7SIlpo Järvinen 	struct tcp_sock *tp = tcp_sk(sk);
663352d4800SArnaldo Carvalho de Melo 	struct tcp_skb_cb *tcb = TCP_SKB_CB(skb);
664352d4800SArnaldo Carvalho de Melo 
665352d4800SArnaldo Carvalho de Melo 	tcb->seq     = tcb->end_seq = tp->write_seq;
6664de075e0SEric Dumazet 	tcb->tcp_flags = TCPHDR_ACK;
667f4a775d1SEric Dumazet 	__skb_header_release(skb);
668fe067e8aSDavid S. Miller 	tcp_add_write_queue_tail(sk, skb);
669ab4e846aSEric Dumazet 	sk_wmem_queued_add(sk, skb->truesize);
6703ab224beSHideo Aoki 	sk_mem_charge(sk, skb->truesize);
67189ebd197SDavid S. Miller 	if (tp->nonagle & TCP_NAGLE_PUSH)
6721da177e4SLinus Torvalds 		tp->nonagle &= ~TCP_NAGLE_PUSH;
6736f021c62SEric Dumazet 
6746f021c62SEric Dumazet 	tcp_slow_start_after_idle_check(sk);
6751da177e4SLinus Torvalds }
6761da177e4SLinus Torvalds 
677afeca340SKrishna Kumar static inline void tcp_mark_urg(struct tcp_sock *tp, int flags)
6781da177e4SLinus Torvalds {
67933f5f57eSIlpo Järvinen 	if (flags & MSG_OOB)
6801da177e4SLinus Torvalds 		tp->snd_up = tp->write_seq;
6811da177e4SLinus Torvalds }
6821da177e4SLinus Torvalds 
683f54b3111SEric Dumazet /* If a not yet filled skb is pushed, do not send it if
684a181ceb5SEric Dumazet  * we have data packets in Qdisc or NIC queues :
685f54b3111SEric Dumazet  * Because TX completion will happen shortly, it gives a chance
686f54b3111SEric Dumazet  * to coalesce future sendmsg() payload into this skb, without
687f54b3111SEric Dumazet  * need for a timer, and with no latency trade off.
688f54b3111SEric Dumazet  * As packets containing data payload have a bigger truesize
689a181ceb5SEric Dumazet  * than pure acks (dataless) packets, the last checks prevent
690a181ceb5SEric Dumazet  * autocorking if we only have an ACK in Qdisc/NIC queues,
691a181ceb5SEric Dumazet  * or if TX completion was delayed after we processed ACK packet.
692f54b3111SEric Dumazet  */
693f54b3111SEric Dumazet static bool tcp_should_autocork(struct sock *sk, struct sk_buff *skb,
694f54b3111SEric Dumazet 				int size_goal)
6951da177e4SLinus Torvalds {
696f54b3111SEric Dumazet 	return skb->len < size_goal &&
69785225e6fSKuniyuki Iwashima 	       READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_autocorking) &&
698114f39feSEric Dumazet 	       !tcp_rtx_queue_empty(sk) &&
699b0de0cf4SEric Dumazet 	       refcount_read(&sk->sk_wmem_alloc) > skb->truesize &&
700b0de0cf4SEric Dumazet 	       tcp_skb_can_collapse_to(skb);
701f54b3111SEric Dumazet }
7029e412ba7SIlpo Järvinen 
70335b2c321SMat Martineau void tcp_push(struct sock *sk, int flags, int mss_now,
704f54b3111SEric Dumazet 	      int nonagle, int size_goal)
705f54b3111SEric Dumazet {
706f54b3111SEric Dumazet 	struct tcp_sock *tp = tcp_sk(sk);
707f54b3111SEric Dumazet 	struct sk_buff *skb;
708f54b3111SEric Dumazet 
709f54b3111SEric Dumazet 	skb = tcp_write_queue_tail(sk);
71075c119afSEric Dumazet 	if (!skb)
71175c119afSEric Dumazet 		return;
7121da177e4SLinus Torvalds 	if (!(flags & MSG_MORE) || forced_push(tp))
713f54b3111SEric Dumazet 		tcp_mark_push(tp, skb);
714afeca340SKrishna Kumar 
715afeca340SKrishna Kumar 	tcp_mark_urg(tp, flags);
716f54b3111SEric Dumazet 
717f54b3111SEric Dumazet 	if (tcp_should_autocork(sk, skb, size_goal)) {
718f54b3111SEric Dumazet 
719f54b3111SEric Dumazet 		/* avoid atomic op if TSQ_THROTTLED bit is already set */
7207aa5470cSEric Dumazet 		if (!test_bit(TSQ_THROTTLED, &sk->sk_tsq_flags)) {
721f54b3111SEric Dumazet 			NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPAUTOCORKING);
7227aa5470cSEric Dumazet 			set_bit(TSQ_THROTTLED, &sk->sk_tsq_flags);
7231da177e4SLinus Torvalds 		}
724a181ceb5SEric Dumazet 		/* It is possible TX completion already happened
725a181ceb5SEric Dumazet 		 * before we set TSQ_THROTTLED.
726a181ceb5SEric Dumazet 		 */
72714afee4bSReshetova, Elena 		if (refcount_read(&sk->sk_wmem_alloc) > skb->truesize)
728f54b3111SEric Dumazet 			return;
729f54b3111SEric Dumazet 	}
730f54b3111SEric Dumazet 
731f54b3111SEric Dumazet 	if (flags & MSG_MORE)
732f54b3111SEric Dumazet 		nonagle = TCP_NAGLE_CORK;
733f54b3111SEric Dumazet 
734f54b3111SEric Dumazet 	__tcp_push_pending_frames(sk, mss_now, nonagle);
7351da177e4SLinus Torvalds }
7361da177e4SLinus Torvalds 
7376ff7751dSAdrian Bunk static int tcp_splice_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb,
7389c55e01cSJens Axboe 				unsigned int offset, size_t len)
7399c55e01cSJens Axboe {
7409c55e01cSJens Axboe 	struct tcp_splice_state *tss = rd_desc->arg.data;
74133966dd0SWilly Tarreau 	int ret;
7429c55e01cSJens Axboe 
743a60e3cc7SHannes Frederic Sowa 	ret = skb_splice_bits(skb, skb->sk, offset, tss->pipe,
74425869262SAl Viro 			      min(rd_desc->count, len), tss->flags);
74533966dd0SWilly Tarreau 	if (ret > 0)
74633966dd0SWilly Tarreau 		rd_desc->count -= ret;
74733966dd0SWilly Tarreau 	return ret;
7489c55e01cSJens Axboe }
7499c55e01cSJens Axboe 
7509c55e01cSJens Axboe static int __tcp_splice_read(struct sock *sk, struct tcp_splice_state *tss)
7519c55e01cSJens Axboe {
7529c55e01cSJens Axboe 	/* Store TCP splice context information in read_descriptor_t. */
7539c55e01cSJens Axboe 	read_descriptor_t rd_desc = {
7549c55e01cSJens Axboe 		.arg.data = tss,
75533966dd0SWilly Tarreau 		.count	  = tss->len,
7569c55e01cSJens Axboe 	};
7579c55e01cSJens Axboe 
7589c55e01cSJens Axboe 	return tcp_read_sock(sk, &rd_desc, tcp_splice_data_recv);
7599c55e01cSJens Axboe }
7609c55e01cSJens Axboe 
7619c55e01cSJens Axboe /**
7629c55e01cSJens Axboe  *  tcp_splice_read - splice data from TCP socket to a pipe
7639c55e01cSJens Axboe  * @sock:	socket to splice from
7649c55e01cSJens Axboe  * @ppos:	position (not valid)
7659c55e01cSJens Axboe  * @pipe:	pipe to splice to
7669c55e01cSJens Axboe  * @len:	number of bytes to splice
7679c55e01cSJens Axboe  * @flags:	splice modifier flags
7689c55e01cSJens Axboe  *
7699c55e01cSJens Axboe  * Description:
7709c55e01cSJens Axboe  *    Will read pages from given socket and fill them into a pipe.
7719c55e01cSJens Axboe  *
7729c55e01cSJens Axboe  **/
7739c55e01cSJens Axboe ssize_t tcp_splice_read(struct socket *sock, loff_t *ppos,
7749c55e01cSJens Axboe 			struct pipe_inode_info *pipe, size_t len,
7759c55e01cSJens Axboe 			unsigned int flags)
7769c55e01cSJens Axboe {
7779c55e01cSJens Axboe 	struct sock *sk = sock->sk;
7789c55e01cSJens Axboe 	struct tcp_splice_state tss = {
7799c55e01cSJens Axboe 		.pipe = pipe,
7809c55e01cSJens Axboe 		.len = len,
7819c55e01cSJens Axboe 		.flags = flags,
7829c55e01cSJens Axboe 	};
7839c55e01cSJens Axboe 	long timeo;
7849c55e01cSJens Axboe 	ssize_t spliced;
7859c55e01cSJens Axboe 	int ret;
7869c55e01cSJens Axboe 
7873a047bf8SChangli Gao 	sock_rps_record_flow(sk);
7889c55e01cSJens Axboe 	/*
7899c55e01cSJens Axboe 	 * We can't seek on a socket input
7909c55e01cSJens Axboe 	 */
7919c55e01cSJens Axboe 	if (unlikely(*ppos))
7929c55e01cSJens Axboe 		return -ESPIPE;
7939c55e01cSJens Axboe 
7949c55e01cSJens Axboe 	ret = spliced = 0;
7959c55e01cSJens Axboe 
7969c55e01cSJens Axboe 	lock_sock(sk);
7979c55e01cSJens Axboe 
79842324c62SEric Dumazet 	timeo = sock_rcvtimeo(sk, sock->file->f_flags & O_NONBLOCK);
7999c55e01cSJens Axboe 	while (tss.len) {
8009c55e01cSJens Axboe 		ret = __tcp_splice_read(sk, &tss);
8019c55e01cSJens Axboe 		if (ret < 0)
8029c55e01cSJens Axboe 			break;
8039c55e01cSJens Axboe 		else if (!ret) {
8049c55e01cSJens Axboe 			if (spliced)
8059c55e01cSJens Axboe 				break;
8069c55e01cSJens Axboe 			if (sock_flag(sk, SOCK_DONE))
8079c55e01cSJens Axboe 				break;
8089c55e01cSJens Axboe 			if (sk->sk_err) {
8099c55e01cSJens Axboe 				ret = sock_error(sk);
8109c55e01cSJens Axboe 				break;
8119c55e01cSJens Axboe 			}
8129c55e01cSJens Axboe 			if (sk->sk_shutdown & RCV_SHUTDOWN)
8139c55e01cSJens Axboe 				break;
8149c55e01cSJens Axboe 			if (sk->sk_state == TCP_CLOSE) {
8159c55e01cSJens Axboe 				/*
8169c55e01cSJens Axboe 				 * This occurs when user tries to read
8179c55e01cSJens Axboe 				 * from never connected socket.
8189c55e01cSJens Axboe 				 */
8199c55e01cSJens Axboe 				ret = -ENOTCONN;
8209c55e01cSJens Axboe 				break;
8219c55e01cSJens Axboe 			}
8229c55e01cSJens Axboe 			if (!timeo) {
8239c55e01cSJens Axboe 				ret = -EAGAIN;
8249c55e01cSJens Axboe 				break;
8259c55e01cSJens Axboe 			}
826ccf7abb9SEric Dumazet 			/* if __tcp_splice_read() got nothing while we have
827ccf7abb9SEric Dumazet 			 * an skb in receive queue, we do not want to loop.
828ccf7abb9SEric Dumazet 			 * This might happen with URG data.
829ccf7abb9SEric Dumazet 			 */
830ccf7abb9SEric Dumazet 			if (!skb_queue_empty(&sk->sk_receive_queue))
831ccf7abb9SEric Dumazet 				break;
832dfbafc99SSabrina Dubroca 			sk_wait_data(sk, &timeo, NULL);
8339c55e01cSJens Axboe 			if (signal_pending(current)) {
8349c55e01cSJens Axboe 				ret = sock_intr_errno(timeo);
8359c55e01cSJens Axboe 				break;
8369c55e01cSJens Axboe 			}
8379c55e01cSJens Axboe 			continue;
8389c55e01cSJens Axboe 		}
8399c55e01cSJens Axboe 		tss.len -= ret;
8409c55e01cSJens Axboe 		spliced += ret;
8419c55e01cSJens Axboe 
8422fe11c9dSPavel Begunkov 		if (!tss.len || !timeo)
84333966dd0SWilly Tarreau 			break;
8449c55e01cSJens Axboe 		release_sock(sk);
8459c55e01cSJens Axboe 		lock_sock(sk);
8469c55e01cSJens Axboe 
8479c55e01cSJens Axboe 		if (sk->sk_err || sk->sk_state == TCP_CLOSE ||
84833966dd0SWilly Tarreau 		    (sk->sk_shutdown & RCV_SHUTDOWN) ||
8499c55e01cSJens Axboe 		    signal_pending(current))
8509c55e01cSJens Axboe 			break;
8519c55e01cSJens Axboe 	}
8529c55e01cSJens Axboe 
8539c55e01cSJens Axboe 	release_sock(sk);
8549c55e01cSJens Axboe 
8559c55e01cSJens Axboe 	if (spliced)
8569c55e01cSJens Axboe 		return spliced;
8579c55e01cSJens Axboe 
8589c55e01cSJens Axboe 	return ret;
8599c55e01cSJens Axboe }
8604bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_splice_read);
8619c55e01cSJens Axboe 
8625882efffSEric Dumazet struct sk_buff *tcp_stream_alloc_skb(struct sock *sk, gfp_t gfp,
863eb934478SEric Dumazet 				     bool force_schedule)
864f561d0f2SPavel Emelyanov {
865f561d0f2SPavel Emelyanov 	struct sk_buff *skb;
866f561d0f2SPavel Emelyanov 
8675882efffSEric Dumazet 	skb = alloc_skb_fclone(MAX_TCP_HEADER, gfp);
8688e4d980aSEric Dumazet 	if (likely(skb)) {
869eb934478SEric Dumazet 		bool mem_scheduled;
8708e4d980aSEric Dumazet 
8719b65b17dSTalal Ahmad 		skb->truesize = SKB_TRUESIZE(skb_end_offset(skb));
872eb934478SEric Dumazet 		if (force_schedule) {
873eb934478SEric Dumazet 			mem_scheduled = true;
8748e4d980aSEric Dumazet 			sk_forced_mem_schedule(sk, skb->truesize);
8758e4d980aSEric Dumazet 		} else {
876eb934478SEric Dumazet 			mem_scheduled = sk_wmem_schedule(sk, skb->truesize);
8778e4d980aSEric Dumazet 		}
878eb934478SEric Dumazet 		if (likely(mem_scheduled)) {
8798a794df6SEric Dumazet 			skb_reserve(skb, MAX_TCP_HEADER);
880a52fe46eSEric Dumazet 			skb->ip_summed = CHECKSUM_PARTIAL;
881e2080072SEric Dumazet 			INIT_LIST_HEAD(&skb->tcp_tsorted_anchor);
882f561d0f2SPavel Emelyanov 			return skb;
883f561d0f2SPavel Emelyanov 		}
884f561d0f2SPavel Emelyanov 		__kfree_skb(skb);
885f561d0f2SPavel Emelyanov 	} else {
8865c52ba17SPavel Emelyanov 		sk->sk_prot->enter_memory_pressure(sk);
887f561d0f2SPavel Emelyanov 		sk_stream_moderate_sndbuf(sk);
888f561d0f2SPavel Emelyanov 	}
889f561d0f2SPavel Emelyanov 	return NULL;
890f561d0f2SPavel Emelyanov }
891f561d0f2SPavel Emelyanov 
8920c54b85fSIlpo Järvinen static unsigned int tcp_xmit_size_goal(struct sock *sk, u32 mss_now,
8930c54b85fSIlpo Järvinen 				       int large_allowed)
8940c54b85fSIlpo Järvinen {
8950c54b85fSIlpo Järvinen 	struct tcp_sock *tp = tcp_sk(sk);
8966c09fa09SEric Dumazet 	u32 new_size_goal, size_goal;
8970c54b85fSIlpo Järvinen 
89874d4a8f8SEric Dumazet 	if (!large_allowed)
899605ad7f1SEric Dumazet 		return mss_now;
9000c54b85fSIlpo Järvinen 
9016c09fa09SEric Dumazet 	/* Note : tcp_tso_autosize() will eventually split this later */
902ab14f180SDavid Ahern 	new_size_goal = tcp_bound_to_half_wnd(tp, sk->sk_gso_max_size);
9032a3a041cSIlpo Järvinen 
9042a3a041cSIlpo Järvinen 	/* We try hard to avoid divides here */
905605ad7f1SEric Dumazet 	size_goal = tp->gso_segs * mss_now;
906605ad7f1SEric Dumazet 	if (unlikely(new_size_goal < size_goal ||
907605ad7f1SEric Dumazet 		     new_size_goal >= size_goal + mss_now)) {
908605ad7f1SEric Dumazet 		tp->gso_segs = min_t(u16, new_size_goal / mss_now,
9091485348dSBen Hutchings 				     sk->sk_gso_max_segs);
910605ad7f1SEric Dumazet 		size_goal = tp->gso_segs * mss_now;
9110c54b85fSIlpo Järvinen 	}
9120c54b85fSIlpo Järvinen 
913605ad7f1SEric Dumazet 	return max(size_goal, mss_now);
9140c54b85fSIlpo Järvinen }
9150c54b85fSIlpo Järvinen 
91635b2c321SMat Martineau int tcp_send_mss(struct sock *sk, int *size_goal, int flags)
9170c54b85fSIlpo Järvinen {
9180c54b85fSIlpo Järvinen 	int mss_now;
9190c54b85fSIlpo Järvinen 
9200c54b85fSIlpo Järvinen 	mss_now = tcp_current_mss(sk);
9210c54b85fSIlpo Järvinen 	*size_goal = tcp_xmit_size_goal(sk, mss_now, !(flags & MSG_OOB));
9220c54b85fSIlpo Järvinen 
9230c54b85fSIlpo Järvinen 	return mss_now;
9240c54b85fSIlpo Järvinen }
9250c54b85fSIlpo Järvinen 
926dc97391eSDavid Howells /* In some cases, both sendmsg() could have added an skb to the write queue,
927dc97391eSDavid Howells  * but failed adding payload on it.  We need to remove it to consume less
928dc97391eSDavid Howells  * memory, but more importantly be able to generate EPOLLOUT for Edge Trigger
929dc97391eSDavid Howells  * epoll() users.
930fdfc5c85SEric Dumazet  */
93127728ba8SEric Dumazet void tcp_remove_empty_skb(struct sock *sk)
932fdfc5c85SEric Dumazet {
93327728ba8SEric Dumazet 	struct sk_buff *skb = tcp_write_queue_tail(sk);
93427728ba8SEric Dumazet 
935cf12e6f9SJon Maxwell 	if (skb && TCP_SKB_CB(skb)->seq == TCP_SKB_CB(skb)->end_seq) {
936fdfc5c85SEric Dumazet 		tcp_unlink_write_queue(skb, sk);
937fdfc5c85SEric Dumazet 		if (tcp_write_queue_empty(sk))
938fdfc5c85SEric Dumazet 			tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
93903271f3aSTalal Ahmad 		tcp_wmem_free_skb(sk, skb);
940fdfc5c85SEric Dumazet 	}
941fdfc5c85SEric Dumazet }
942fdfc5c85SEric Dumazet 
943f8d9d938SEric Dumazet /* skb changing from pure zc to mixed, must charge zc */
944f8d9d938SEric Dumazet static int tcp_downgrade_zcopy_pure(struct sock *sk, struct sk_buff *skb)
945f8d9d938SEric Dumazet {
946f8d9d938SEric Dumazet 	if (unlikely(skb_zcopy_pure(skb))) {
947f8d9d938SEric Dumazet 		u32 extra = skb->truesize -
948f8d9d938SEric Dumazet 			    SKB_TRUESIZE(skb_end_offset(skb));
949f8d9d938SEric Dumazet 
950f8d9d938SEric Dumazet 		if (!sk_wmem_schedule(sk, extra))
951f8d9d938SEric Dumazet 			return -ENOMEM;
952f8d9d938SEric Dumazet 
953f8d9d938SEric Dumazet 		sk_mem_charge(sk, extra);
954f8d9d938SEric Dumazet 		skb_shinfo(skb)->flags &= ~SKBFL_PURE_ZEROCOPY;
955f8d9d938SEric Dumazet 	}
956f8d9d938SEric Dumazet 	return 0;
957f8d9d938SEric Dumazet }
958f8d9d938SEric Dumazet 
959849b425cSEric Dumazet 
960fbf93406SEric Dumazet int tcp_wmem_schedule(struct sock *sk, int copy)
961f54755f6SEric Dumazet {
962f54755f6SEric Dumazet 	int left;
963f54755f6SEric Dumazet 
964f54755f6SEric Dumazet 	if (likely(sk_wmem_schedule(sk, copy)))
965f54755f6SEric Dumazet 		return copy;
966f54755f6SEric Dumazet 
967f54755f6SEric Dumazet 	/* We could be in trouble if we have nothing queued.
968f54755f6SEric Dumazet 	 * Use whatever is left in sk->sk_forward_alloc and tcp_wmem[0]
969f54755f6SEric Dumazet 	 * to guarantee some progress.
970f54755f6SEric Dumazet 	 */
971f54755f6SEric Dumazet 	left = sock_net(sk)->ipv4.sysctl_tcp_wmem[0] - sk->sk_wmem_queued;
972f54755f6SEric Dumazet 	if (left > 0)
973f54755f6SEric Dumazet 		sk_forced_mem_schedule(sk, min(left, copy));
974f54755f6SEric Dumazet 	return min(copy, sk->sk_forward_alloc);
975f54755f6SEric Dumazet }
976f54755f6SEric Dumazet 
977cf60af03SYuchung Cheng void tcp_free_fastopen_req(struct tcp_sock *tp)
978cf60af03SYuchung Cheng {
97900db4124SIan Morris 	if (tp->fastopen_req) {
980cf60af03SYuchung Cheng 		kfree(tp->fastopen_req);
981cf60af03SYuchung Cheng 		tp->fastopen_req = NULL;
982cf60af03SYuchung Cheng 	}
983cf60af03SYuchung Cheng }
984cf60af03SYuchung Cheng 
9853242abebSBenjamin Hesmans int tcp_sendmsg_fastopen(struct sock *sk, struct msghdr *msg, int *copied,
9863242abebSBenjamin Hesmans 			 size_t size, struct ubuf_info *uarg)
987cf60af03SYuchung Cheng {
988cf60af03SYuchung Cheng 	struct tcp_sock *tp = tcp_sk(sk);
98919f6d3f3SWei Wang 	struct inet_sock *inet = inet_sk(sk);
990ba615f67SWei Wang 	struct sockaddr *uaddr = msg->msg_name;
991cf60af03SYuchung Cheng 	int err, flags;
992cf60af03SYuchung Cheng 
9935a542133SKuniyuki Iwashima 	if (!(READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_fastopen) &
9945a542133SKuniyuki Iwashima 	      TFO_CLIENT_ENABLE) ||
995ba615f67SWei Wang 	    (uaddr && msg->msg_namelen >= sizeof(uaddr->sa_family) &&
996ba615f67SWei Wang 	     uaddr->sa_family == AF_UNSPEC))
997cf60af03SYuchung Cheng 		return -EOPNOTSUPP;
99800db4124SIan Morris 	if (tp->fastopen_req)
999cf60af03SYuchung Cheng 		return -EALREADY; /* Another Fast Open is in progress */
1000cf60af03SYuchung Cheng 
1001cf60af03SYuchung Cheng 	tp->fastopen_req = kzalloc(sizeof(struct tcp_fastopen_request),
1002cf60af03SYuchung Cheng 				   sk->sk_allocation);
100351456b29SIan Morris 	if (unlikely(!tp->fastopen_req))
1004cf60af03SYuchung Cheng 		return -ENOBUFS;
1005cf60af03SYuchung Cheng 	tp->fastopen_req->data = msg;
1006f5ddcbbbSEric Dumazet 	tp->fastopen_req->size = size;
1007f859a448SWillem de Bruijn 	tp->fastopen_req->uarg = uarg;
1008cf60af03SYuchung Cheng 
100919f6d3f3SWei Wang 	if (inet->defer_connect) {
101019f6d3f3SWei Wang 		err = tcp_connect(sk);
101119f6d3f3SWei Wang 		/* Same failure procedure as in tcp_v4/6_connect */
101219f6d3f3SWei Wang 		if (err) {
101319f6d3f3SWei Wang 			tcp_set_state(sk, TCP_CLOSE);
101419f6d3f3SWei Wang 			inet->inet_dport = 0;
101519f6d3f3SWei Wang 			sk->sk_route_caps = 0;
101619f6d3f3SWei Wang 		}
101719f6d3f3SWei Wang 	}
1018cf60af03SYuchung Cheng 	flags = (msg->msg_flags & MSG_DONTWAIT) ? O_NONBLOCK : 0;
1019ba615f67SWei Wang 	err = __inet_stream_connect(sk->sk_socket, uaddr,
10203979ad7eSWilly Tarreau 				    msg->msg_namelen, flags, 1);
10217db92362SWei Wang 	/* fastopen_req could already be freed in __inet_stream_connect
10227db92362SWei Wang 	 * if the connection times out or gets rst
10237db92362SWei Wang 	 */
10247db92362SWei Wang 	if (tp->fastopen_req) {
1025f5ddcbbbSEric Dumazet 		*copied = tp->fastopen_req->copied;
1026cf60af03SYuchung Cheng 		tcp_free_fastopen_req(tp);
10277db92362SWei Wang 		inet->defer_connect = 0;
10287db92362SWei Wang 	}
1029cf60af03SYuchung Cheng 	return err;
1030cf60af03SYuchung Cheng }
1031cf60af03SYuchung Cheng 
1032306b13ebSTom Herbert int tcp_sendmsg_locked(struct sock *sk, struct msghdr *msg, size_t size)
10331da177e4SLinus Torvalds {
10341da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
1035f214f915SWillem de Bruijn 	struct ubuf_info *uarg = NULL;
10361da177e4SLinus Torvalds 	struct sk_buff *skb;
1037c14ac945SSoheil Hassas Yeganeh 	struct sockcm_cookie sockc;
103857be5bdaSAl Viro 	int flags, err, copied = 0;
103957be5bdaSAl Viro 	int mss_now = 0, size_goal, copied_syn = 0;
10401a991488SEric Dumazet 	int process_backlog = 0;
1041270a1c3dSDavid Howells 	int zc = 0;
10421da177e4SLinus Torvalds 	long timeo;
10431da177e4SLinus Torvalds 
10441da177e4SLinus Torvalds 	flags = msg->msg_flags;
1045f214f915SWillem de Bruijn 
1046eb315a7dSPavel Begunkov 	if ((flags & MSG_ZEROCOPY) && size) {
1047eb315a7dSPavel Begunkov 		if (msg->msg_ubuf) {
1048eb315a7dSPavel Begunkov 			uarg = msg->msg_ubuf;
1049270a1c3dSDavid Howells 			if (sk->sk_route_caps & NETIF_F_SG)
1050270a1c3dSDavid Howells 				zc = MSG_ZEROCOPY;
1051eb315a7dSPavel Begunkov 		} else if (sock_flag(sk, SOCK_ZEROCOPY)) {
1052eea96a3eSPavel Begunkov 			skb = tcp_write_queue_tail(sk);
10538c793822SJonathan Lemon 			uarg = msg_zerocopy_realloc(sk, size, skb_zcopy(skb));
1054f214f915SWillem de Bruijn 			if (!uarg) {
1055f214f915SWillem de Bruijn 				err = -ENOBUFS;
1056f214f915SWillem de Bruijn 				goto out_err;
1057f214f915SWillem de Bruijn 			}
1058270a1c3dSDavid Howells 			if (sk->sk_route_caps & NETIF_F_SG)
1059270a1c3dSDavid Howells 				zc = MSG_ZEROCOPY;
1060270a1c3dSDavid Howells 			else
1061e7d2b510SPavel Begunkov 				uarg_to_msgzc(uarg)->zerocopy = 0;
1062f214f915SWillem de Bruijn 		}
1063270a1c3dSDavid Howells 	} else if (unlikely(msg->msg_flags & MSG_SPLICE_PAGES) && size) {
1064270a1c3dSDavid Howells 		if (sk->sk_route_caps & NETIF_F_SG)
1065270a1c3dSDavid Howells 			zc = MSG_SPLICE_PAGES;
1066eb315a7dSPavel Begunkov 	}
1067f214f915SWillem de Bruijn 
106816ae6aa1SYuchung Cheng 	if (unlikely(flags & MSG_FASTOPEN || inet_sk(sk)->defer_connect) &&
106916ae6aa1SYuchung Cheng 	    !tp->repair) {
1070f859a448SWillem de Bruijn 		err = tcp_sendmsg_fastopen(sk, msg, &copied_syn, size, uarg);
1071cf60af03SYuchung Cheng 		if (err == -EINPROGRESS && copied_syn > 0)
1072cf60af03SYuchung Cheng 			goto out;
1073cf60af03SYuchung Cheng 		else if (err)
1074cf60af03SYuchung Cheng 			goto out_err;
1075cf60af03SYuchung Cheng 	}
1076cf60af03SYuchung Cheng 
10771da177e4SLinus Torvalds 	timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
10781da177e4SLinus Torvalds 
1079d7722e85SSoheil Hassas Yeganeh 	tcp_rate_check_app_limited(sk);  /* is sending application-limited? */
1080d7722e85SSoheil Hassas Yeganeh 
10818336886fSJerry Chu 	/* Wait for a connection to finish. One exception is TCP Fast Open
10828336886fSJerry Chu 	 * (passive side) where data is allowed to be sent before a connection
10838336886fSJerry Chu 	 * is fully established.
10848336886fSJerry Chu 	 */
10858336886fSJerry Chu 	if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) &&
10868336886fSJerry Chu 	    !tcp_passive_fastopen(sk)) {
1087686a5624SYuvaraja Mariappan 		err = sk_stream_wait_connect(sk, &timeo);
1088686a5624SYuvaraja Mariappan 		if (err != 0)
1089cf60af03SYuchung Cheng 			goto do_error;
10908336886fSJerry Chu 	}
10911da177e4SLinus Torvalds 
1092c0e88ff0SPavel Emelyanov 	if (unlikely(tp->repair)) {
1093c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_RECV_QUEUE) {
1094c0e88ff0SPavel Emelyanov 			copied = tcp_send_rcvq(sk, msg, size);
10955924f17aSChristoph Paasch 			goto out_nopush;
1096c0e88ff0SPavel Emelyanov 		}
1097c0e88ff0SPavel Emelyanov 
1098c0e88ff0SPavel Emelyanov 		err = -EINVAL;
1099c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_NO_QUEUE)
1100c0e88ff0SPavel Emelyanov 			goto out_err;
1101c0e88ff0SPavel Emelyanov 
1102c0e88ff0SPavel Emelyanov 		/* 'common' sending to sendq */
1103c0e88ff0SPavel Emelyanov 	}
1104c0e88ff0SPavel Emelyanov 
1105657a0667SWillem de Bruijn 	sockcm_init(&sockc, sk);
1106c14ac945SSoheil Hassas Yeganeh 	if (msg->msg_controllen) {
1107c14ac945SSoheil Hassas Yeganeh 		err = sock_cmsg_send(sk, msg, &sockc);
1108c14ac945SSoheil Hassas Yeganeh 		if (unlikely(err)) {
1109c14ac945SSoheil Hassas Yeganeh 			err = -EINVAL;
1110c14ac945SSoheil Hassas Yeganeh 			goto out_err;
1111c14ac945SSoheil Hassas Yeganeh 		}
1112c14ac945SSoheil Hassas Yeganeh 	}
1113c14ac945SSoheil Hassas Yeganeh 
11141da177e4SLinus Torvalds 	/* This should be in poll */
11159cd3e072SEric Dumazet 	sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
11161da177e4SLinus Torvalds 
11171da177e4SLinus Torvalds 	/* Ok commence sending. */
11181da177e4SLinus Torvalds 	copied = 0;
11191da177e4SLinus Torvalds 
1120d41a69f1SEric Dumazet restart:
1121d41a69f1SEric Dumazet 	mss_now = tcp_send_mss(sk, &size_goal, flags);
1122d41a69f1SEric Dumazet 
11231da177e4SLinus Torvalds 	err = -EPIPE;
11241da177e4SLinus Torvalds 	if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
112579d8665bSEric Dumazet 		goto do_error;
11261da177e4SLinus Torvalds 
112701e97e65SAl Viro 	while (msg_data_left(msg)) {
1128270a1c3dSDavid Howells 		ssize_t copy = 0;
11291da177e4SLinus Torvalds 
1130fe067e8aSDavid S. Miller 		skb = tcp_write_queue_tail(sk);
113165ec6097SEric Dumazet 		if (skb)
113265ec6097SEric Dumazet 			copy = size_goal - skb->len;
11331da177e4SLinus Torvalds 
1134c134ecb8SMartin KaFai Lau 		if (copy <= 0 || !tcp_skb_can_collapse_to(skb)) {
11353613b3dbSEric Dumazet 			bool first_skb;
11363613b3dbSEric Dumazet 
11371da177e4SLinus Torvalds new_segment:
11381da177e4SLinus Torvalds 			if (!sk_stream_memory_free(sk))
1139afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
11401da177e4SLinus Torvalds 
11411a991488SEric Dumazet 			if (unlikely(process_backlog >= 16)) {
11421a991488SEric Dumazet 				process_backlog = 0;
11431a991488SEric Dumazet 				if (sk_flush_backlog(sk))
1144d41a69f1SEric Dumazet 					goto restart;
1145d4011239SEric Dumazet 			}
114675c119afSEric Dumazet 			first_skb = tcp_rtx_and_write_queues_empty(sk);
11475882efffSEric Dumazet 			skb = tcp_stream_alloc_skb(sk, sk->sk_allocation,
11483613b3dbSEric Dumazet 						   first_skb);
11491da177e4SLinus Torvalds 			if (!skb)
1150afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
11511da177e4SLinus Torvalds 
11521a991488SEric Dumazet 			process_backlog++;
11531da177e4SLinus Torvalds 
115404d8825cSPaolo Abeni 			tcp_skb_entail(sk, skb);
1155c1b4a7e6SDavid S. Miller 			copy = size_goal;
11569d186cacSAndrey Vagin 
11579d186cacSAndrey Vagin 			/* All packets are restored as if they have
1158d3edd06eSEric Dumazet 			 * already been sent. skb_mstamp_ns isn't set to
11599d186cacSAndrey Vagin 			 * avoid wrong rtt estimation.
11609d186cacSAndrey Vagin 			 */
11619d186cacSAndrey Vagin 			if (tp->repair)
11629d186cacSAndrey Vagin 				TCP_SKB_CB(skb)->sacked |= TCPCB_REPAIRED;
11631da177e4SLinus Torvalds 		}
11641da177e4SLinus Torvalds 
11651da177e4SLinus Torvalds 		/* Try to append data to the end of skb. */
116601e97e65SAl Viro 		if (copy > msg_data_left(msg))
116701e97e65SAl Viro 			copy = msg_data_left(msg);
11681da177e4SLinus Torvalds 
1169270a1c3dSDavid Howells 		if (zc == 0) {
11705640f768SEric Dumazet 			bool merge = true;
11711da177e4SLinus Torvalds 			int i = skb_shinfo(skb)->nr_frags;
11725640f768SEric Dumazet 			struct page_frag *pfrag = sk_page_frag(sk);
1173761965eaSEric Dumazet 
11745640f768SEric Dumazet 			if (!sk_page_frag_refill(sk, pfrag))
1175afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
1176761965eaSEric Dumazet 
11775640f768SEric Dumazet 			if (!skb_can_coalesce(skb, i, pfrag->page,
11785640f768SEric Dumazet 					      pfrag->offset)) {
1179657b991aSKuniyuki Iwashima 				if (i >= READ_ONCE(sysctl_max_skb_frags)) {
11801da177e4SLinus Torvalds 					tcp_mark_push(tp, skb);
11811da177e4SLinus Torvalds 					goto new_segment;
11821da177e4SLinus Torvalds 				}
11835640f768SEric Dumazet 				merge = false;
11845640f768SEric Dumazet 			}
1185ef015786SHerbert Xu 
11865640f768SEric Dumazet 			copy = min_t(int, copy, pfrag->size - pfrag->offset);
1187ef015786SHerbert Xu 
1188eb315a7dSPavel Begunkov 			if (unlikely(skb_zcopy_pure(skb) || skb_zcopy_managed(skb))) {
1189849b425cSEric Dumazet 				if (tcp_downgrade_zcopy_pure(sk, skb))
1190849b425cSEric Dumazet 					goto wait_for_space;
1191eb315a7dSPavel Begunkov 				skb_zcopy_downgrade_managed(skb);
1192eb315a7dSPavel Begunkov 			}
1193849b425cSEric Dumazet 
1194849b425cSEric Dumazet 			copy = tcp_wmem_schedule(sk, copy);
1195849b425cSEric Dumazet 			if (!copy)
1196afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
11971da177e4SLinus Torvalds 
119857be5bdaSAl Viro 			err = skb_copy_to_page_nocache(sk, &msg->msg_iter, skb,
11995640f768SEric Dumazet 						       pfrag->page,
12005640f768SEric Dumazet 						       pfrag->offset,
12015640f768SEric Dumazet 						       copy);
12025640f768SEric Dumazet 			if (err)
12031da177e4SLinus Torvalds 				goto do_error;
12041da177e4SLinus Torvalds 
12051da177e4SLinus Torvalds 			/* Update the skb. */
12061da177e4SLinus Torvalds 			if (merge) {
12079e903e08SEric Dumazet 				skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy);
12081da177e4SLinus Torvalds 			} else {
12095640f768SEric Dumazet 				skb_fill_page_desc(skb, i, pfrag->page,
12105640f768SEric Dumazet 						   pfrag->offset, copy);
12114e33e346SEric Dumazet 				page_ref_inc(pfrag->page);
12121da177e4SLinus Torvalds 			}
12135640f768SEric Dumazet 			pfrag->offset += copy;
1214270a1c3dSDavid Howells 		} else if (zc == MSG_ZEROCOPY)  {
12159b65b17dSTalal Ahmad 			/* First append to a fragless skb builds initial
12169b65b17dSTalal Ahmad 			 * pure zerocopy skb
12179b65b17dSTalal Ahmad 			 */
12189b65b17dSTalal Ahmad 			if (!skb->len)
12199b65b17dSTalal Ahmad 				skb_shinfo(skb)->flags |= SKBFL_PURE_ZEROCOPY;
12209b65b17dSTalal Ahmad 
12219b65b17dSTalal Ahmad 			if (!skb_zcopy_pure(skb)) {
1222849b425cSEric Dumazet 				copy = tcp_wmem_schedule(sk, copy);
1223849b425cSEric Dumazet 				if (!copy)
1224358ed624STalal Ahmad 					goto wait_for_space;
12259b65b17dSTalal Ahmad 			}
1226358ed624STalal Ahmad 
1227f214f915SWillem de Bruijn 			err = skb_zerocopy_iter_stream(sk, skb, msg, copy, uarg);
1228111856c7SWillem de Bruijn 			if (err == -EMSGSIZE || err == -EEXIST) {
1229111856c7SWillem de Bruijn 				tcp_mark_push(tp, skb);
1230f214f915SWillem de Bruijn 				goto new_segment;
1231111856c7SWillem de Bruijn 			}
1232f214f915SWillem de Bruijn 			if (err < 0)
1233f214f915SWillem de Bruijn 				goto do_error;
1234f214f915SWillem de Bruijn 			copy = err;
1235270a1c3dSDavid Howells 		} else if (zc == MSG_SPLICE_PAGES) {
1236270a1c3dSDavid Howells 			/* Splice in data if we can; copy if we can't. */
1237270a1c3dSDavid Howells 			if (tcp_downgrade_zcopy_pure(sk, skb))
1238270a1c3dSDavid Howells 				goto wait_for_space;
1239270a1c3dSDavid Howells 			copy = tcp_wmem_schedule(sk, copy);
1240270a1c3dSDavid Howells 			if (!copy)
1241270a1c3dSDavid Howells 				goto wait_for_space;
1242270a1c3dSDavid Howells 
1243270a1c3dSDavid Howells 			err = skb_splice_from_iter(skb, &msg->msg_iter, copy,
1244270a1c3dSDavid Howells 						   sk->sk_allocation);
1245270a1c3dSDavid Howells 			if (err < 0) {
1246270a1c3dSDavid Howells 				if (err == -EMSGSIZE) {
1247270a1c3dSDavid Howells 					tcp_mark_push(tp, skb);
1248270a1c3dSDavid Howells 					goto new_segment;
1249270a1c3dSDavid Howells 				}
1250270a1c3dSDavid Howells 				goto do_error;
1251270a1c3dSDavid Howells 			}
1252270a1c3dSDavid Howells 			copy = err;
1253270a1c3dSDavid Howells 
1254270a1c3dSDavid Howells 			if (!(flags & MSG_NO_SHARED_FRAGS))
1255270a1c3dSDavid Howells 				skb_shinfo(skb)->flags |= SKBFL_SHARED_FRAG;
1256270a1c3dSDavid Howells 
1257270a1c3dSDavid Howells 			sk_wmem_queued_add(sk, copy);
1258270a1c3dSDavid Howells 			sk_mem_charge(sk, copy);
12591da177e4SLinus Torvalds 		}
12601da177e4SLinus Torvalds 
12611da177e4SLinus Torvalds 		if (!copied)
12624de075e0SEric Dumazet 			TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH;
12631da177e4SLinus Torvalds 
12640f317464SEric Dumazet 		WRITE_ONCE(tp->write_seq, tp->write_seq + copy);
12651da177e4SLinus Torvalds 		TCP_SKB_CB(skb)->end_seq += copy;
1266cd7d8498SEric Dumazet 		tcp_skb_pcount_set(skb, 0);
12671da177e4SLinus Torvalds 
12681da177e4SLinus Torvalds 		copied += copy;
126901e97e65SAl Viro 		if (!msg_data_left(msg)) {
1270c134ecb8SMartin KaFai Lau 			if (unlikely(flags & MSG_EOR))
1271c134ecb8SMartin KaFai Lau 				TCP_SKB_CB(skb)->eor = 1;
12721da177e4SLinus Torvalds 			goto out;
12734ed2d765SWillem de Bruijn 		}
12741da177e4SLinus Torvalds 
127565ec6097SEric Dumazet 		if (skb->len < size_goal || (flags & MSG_OOB) || unlikely(tp->repair))
12761da177e4SLinus Torvalds 			continue;
12771da177e4SLinus Torvalds 
12781da177e4SLinus Torvalds 		if (forced_push(tp)) {
12791da177e4SLinus Torvalds 			tcp_mark_push(tp, skb);
12809e412ba7SIlpo Järvinen 			__tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);
1281fe067e8aSDavid S. Miller 		} else if (skb == tcp_send_head(sk))
12821da177e4SLinus Torvalds 			tcp_push_one(sk, mss_now);
12831da177e4SLinus Torvalds 		continue;
12841da177e4SLinus Torvalds 
1285afb83012SSoheil Hassas Yeganeh wait_for_space:
12861da177e4SLinus Torvalds 		set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1287ec342325SAndrew Vagin 		if (copied)
1288f54b3111SEric Dumazet 			tcp_push(sk, flags & ~MSG_MORE, mss_now,
1289f54b3111SEric Dumazet 				 TCP_NAGLE_PUSH, size_goal);
12901da177e4SLinus Torvalds 
1291686a5624SYuvaraja Mariappan 		err = sk_stream_wait_memory(sk, &timeo);
1292686a5624SYuvaraja Mariappan 		if (err != 0)
12931da177e4SLinus Torvalds 			goto do_error;
12941da177e4SLinus Torvalds 
12950c54b85fSIlpo Järvinen 		mss_now = tcp_send_mss(sk, &size_goal, flags);
12961da177e4SLinus Torvalds 	}
12971da177e4SLinus Torvalds 
12981da177e4SLinus Torvalds out:
1299ad02c4f5SSoheil Hassas Yeganeh 	if (copied) {
13004e8cc228SEric Dumazet 		tcp_tx_timestamp(sk, sockc.tsflags);
1301f54b3111SEric Dumazet 		tcp_push(sk, flags, mss_now, tp->nonagle, size_goal);
1302ad02c4f5SSoheil Hassas Yeganeh 	}
13035924f17aSChristoph Paasch out_nopush:
1304a7533584SPavel Begunkov 	/* msg->msg_ubuf is pinned by the caller so we don't take extra refs */
1305a7533584SPavel Begunkov 	if (uarg && !msg->msg_ubuf)
13068e044917SJonathan Lemon 		net_zcopy_put(uarg);
1307cf60af03SYuchung Cheng 	return copied + copied_syn;
13081da177e4SLinus Torvalds 
13091da177e4SLinus Torvalds do_error:
131027728ba8SEric Dumazet 	tcp_remove_empty_skb(sk);
1311fdfc5c85SEric Dumazet 
1312cf60af03SYuchung Cheng 	if (copied + copied_syn)
13131da177e4SLinus Torvalds 		goto out;
13141da177e4SLinus Torvalds out_err:
1315a7533584SPavel Begunkov 	/* msg->msg_ubuf is pinned by the caller so we don't take extra refs */
1316a7533584SPavel Begunkov 	if (uarg && !msg->msg_ubuf)
13178e044917SJonathan Lemon 		net_zcopy_put_abort(uarg, true);
13181da177e4SLinus Torvalds 	err = sk_stream_error(sk, flags, err);
1319ce5ec440SJason Baron 	/* make sure we wake any epoll edge trigger waiter */
1320216808c6SEric Dumazet 	if (unlikely(tcp_rtx_and_write_queues_empty(sk) && err == -EAGAIN)) {
1321ce5ec440SJason Baron 		sk->sk_write_space(sk);
1322b0f71bd3SFrancis Yan 		tcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED);
1323b0f71bd3SFrancis Yan 	}
13241da177e4SLinus Torvalds 	return err;
13251da177e4SLinus Torvalds }
1326774c4673SDavid S. Miller EXPORT_SYMBOL_GPL(tcp_sendmsg_locked);
1327306b13ebSTom Herbert 
1328306b13ebSTom Herbert int tcp_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
1329306b13ebSTom Herbert {
1330306b13ebSTom Herbert 	int ret;
1331306b13ebSTom Herbert 
1332306b13ebSTom Herbert 	lock_sock(sk);
1333306b13ebSTom Herbert 	ret = tcp_sendmsg_locked(sk, msg, size);
1334306b13ebSTom Herbert 	release_sock(sk);
1335306b13ebSTom Herbert 
1336306b13ebSTom Herbert 	return ret;
1337306b13ebSTom Herbert }
13384bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_sendmsg);
13391da177e4SLinus Torvalds 
13401d7e4538SDavid Howells void tcp_splice_eof(struct socket *sock)
13411d7e4538SDavid Howells {
13421d7e4538SDavid Howells 	struct sock *sk = sock->sk;
13431d7e4538SDavid Howells 	struct tcp_sock *tp = tcp_sk(sk);
13441d7e4538SDavid Howells 	int mss_now, size_goal;
13451d7e4538SDavid Howells 
13461d7e4538SDavid Howells 	if (!tcp_write_queue_tail(sk))
13471d7e4538SDavid Howells 		return;
13481d7e4538SDavid Howells 
13491d7e4538SDavid Howells 	lock_sock(sk);
13501d7e4538SDavid Howells 	mss_now = tcp_send_mss(sk, &size_goal, 0);
13511d7e4538SDavid Howells 	tcp_push(sk, 0, mss_now, tp->nonagle, size_goal);
13521d7e4538SDavid Howells 	release_sock(sk);
13531d7e4538SDavid Howells }
13541d7e4538SDavid Howells EXPORT_SYMBOL_GPL(tcp_splice_eof);
13551d7e4538SDavid Howells 
13561da177e4SLinus Torvalds /*
13571da177e4SLinus Torvalds  *	Handle reading urgent data. BSD has very simple semantics for
13581da177e4SLinus Torvalds  *	this, no blocking and very strange errors 8)
13591da177e4SLinus Torvalds  */
13601da177e4SLinus Torvalds 
1361377f0a08SRami Rosen static int tcp_recv_urg(struct sock *sk, struct msghdr *msg, int len, int flags)
13621da177e4SLinus Torvalds {
13631da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
13641da177e4SLinus Torvalds 
13651da177e4SLinus Torvalds 	/* No URG data to read. */
13661da177e4SLinus Torvalds 	if (sock_flag(sk, SOCK_URGINLINE) || !tp->urg_data ||
13671da177e4SLinus Torvalds 	    tp->urg_data == TCP_URG_READ)
13681da177e4SLinus Torvalds 		return -EINVAL;	/* Yes this is right ! */
13691da177e4SLinus Torvalds 
13701da177e4SLinus Torvalds 	if (sk->sk_state == TCP_CLOSE && !sock_flag(sk, SOCK_DONE))
13711da177e4SLinus Torvalds 		return -ENOTCONN;
13721da177e4SLinus Torvalds 
13731da177e4SLinus Torvalds 	if (tp->urg_data & TCP_URG_VALID) {
13741da177e4SLinus Torvalds 		int err = 0;
13751da177e4SLinus Torvalds 		char c = tp->urg_data;
13761da177e4SLinus Torvalds 
13771da177e4SLinus Torvalds 		if (!(flags & MSG_PEEK))
13787b6a893aSEric Dumazet 			WRITE_ONCE(tp->urg_data, TCP_URG_READ);
13791da177e4SLinus Torvalds 
13801da177e4SLinus Torvalds 		/* Read urgent data. */
13811da177e4SLinus Torvalds 		msg->msg_flags |= MSG_OOB;
13821da177e4SLinus Torvalds 
13831da177e4SLinus Torvalds 		if (len > 0) {
13841da177e4SLinus Torvalds 			if (!(flags & MSG_TRUNC))
13857eab8d9eSAl Viro 				err = memcpy_to_msg(msg, &c, 1);
13861da177e4SLinus Torvalds 			len = 1;
13871da177e4SLinus Torvalds 		} else
13881da177e4SLinus Torvalds 			msg->msg_flags |= MSG_TRUNC;
13891da177e4SLinus Torvalds 
13901da177e4SLinus Torvalds 		return err ? -EFAULT : len;
13911da177e4SLinus Torvalds 	}
13921da177e4SLinus Torvalds 
13931da177e4SLinus Torvalds 	if (sk->sk_state == TCP_CLOSE || (sk->sk_shutdown & RCV_SHUTDOWN))
13941da177e4SLinus Torvalds 		return 0;
13951da177e4SLinus Torvalds 
13961da177e4SLinus Torvalds 	/* Fixed the recv(..., MSG_OOB) behaviour.  BSD docs and
13971da177e4SLinus Torvalds 	 * the available implementations agree in this case:
13981da177e4SLinus Torvalds 	 * this call should never block, independent of the
13991da177e4SLinus Torvalds 	 * blocking state of the socket.
14001da177e4SLinus Torvalds 	 * Mike <pall@rz.uni-karlsruhe.de>
14011da177e4SLinus Torvalds 	 */
14021da177e4SLinus Torvalds 	return -EAGAIN;
14031da177e4SLinus Torvalds }
14041da177e4SLinus Torvalds 
1405c0e88ff0SPavel Emelyanov static int tcp_peek_sndq(struct sock *sk, struct msghdr *msg, int len)
1406c0e88ff0SPavel Emelyanov {
1407c0e88ff0SPavel Emelyanov 	struct sk_buff *skb;
1408c0e88ff0SPavel Emelyanov 	int copied = 0, err = 0;
1409c0e88ff0SPavel Emelyanov 
1410c0e88ff0SPavel Emelyanov 	/* XXX -- need to support SO_PEEK_OFF */
1411c0e88ff0SPavel Emelyanov 
141275c119afSEric Dumazet 	skb_rbtree_walk(skb, &sk->tcp_rtx_queue) {
141375c119afSEric Dumazet 		err = skb_copy_datagram_msg(skb, 0, msg, skb->len);
141475c119afSEric Dumazet 		if (err)
141575c119afSEric Dumazet 			return err;
141675c119afSEric Dumazet 		copied += skb->len;
141775c119afSEric Dumazet 	}
141875c119afSEric Dumazet 
1419c0e88ff0SPavel Emelyanov 	skb_queue_walk(&sk->sk_write_queue, skb) {
142051f3d02bSDavid S. Miller 		err = skb_copy_datagram_msg(skb, 0, msg, skb->len);
1421c0e88ff0SPavel Emelyanov 		if (err)
1422c0e88ff0SPavel Emelyanov 			break;
1423c0e88ff0SPavel Emelyanov 
1424c0e88ff0SPavel Emelyanov 		copied += skb->len;
1425c0e88ff0SPavel Emelyanov 	}
1426c0e88ff0SPavel Emelyanov 
1427c0e88ff0SPavel Emelyanov 	return err ?: copied;
1428c0e88ff0SPavel Emelyanov }
1429c0e88ff0SPavel Emelyanov 
14301da177e4SLinus Torvalds /* Clean up the receive buffer for full frames taken by the user,
14311da177e4SLinus Torvalds  * then send an ACK if necessary.  COPIED is the number of bytes
14321da177e4SLinus Torvalds  * tcp_recvmsg has given to the user so far, it speeds up the
14331da177e4SLinus Torvalds  * calculation of whether or not we must ACK for the sake of
14341da177e4SLinus Torvalds  * a window update.
14351da177e4SLinus Torvalds  */
1436e5c6de5fSJohn Fastabend void __tcp_cleanup_rbuf(struct sock *sk, int copied)
14371da177e4SLinus Torvalds {
14381da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
1439a2a385d6SEric Dumazet 	bool time_to_ack = false;
14401da177e4SLinus Torvalds 
1441463c84b9SArnaldo Carvalho de Melo 	if (inet_csk_ack_scheduled(sk)) {
1442463c84b9SArnaldo Carvalho de Melo 		const struct inet_connection_sock *icsk = inet_csk(sk);
1443b6b6d653SEric Dumazet 
1444b6b6d653SEric Dumazet 		if (/* Once-per-two-segments ACK was not sent by tcp_input.c */
1445463c84b9SArnaldo Carvalho de Melo 		    tp->rcv_nxt - tp->rcv_wup > icsk->icsk_ack.rcv_mss ||
14461da177e4SLinus Torvalds 		    /*
14471da177e4SLinus Torvalds 		     * If this read emptied read buffer, we send ACK, if
14481da177e4SLinus Torvalds 		     * connection is not bidirectional, user drained
14491da177e4SLinus Torvalds 		     * receive buffer and there was a small segment
14501da177e4SLinus Torvalds 		     * in queue.
14511da177e4SLinus Torvalds 		     */
14521ef9696cSAlexey Kuznetsov 		    (copied > 0 &&
14531ef9696cSAlexey Kuznetsov 		     ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED2) ||
14541ef9696cSAlexey Kuznetsov 		      ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED) &&
145531954cd8SWei Wang 		       !inet_csk_in_pingpong_mode(sk))) &&
14561ef9696cSAlexey Kuznetsov 		      !atomic_read(&sk->sk_rmem_alloc)))
1457a2a385d6SEric Dumazet 			time_to_ack = true;
14581da177e4SLinus Torvalds 	}
14591da177e4SLinus Torvalds 
14601da177e4SLinus Torvalds 	/* We send an ACK if we can now advertise a non-zero window
14611da177e4SLinus Torvalds 	 * which has been raised "significantly".
14621da177e4SLinus Torvalds 	 *
14631da177e4SLinus Torvalds 	 * Even if window raised up to infinity, do not send window open ACK
14641da177e4SLinus Torvalds 	 * in states, where we will not receive more. It is useless.
14651da177e4SLinus Torvalds 	 */
14661da177e4SLinus Torvalds 	if (copied > 0 && !time_to_ack && !(sk->sk_shutdown & RCV_SHUTDOWN)) {
14671da177e4SLinus Torvalds 		__u32 rcv_window_now = tcp_receive_window(tp);
14681da177e4SLinus Torvalds 
14691da177e4SLinus Torvalds 		/* Optimize, __tcp_select_window() is not cheap. */
14701da177e4SLinus Torvalds 		if (2*rcv_window_now <= tp->window_clamp) {
14711da177e4SLinus Torvalds 			__u32 new_window = __tcp_select_window(sk);
14721da177e4SLinus Torvalds 
14731da177e4SLinus Torvalds 			/* Send ACK now, if this read freed lots of space
14741da177e4SLinus Torvalds 			 * in our buffer. Certainly, new_window is new window.
14751da177e4SLinus Torvalds 			 * We can advertise it now, if it is not less than current one.
14761da177e4SLinus Torvalds 			 * "Lots" means "at least twice" here.
14771da177e4SLinus Torvalds 			 */
14781da177e4SLinus Torvalds 			if (new_window && new_window >= 2 * rcv_window_now)
1479a2a385d6SEric Dumazet 				time_to_ack = true;
14801da177e4SLinus Torvalds 		}
14811da177e4SLinus Torvalds 	}
14821da177e4SLinus Torvalds 	if (time_to_ack)
14831da177e4SLinus Torvalds 		tcp_send_ack(sk);
14841da177e4SLinus Torvalds }
14851da177e4SLinus Torvalds 
1486c457985aSCong Wang void tcp_cleanup_rbuf(struct sock *sk, int copied)
1487c457985aSCong Wang {
1488c457985aSCong Wang 	struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
1489c457985aSCong Wang 	struct tcp_sock *tp = tcp_sk(sk);
1490c457985aSCong Wang 
1491c457985aSCong Wang 	WARN(skb && !before(tp->copied_seq, TCP_SKB_CB(skb)->end_seq),
1492c457985aSCong Wang 	     "cleanup rbuf bug: copied %X seq %X rcvnxt %X\n",
1493c457985aSCong Wang 	     tp->copied_seq, TCP_SKB_CB(skb)->end_seq, tp->rcv_nxt);
1494c457985aSCong Wang 	__tcp_cleanup_rbuf(sk, copied);
1495c457985aSCong Wang }
1496c457985aSCong Wang 
14973df684c1SEric Dumazet static void tcp_eat_recv_skb(struct sock *sk, struct sk_buff *skb)
14983df684c1SEric Dumazet {
1499f35f8219SEric Dumazet 	__skb_unlink(skb, &sk->sk_receive_queue);
15003df684c1SEric Dumazet 	if (likely(skb->destructor == sock_rfree)) {
15013df684c1SEric Dumazet 		sock_rfree(skb);
15023df684c1SEric Dumazet 		skb->destructor = NULL;
15033df684c1SEric Dumazet 		skb->sk = NULL;
150468822bdfSEric Dumazet 		return skb_attempt_defer_free(skb);
1505f35f8219SEric Dumazet 	}
1506f35f8219SEric Dumazet 	__kfree_skb(skb);
15073df684c1SEric Dumazet }
15083df684c1SEric Dumazet 
15093f92a64eSJakub Kicinski struct sk_buff *tcp_recv_skb(struct sock *sk, u32 seq, u32 *off)
15101da177e4SLinus Torvalds {
15111da177e4SLinus Torvalds 	struct sk_buff *skb;
15121da177e4SLinus Torvalds 	u32 offset;
15131da177e4SLinus Torvalds 
1514f26845b4SEric Dumazet 	while ((skb = skb_peek(&sk->sk_receive_queue)) != NULL) {
15151da177e4SLinus Torvalds 		offset = seq - TCP_SKB_CB(skb)->seq;
15169d691539SEric Dumazet 		if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) {
15179d691539SEric Dumazet 			pr_err_once("%s: found a SYN, please report !\n", __func__);
15181da177e4SLinus Torvalds 			offset--;
15199d691539SEric Dumazet 		}
1520e11ecddfSEric Dumazet 		if (offset < skb->len || (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)) {
15211da177e4SLinus Torvalds 			*off = offset;
15221da177e4SLinus Torvalds 			return skb;
15231da177e4SLinus Torvalds 		}
1524f26845b4SEric Dumazet 		/* This looks weird, but this can happen if TCP collapsing
1525f26845b4SEric Dumazet 		 * splitted a fat GRO packet, while we released socket lock
1526f26845b4SEric Dumazet 		 * in skb_splice_bits()
1527f26845b4SEric Dumazet 		 */
15283df684c1SEric Dumazet 		tcp_eat_recv_skb(sk, skb);
15291da177e4SLinus Torvalds 	}
15301da177e4SLinus Torvalds 	return NULL;
15311da177e4SLinus Torvalds }
15323f92a64eSJakub Kicinski EXPORT_SYMBOL(tcp_recv_skb);
15331da177e4SLinus Torvalds 
15341da177e4SLinus Torvalds /*
15351da177e4SLinus Torvalds  * This routine provides an alternative to tcp_recvmsg() for routines
15361da177e4SLinus Torvalds  * that would like to handle copying from skbuffs directly in 'sendfile'
15371da177e4SLinus Torvalds  * fashion.
15381da177e4SLinus Torvalds  * Note:
15391da177e4SLinus Torvalds  *	- It is assumed that the socket was locked by the caller.
15401da177e4SLinus Torvalds  *	- The routine does not block.
15411da177e4SLinus Torvalds  *	- At present, there is no support for reading OOB data
15421da177e4SLinus Torvalds  *	  or for 'peeking' the socket using this routine
15431da177e4SLinus Torvalds  *	  (although both would be easy to implement).
15441da177e4SLinus Torvalds  */
15451da177e4SLinus Torvalds int tcp_read_sock(struct sock *sk, read_descriptor_t *desc,
15461da177e4SLinus Torvalds 		  sk_read_actor_t recv_actor)
15471da177e4SLinus Torvalds {
15481da177e4SLinus Torvalds 	struct sk_buff *skb;
15491da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
15501da177e4SLinus Torvalds 	u32 seq = tp->copied_seq;
15511da177e4SLinus Torvalds 	u32 offset;
15521da177e4SLinus Torvalds 	int copied = 0;
15531da177e4SLinus Torvalds 
15541da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN)
15551da177e4SLinus Torvalds 		return -ENOTCONN;
15561da177e4SLinus Torvalds 	while ((skb = tcp_recv_skb(sk, seq, &offset)) != NULL) {
15571da177e4SLinus Torvalds 		if (offset < skb->len) {
1558374e7b59SOctavian Purdila 			int used;
1559374e7b59SOctavian Purdila 			size_t len;
15601da177e4SLinus Torvalds 
15611da177e4SLinus Torvalds 			len = skb->len - offset;
15621da177e4SLinus Torvalds 			/* Stop reading if we hit a patch of urgent data */
1563b96c51bdSEric Dumazet 			if (unlikely(tp->urg_data)) {
15641da177e4SLinus Torvalds 				u32 urg_offset = tp->urg_seq - seq;
15651da177e4SLinus Torvalds 				if (urg_offset < len)
15661da177e4SLinus Torvalds 					len = urg_offset;
15671da177e4SLinus Torvalds 				if (!len)
15681da177e4SLinus Torvalds 					break;
15691da177e4SLinus Torvalds 			}
15701da177e4SLinus Torvalds 			used = recv_actor(desc, skb, offset, len);
1571ff905b1eSEric Dumazet 			if (used <= 0) {
1572ddb61a57SJens Axboe 				if (!copied)
1573ddb61a57SJens Axboe 					copied = used;
1574ddb61a57SJens Axboe 				break;
1575e3d5ea2cSEric Dumazet 			}
1576e3d5ea2cSEric Dumazet 			if (WARN_ON_ONCE(used > len))
1577e3d5ea2cSEric Dumazet 				used = len;
15781da177e4SLinus Torvalds 			seq += used;
15791da177e4SLinus Torvalds 			copied += used;
15801da177e4SLinus Torvalds 			offset += used;
1581e3d5ea2cSEric Dumazet 
158202275a2eSWilly Tarreau 			/* If recv_actor drops the lock (e.g. TCP splice
1583293ad604SOctavian Purdila 			 * receive) the skb pointer might be invalid when
1584293ad604SOctavian Purdila 			 * getting here: tcp_collapse might have deleted it
1585293ad604SOctavian Purdila 			 * while aggregating skbs from the socket queue.
1586293ad604SOctavian Purdila 			 */
1587293ad604SOctavian Purdila 			skb = tcp_recv_skb(sk, seq - 1, &offset);
158802275a2eSWilly Tarreau 			if (!skb)
15891da177e4SLinus Torvalds 				break;
159002275a2eSWilly Tarreau 			/* TCP coalescing might have appended data to the skb.
159102275a2eSWilly Tarreau 			 * Try to splice more frags
159202275a2eSWilly Tarreau 			 */
159302275a2eSWilly Tarreau 			if (offset + 1 != skb->len)
159402275a2eSWilly Tarreau 				continue;
15951da177e4SLinus Torvalds 		}
1596e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) {
15973df684c1SEric Dumazet 			tcp_eat_recv_skb(sk, skb);
15981da177e4SLinus Torvalds 			++seq;
15991da177e4SLinus Torvalds 			break;
16001da177e4SLinus Torvalds 		}
16013df684c1SEric Dumazet 		tcp_eat_recv_skb(sk, skb);
16021da177e4SLinus Torvalds 		if (!desc->count)
16031da177e4SLinus Torvalds 			break;
16047db48e98SEric Dumazet 		WRITE_ONCE(tp->copied_seq, seq);
16051da177e4SLinus Torvalds 	}
16067db48e98SEric Dumazet 	WRITE_ONCE(tp->copied_seq, seq);
16071da177e4SLinus Torvalds 
16081da177e4SLinus Torvalds 	tcp_rcv_space_adjust(sk);
16091da177e4SLinus Torvalds 
16101da177e4SLinus Torvalds 	/* Clean up data we have read: This will do ACK frames. */
1611f26845b4SEric Dumazet 	if (copied > 0) {
1612f26845b4SEric Dumazet 		tcp_recv_skb(sk, seq, &offset);
16130e4b4992SChris Leech 		tcp_cleanup_rbuf(sk, copied);
1614f26845b4SEric Dumazet 	}
16151da177e4SLinus Torvalds 	return copied;
16161da177e4SLinus Torvalds }
16174bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_read_sock);
16181da177e4SLinus Torvalds 
1619965b57b4SCong Wang int tcp_read_skb(struct sock *sk, skb_read_actor_t recv_actor)
162004919bedSCong Wang {
162104919bedSCong Wang 	struct tcp_sock *tp = tcp_sk(sk);
162204919bedSCong Wang 	u32 seq = tp->copied_seq;
162304919bedSCong Wang 	struct sk_buff *skb;
162404919bedSCong Wang 	int copied = 0;
162504919bedSCong Wang 	u32 offset;
162604919bedSCong Wang 
162704919bedSCong Wang 	if (sk->sk_state == TCP_LISTEN)
162804919bedSCong Wang 		return -ENOTCONN;
162904919bedSCong Wang 
1630db4192a7SCong Wang 	while ((skb = tcp_recv_skb(sk, seq, &offset)) != NULL) {
1631db4192a7SCong Wang 		u8 tcp_flags;
1632db4192a7SCong Wang 		int used;
163304919bedSCong Wang 
163404919bedSCong Wang 		__skb_unlink(skb, &sk->sk_receive_queue);
163596628951SPeilin Ye 		WARN_ON_ONCE(!skb_set_owner_sk_safe(skb, sk));
1636db4192a7SCong Wang 		tcp_flags = TCP_SKB_CB(skb)->tcp_flags;
1637db4192a7SCong Wang 		used = recv_actor(sk, skb);
1638db4192a7SCong Wang 		if (used < 0) {
1639db4192a7SCong Wang 			if (!copied)
1640db4192a7SCong Wang 				copied = used;
1641db4192a7SCong Wang 			break;
1642db4192a7SCong Wang 		}
1643db4192a7SCong Wang 		seq += used;
1644db4192a7SCong Wang 		copied += used;
1645db4192a7SCong Wang 
1646db4192a7SCong Wang 		if (tcp_flags & TCPHDR_FIN) {
1647db4192a7SCong Wang 			++seq;
1648db4192a7SCong Wang 			break;
1649db4192a7SCong Wang 		}
1650db4192a7SCong Wang 	}
165104919bedSCong Wang 	return copied;
165204919bedSCong Wang }
165304919bedSCong Wang EXPORT_SYMBOL(tcp_read_skb);
165404919bedSCong Wang 
16553f92a64eSJakub Kicinski void tcp_read_done(struct sock *sk, size_t len)
16563f92a64eSJakub Kicinski {
16573f92a64eSJakub Kicinski 	struct tcp_sock *tp = tcp_sk(sk);
16583f92a64eSJakub Kicinski 	u32 seq = tp->copied_seq;
16593f92a64eSJakub Kicinski 	struct sk_buff *skb;
16603f92a64eSJakub Kicinski 	size_t left;
16613f92a64eSJakub Kicinski 	u32 offset;
16623f92a64eSJakub Kicinski 
16633f92a64eSJakub Kicinski 	if (sk->sk_state == TCP_LISTEN)
16643f92a64eSJakub Kicinski 		return;
16653f92a64eSJakub Kicinski 
16663f92a64eSJakub Kicinski 	left = len;
16673f92a64eSJakub Kicinski 	while (left && (skb = tcp_recv_skb(sk, seq, &offset)) != NULL) {
16683f92a64eSJakub Kicinski 		int used;
16693f92a64eSJakub Kicinski 
16703f92a64eSJakub Kicinski 		used = min_t(size_t, skb->len - offset, left);
16713f92a64eSJakub Kicinski 		seq += used;
16723f92a64eSJakub Kicinski 		left -= used;
16733f92a64eSJakub Kicinski 
16743f92a64eSJakub Kicinski 		if (skb->len > offset + used)
16753f92a64eSJakub Kicinski 			break;
16763f92a64eSJakub Kicinski 
16773f92a64eSJakub Kicinski 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) {
16783f92a64eSJakub Kicinski 			tcp_eat_recv_skb(sk, skb);
16793f92a64eSJakub Kicinski 			++seq;
16803f92a64eSJakub Kicinski 			break;
16813f92a64eSJakub Kicinski 		}
16823f92a64eSJakub Kicinski 		tcp_eat_recv_skb(sk, skb);
16833f92a64eSJakub Kicinski 	}
16843f92a64eSJakub Kicinski 	WRITE_ONCE(tp->copied_seq, seq);
16853f92a64eSJakub Kicinski 
16863f92a64eSJakub Kicinski 	tcp_rcv_space_adjust(sk);
16873f92a64eSJakub Kicinski 
16883f92a64eSJakub Kicinski 	/* Clean up data we have read: This will do ACK frames. */
16893f92a64eSJakub Kicinski 	if (left != len)
16903f92a64eSJakub Kicinski 		tcp_cleanup_rbuf(sk, len - left);
16913f92a64eSJakub Kicinski }
16923f92a64eSJakub Kicinski EXPORT_SYMBOL(tcp_read_done);
16933f92a64eSJakub Kicinski 
169432035585STom Herbert int tcp_peek_len(struct socket *sock)
169532035585STom Herbert {
169632035585STom Herbert 	return tcp_inq(sock->sk);
169732035585STom Herbert }
169832035585STom Herbert EXPORT_SYMBOL(tcp_peek_len);
169932035585STom Herbert 
1700d1361840SEric Dumazet /* Make sure sk_rcvbuf is big enough to satisfy SO_RCVLOWAT hint */
1701d1361840SEric Dumazet int tcp_set_rcvlowat(struct sock *sk, int val)
1702d1361840SEric Dumazet {
1703867f816bSSoheil Hassas Yeganeh 	int cap;
1704867f816bSSoheil Hassas Yeganeh 
1705867f816bSSoheil Hassas Yeganeh 	if (sk->sk_userlocks & SOCK_RCVBUF_LOCK)
1706867f816bSSoheil Hassas Yeganeh 		cap = sk->sk_rcvbuf >> 1;
1707867f816bSSoheil Hassas Yeganeh 	else
170802739545SKuniyuki Iwashima 		cap = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_rmem[2]) >> 1;
1709867f816bSSoheil Hassas Yeganeh 	val = min(val, cap);
1710eac66402SEric Dumazet 	WRITE_ONCE(sk->sk_rcvlowat, val ? : 1);
171103f45c88SEric Dumazet 
171203f45c88SEric Dumazet 	/* Check if we need to signal EPOLLIN right now */
171303f45c88SEric Dumazet 	tcp_data_ready(sk);
171403f45c88SEric Dumazet 
1715d1361840SEric Dumazet 	if (sk->sk_userlocks & SOCK_RCVBUF_LOCK)
1716d1361840SEric Dumazet 		return 0;
1717d1361840SEric Dumazet 
1718d1361840SEric Dumazet 	val <<= 1;
1719d1361840SEric Dumazet 	if (val > sk->sk_rcvbuf) {
1720ebb3b78dSEric Dumazet 		WRITE_ONCE(sk->sk_rcvbuf, val);
1721d1361840SEric Dumazet 		tcp_sk(sk)->window_clamp = tcp_win_from_space(sk, val);
1722d1361840SEric Dumazet 	}
1723d1361840SEric Dumazet 	return 0;
1724d1361840SEric Dumazet }
1725d1361840SEric Dumazet EXPORT_SYMBOL(tcp_set_rcvlowat);
1726d1361840SEric Dumazet 
1727892bfd3dSFlorian Westphal void tcp_update_recv_tstamps(struct sk_buff *skb,
17287eeba170SArjun Roy 			     struct scm_timestamping_internal *tss)
17297eeba170SArjun Roy {
17307eeba170SArjun Roy 	if (skb->tstamp)
17317eeba170SArjun Roy 		tss->ts[0] = ktime_to_timespec64(skb->tstamp);
17327eeba170SArjun Roy 	else
17337eeba170SArjun Roy 		tss->ts[0] = (struct timespec64) {0};
17347eeba170SArjun Roy 
17357eeba170SArjun Roy 	if (skb_hwtstamps(skb)->hwtstamp)
17367eeba170SArjun Roy 		tss->ts[2] = ktime_to_timespec64(skb_hwtstamps(skb)->hwtstamp);
17377eeba170SArjun Roy 	else
17387eeba170SArjun Roy 		tss->ts[2] = (struct timespec64) {0};
17397eeba170SArjun Roy }
17407eeba170SArjun Roy 
174105255b82SEric Dumazet #ifdef CONFIG_MMU
17427a7f0946SArjun Roy const struct vm_operations_struct tcp_vm_ops = {
174305255b82SEric Dumazet };
174405255b82SEric Dumazet 
174593ab6cc6SEric Dumazet int tcp_mmap(struct file *file, struct socket *sock,
174693ab6cc6SEric Dumazet 	     struct vm_area_struct *vma)
174793ab6cc6SEric Dumazet {
174805255b82SEric Dumazet 	if (vma->vm_flags & (VM_WRITE | VM_EXEC))
174905255b82SEric Dumazet 		return -EPERM;
17501c71222eSSuren Baghdasaryan 	vm_flags_clear(vma, VM_MAYWRITE | VM_MAYEXEC);
175105255b82SEric Dumazet 
17523e4e28c5SMichel Lespinasse 	/* Instruct vm_insert_page() to not mmap_read_lock(mm) */
17531c71222eSSuren Baghdasaryan 	vm_flags_set(vma, VM_MIXEDMAP);
175405255b82SEric Dumazet 
175505255b82SEric Dumazet 	vma->vm_ops = &tcp_vm_ops;
175605255b82SEric Dumazet 	return 0;
175705255b82SEric Dumazet }
175805255b82SEric Dumazet EXPORT_SYMBOL(tcp_mmap);
175905255b82SEric Dumazet 
17607fba5309SArjun Roy static skb_frag_t *skb_advance_to_frag(struct sk_buff *skb, u32 offset_skb,
17617fba5309SArjun Roy 				       u32 *offset_frag)
17627fba5309SArjun Roy {
17637fba5309SArjun Roy 	skb_frag_t *frag;
17647fba5309SArjun Roy 
176570701b83SArjun Roy 	if (unlikely(offset_skb >= skb->len))
176670701b83SArjun Roy 		return NULL;
176770701b83SArjun Roy 
17687fba5309SArjun Roy 	offset_skb -= skb_headlen(skb);
17697fba5309SArjun Roy 	if ((int)offset_skb < 0 || skb_has_frag_list(skb))
17707fba5309SArjun Roy 		return NULL;
17717fba5309SArjun Roy 
17727fba5309SArjun Roy 	frag = skb_shinfo(skb)->frags;
17737fba5309SArjun Roy 	while (offset_skb) {
17747fba5309SArjun Roy 		if (skb_frag_size(frag) > offset_skb) {
17757fba5309SArjun Roy 			*offset_frag = offset_skb;
17767fba5309SArjun Roy 			return frag;
17777fba5309SArjun Roy 		}
17787fba5309SArjun Roy 		offset_skb -= skb_frag_size(frag);
17797fba5309SArjun Roy 		++frag;
17807fba5309SArjun Roy 	}
17817fba5309SArjun Roy 	*offset_frag = 0;
17827fba5309SArjun Roy 	return frag;
17837fba5309SArjun Roy }
17847fba5309SArjun Roy 
178598917cf0SArjun Roy static bool can_map_frag(const skb_frag_t *frag)
178698917cf0SArjun Roy {
178798917cf0SArjun Roy 	return skb_frag_size(frag) == PAGE_SIZE && !skb_frag_off(frag);
178898917cf0SArjun Roy }
178998917cf0SArjun Roy 
179098917cf0SArjun Roy static int find_next_mappable_frag(const skb_frag_t *frag,
179198917cf0SArjun Roy 				   int remaining_in_skb)
179298917cf0SArjun Roy {
179398917cf0SArjun Roy 	int offset = 0;
179498917cf0SArjun Roy 
179598917cf0SArjun Roy 	if (likely(can_map_frag(frag)))
179698917cf0SArjun Roy 		return 0;
179798917cf0SArjun Roy 
179898917cf0SArjun Roy 	while (offset < remaining_in_skb && !can_map_frag(frag)) {
179998917cf0SArjun Roy 		offset += skb_frag_size(frag);
180098917cf0SArjun Roy 		++frag;
180198917cf0SArjun Roy 	}
180298917cf0SArjun Roy 	return offset;
180398917cf0SArjun Roy }
180498917cf0SArjun Roy 
18050c3936d3SArjun Roy static void tcp_zerocopy_set_hint_for_skb(struct sock *sk,
18060c3936d3SArjun Roy 					  struct tcp_zerocopy_receive *zc,
18070c3936d3SArjun Roy 					  struct sk_buff *skb, u32 offset)
18080c3936d3SArjun Roy {
18090c3936d3SArjun Roy 	u32 frag_offset, partial_frag_remainder = 0;
18100c3936d3SArjun Roy 	int mappable_offset;
18110c3936d3SArjun Roy 	skb_frag_t *frag;
18120c3936d3SArjun Roy 
18130c3936d3SArjun Roy 	/* worst case: skip to next skb. try to improve on this case below */
18140c3936d3SArjun Roy 	zc->recv_skip_hint = skb->len - offset;
18150c3936d3SArjun Roy 
18160c3936d3SArjun Roy 	/* Find the frag containing this offset (and how far into that frag) */
18170c3936d3SArjun Roy 	frag = skb_advance_to_frag(skb, offset, &frag_offset);
18180c3936d3SArjun Roy 	if (!frag)
18190c3936d3SArjun Roy 		return;
18200c3936d3SArjun Roy 
18210c3936d3SArjun Roy 	if (frag_offset) {
18220c3936d3SArjun Roy 		struct skb_shared_info *info = skb_shinfo(skb);
18230c3936d3SArjun Roy 
18240c3936d3SArjun Roy 		/* We read part of the last frag, must recvmsg() rest of skb. */
18250c3936d3SArjun Roy 		if (frag == &info->frags[info->nr_frags - 1])
18260c3936d3SArjun Roy 			return;
18270c3936d3SArjun Roy 
18280c3936d3SArjun Roy 		/* Else, we must at least read the remainder in this frag. */
18290c3936d3SArjun Roy 		partial_frag_remainder = skb_frag_size(frag) - frag_offset;
18300c3936d3SArjun Roy 		zc->recv_skip_hint -= partial_frag_remainder;
18310c3936d3SArjun Roy 		++frag;
18320c3936d3SArjun Roy 	}
18330c3936d3SArjun Roy 
18340c3936d3SArjun Roy 	/* partial_frag_remainder: If part way through a frag, must read rest.
18350c3936d3SArjun Roy 	 * mappable_offset: Bytes till next mappable frag, *not* counting bytes
18360c3936d3SArjun Roy 	 * in partial_frag_remainder.
18370c3936d3SArjun Roy 	 */
18380c3936d3SArjun Roy 	mappable_offset = find_next_mappable_frag(frag, zc->recv_skip_hint);
18390c3936d3SArjun Roy 	zc->recv_skip_hint = mappable_offset + partial_frag_remainder;
18400c3936d3SArjun Roy }
18410c3936d3SArjun Roy 
1842f21a3c48SArjun Roy static int tcp_recvmsg_locked(struct sock *sk, struct msghdr *msg, size_t len,
1843ec095263SOliver Hartkopp 			      int flags, struct scm_timestamping_internal *tss,
1844f21a3c48SArjun Roy 			      int *cmsg_flags);
1845f21a3c48SArjun Roy static int receive_fallback_to_copy(struct sock *sk,
18467eeba170SArjun Roy 				    struct tcp_zerocopy_receive *zc, int inq,
18477eeba170SArjun Roy 				    struct scm_timestamping_internal *tss)
1848f21a3c48SArjun Roy {
1849f21a3c48SArjun Roy 	unsigned long copy_address = (unsigned long)zc->copybuf_address;
1850f21a3c48SArjun Roy 	struct msghdr msg = {};
1851f21a3c48SArjun Roy 	struct iovec iov;
18527eeba170SArjun Roy 	int err;
1853f21a3c48SArjun Roy 
1854f21a3c48SArjun Roy 	zc->length = 0;
1855f21a3c48SArjun Roy 	zc->recv_skip_hint = 0;
1856f21a3c48SArjun Roy 
1857f21a3c48SArjun Roy 	if (copy_address != zc->copybuf_address)
1858f21a3c48SArjun Roy 		return -EINVAL;
1859f21a3c48SArjun Roy 
1860de4eda9dSAl Viro 	err = import_single_range(ITER_DEST, (void __user *)copy_address,
1861f21a3c48SArjun Roy 				  inq, &iov, &msg.msg_iter);
1862f21a3c48SArjun Roy 	if (err)
1863f21a3c48SArjun Roy 		return err;
1864f21a3c48SArjun Roy 
1865ec095263SOliver Hartkopp 	err = tcp_recvmsg_locked(sk, &msg, inq, MSG_DONTWAIT,
18667eeba170SArjun Roy 				 tss, &zc->msg_flags);
1867f21a3c48SArjun Roy 	if (err < 0)
1868f21a3c48SArjun Roy 		return err;
1869f21a3c48SArjun Roy 
1870f21a3c48SArjun Roy 	zc->copybuf_len = err;
18710c3936d3SArjun Roy 	if (likely(zc->copybuf_len)) {
18720c3936d3SArjun Roy 		struct sk_buff *skb;
18730c3936d3SArjun Roy 		u32 offset;
18740c3936d3SArjun Roy 
18750c3936d3SArjun Roy 		skb = tcp_recv_skb(sk, tcp_sk(sk)->copied_seq, &offset);
18760c3936d3SArjun Roy 		if (skb)
18770c3936d3SArjun Roy 			tcp_zerocopy_set_hint_for_skb(sk, zc, skb, offset);
18780c3936d3SArjun Roy 	}
1879f21a3c48SArjun Roy 	return 0;
1880f21a3c48SArjun Roy }
1881f21a3c48SArjun Roy 
188218fb76edSArjun Roy static int tcp_copy_straggler_data(struct tcp_zerocopy_receive *zc,
188318fb76edSArjun Roy 				   struct sk_buff *skb, u32 copylen,
188418fb76edSArjun Roy 				   u32 *offset, u32 *seq)
188518fb76edSArjun Roy {
188618fb76edSArjun Roy 	unsigned long copy_address = (unsigned long)zc->copybuf_address;
188718fb76edSArjun Roy 	struct msghdr msg = {};
188818fb76edSArjun Roy 	struct iovec iov;
188918fb76edSArjun Roy 	int err;
189018fb76edSArjun Roy 
189118fb76edSArjun Roy 	if (copy_address != zc->copybuf_address)
189218fb76edSArjun Roy 		return -EINVAL;
189318fb76edSArjun Roy 
1894de4eda9dSAl Viro 	err = import_single_range(ITER_DEST, (void __user *)copy_address,
189518fb76edSArjun Roy 				  copylen, &iov, &msg.msg_iter);
189618fb76edSArjun Roy 	if (err)
189718fb76edSArjun Roy 		return err;
189818fb76edSArjun Roy 	err = skb_copy_datagram_msg(skb, *offset, &msg, copylen);
189918fb76edSArjun Roy 	if (err)
190018fb76edSArjun Roy 		return err;
190118fb76edSArjun Roy 	zc->recv_skip_hint -= copylen;
190218fb76edSArjun Roy 	*offset += copylen;
190318fb76edSArjun Roy 	*seq += copylen;
190418fb76edSArjun Roy 	return (__s32)copylen;
190518fb76edSArjun Roy }
190618fb76edSArjun Roy 
19077eeba170SArjun Roy static int tcp_zc_handle_leftover(struct tcp_zerocopy_receive *zc,
190818fb76edSArjun Roy 				  struct sock *sk,
190918fb76edSArjun Roy 				  struct sk_buff *skb,
191018fb76edSArjun Roy 				  u32 *seq,
19117eeba170SArjun Roy 				  s32 copybuf_len,
19127eeba170SArjun Roy 				  struct scm_timestamping_internal *tss)
191318fb76edSArjun Roy {
191418fb76edSArjun Roy 	u32 offset, copylen = min_t(u32, copybuf_len, zc->recv_skip_hint);
191518fb76edSArjun Roy 
191618fb76edSArjun Roy 	if (!copylen)
191718fb76edSArjun Roy 		return 0;
191818fb76edSArjun Roy 	/* skb is null if inq < PAGE_SIZE. */
19197eeba170SArjun Roy 	if (skb) {
192018fb76edSArjun Roy 		offset = *seq - TCP_SKB_CB(skb)->seq;
19217eeba170SArjun Roy 	} else {
192218fb76edSArjun Roy 		skb = tcp_recv_skb(sk, *seq, &offset);
19237eeba170SArjun Roy 		if (TCP_SKB_CB(skb)->has_rxtstamp) {
19247eeba170SArjun Roy 			tcp_update_recv_tstamps(skb, tss);
19257eeba170SArjun Roy 			zc->msg_flags |= TCP_CMSG_TS;
19267eeba170SArjun Roy 		}
19277eeba170SArjun Roy 	}
192818fb76edSArjun Roy 
192918fb76edSArjun Roy 	zc->copybuf_len = tcp_copy_straggler_data(zc, skb, copylen, &offset,
193018fb76edSArjun Roy 						  seq);
193118fb76edSArjun Roy 	return zc->copybuf_len < 0 ? 0 : copylen;
193218fb76edSArjun Roy }
193318fb76edSArjun Roy 
193494ab9eb9SArjun Roy static int tcp_zerocopy_vm_insert_batch_error(struct vm_area_struct *vma,
193594ab9eb9SArjun Roy 					      struct page **pending_pages,
193694ab9eb9SArjun Roy 					      unsigned long pages_remaining,
193794ab9eb9SArjun Roy 					      unsigned long *address,
193894ab9eb9SArjun Roy 					      u32 *length,
193994ab9eb9SArjun Roy 					      u32 *seq,
194094ab9eb9SArjun Roy 					      struct tcp_zerocopy_receive *zc,
194194ab9eb9SArjun Roy 					      u32 total_bytes_to_map,
194294ab9eb9SArjun Roy 					      int err)
194394ab9eb9SArjun Roy {
194494ab9eb9SArjun Roy 	/* At least one page did not map. Try zapping if we skipped earlier. */
194594ab9eb9SArjun Roy 	if (err == -EBUSY &&
194694ab9eb9SArjun Roy 	    zc->flags & TCP_RECEIVE_ZEROCOPY_FLAG_TLB_CLEAN_HINT) {
194794ab9eb9SArjun Roy 		u32 maybe_zap_len;
194894ab9eb9SArjun Roy 
194994ab9eb9SArjun Roy 		maybe_zap_len = total_bytes_to_map -  /* All bytes to map */
195094ab9eb9SArjun Roy 				*length + /* Mapped or pending */
195194ab9eb9SArjun Roy 				(pages_remaining * PAGE_SIZE); /* Failed map. */
1952e9adcfecSMike Kravetz 		zap_page_range_single(vma, *address, maybe_zap_len, NULL);
195394ab9eb9SArjun Roy 		err = 0;
195494ab9eb9SArjun Roy 	}
195594ab9eb9SArjun Roy 
195694ab9eb9SArjun Roy 	if (!err) {
195794ab9eb9SArjun Roy 		unsigned long leftover_pages = pages_remaining;
195894ab9eb9SArjun Roy 		int bytes_mapped;
195994ab9eb9SArjun Roy 
1960e9adcfecSMike Kravetz 		/* We called zap_page_range_single, try to reinsert. */
196194ab9eb9SArjun Roy 		err = vm_insert_pages(vma, *address,
196294ab9eb9SArjun Roy 				      pending_pages,
196394ab9eb9SArjun Roy 				      &pages_remaining);
196494ab9eb9SArjun Roy 		bytes_mapped = PAGE_SIZE * (leftover_pages - pages_remaining);
196594ab9eb9SArjun Roy 		*seq += bytes_mapped;
196694ab9eb9SArjun Roy 		*address += bytes_mapped;
196794ab9eb9SArjun Roy 	}
196894ab9eb9SArjun Roy 	if (err) {
196994ab9eb9SArjun Roy 		/* Either we were unable to zap, OR we zapped, retried an
197094ab9eb9SArjun Roy 		 * insert, and still had an issue. Either ways, pages_remaining
197194ab9eb9SArjun Roy 		 * is the number of pages we were unable to map, and we unroll
197294ab9eb9SArjun Roy 		 * some state we speculatively touched before.
197394ab9eb9SArjun Roy 		 */
197494ab9eb9SArjun Roy 		const int bytes_not_mapped = PAGE_SIZE * pages_remaining;
197594ab9eb9SArjun Roy 
197694ab9eb9SArjun Roy 		*length -= bytes_not_mapped;
197794ab9eb9SArjun Roy 		zc->recv_skip_hint += bytes_not_mapped;
197894ab9eb9SArjun Roy 	}
197994ab9eb9SArjun Roy 	return err;
198094ab9eb9SArjun Roy }
198194ab9eb9SArjun Roy 
19823763a24cSArjun Roy static int tcp_zerocopy_vm_insert_batch(struct vm_area_struct *vma,
19833763a24cSArjun Roy 					struct page **pages,
198494ab9eb9SArjun Roy 					unsigned int pages_to_map,
198594ab9eb9SArjun Roy 					unsigned long *address,
198694ab9eb9SArjun Roy 					u32 *length,
19873763a24cSArjun Roy 					u32 *seq,
198894ab9eb9SArjun Roy 					struct tcp_zerocopy_receive *zc,
198994ab9eb9SArjun Roy 					u32 total_bytes_to_map)
19903763a24cSArjun Roy {
19913763a24cSArjun Roy 	unsigned long pages_remaining = pages_to_map;
199294ab9eb9SArjun Roy 	unsigned int pages_mapped;
199394ab9eb9SArjun Roy 	unsigned int bytes_mapped;
199494ab9eb9SArjun Roy 	int err;
19953763a24cSArjun Roy 
199694ab9eb9SArjun Roy 	err = vm_insert_pages(vma, *address, pages, &pages_remaining);
199794ab9eb9SArjun Roy 	pages_mapped = pages_to_map - (unsigned int)pages_remaining;
199894ab9eb9SArjun Roy 	bytes_mapped = PAGE_SIZE * pages_mapped;
19993763a24cSArjun Roy 	/* Even if vm_insert_pages fails, it may have partially succeeded in
20003763a24cSArjun Roy 	 * mapping (some but not all of the pages).
20013763a24cSArjun Roy 	 */
20023763a24cSArjun Roy 	*seq += bytes_mapped;
200394ab9eb9SArjun Roy 	*address += bytes_mapped;
200494ab9eb9SArjun Roy 
200594ab9eb9SArjun Roy 	if (likely(!err))
200694ab9eb9SArjun Roy 		return 0;
200794ab9eb9SArjun Roy 
200894ab9eb9SArjun Roy 	/* Error: maybe zap and retry + rollback state for failed inserts. */
200994ab9eb9SArjun Roy 	return tcp_zerocopy_vm_insert_batch_error(vma, pages + pages_mapped,
201094ab9eb9SArjun Roy 		pages_remaining, address, length, seq, zc, total_bytes_to_map,
201194ab9eb9SArjun Roy 		err);
20123763a24cSArjun Roy }
20133763a24cSArjun Roy 
20143c5a2fd0SArjun Roy #define TCP_VALID_ZC_MSG_FLAGS   (TCP_CMSG_TS)
20157eeba170SArjun Roy static void tcp_zc_finalize_rx_tstamp(struct sock *sk,
20167eeba170SArjun Roy 				      struct tcp_zerocopy_receive *zc,
20177eeba170SArjun Roy 				      struct scm_timestamping_internal *tss)
20187eeba170SArjun Roy {
20197eeba170SArjun Roy 	unsigned long msg_control_addr;
20207eeba170SArjun Roy 	struct msghdr cmsg_dummy;
20217eeba170SArjun Roy 
20227eeba170SArjun Roy 	msg_control_addr = (unsigned long)zc->msg_control;
2023c39ef213SKevin Brodsky 	cmsg_dummy.msg_control_user = (void __user *)msg_control_addr;
20247eeba170SArjun Roy 	cmsg_dummy.msg_controllen =
20257eeba170SArjun Roy 		(__kernel_size_t)zc->msg_controllen;
20267eeba170SArjun Roy 	cmsg_dummy.msg_flags = in_compat_syscall()
20277eeba170SArjun Roy 		? MSG_CMSG_COMPAT : 0;
2028a6f8ee58SArjun Roy 	cmsg_dummy.msg_control_is_user = true;
20297eeba170SArjun Roy 	zc->msg_flags = 0;
20307eeba170SArjun Roy 	if (zc->msg_control == msg_control_addr &&
20317eeba170SArjun Roy 	    zc->msg_controllen == cmsg_dummy.msg_controllen) {
20327eeba170SArjun Roy 		tcp_recv_timestamp(&cmsg_dummy, sk, tss);
20337eeba170SArjun Roy 		zc->msg_control = (__u64)
2034c39ef213SKevin Brodsky 			((uintptr_t)cmsg_dummy.msg_control_user);
20357eeba170SArjun Roy 		zc->msg_controllen =
20367eeba170SArjun Roy 			(__u64)cmsg_dummy.msg_controllen;
20377eeba170SArjun Roy 		zc->msg_flags = (__u32)cmsg_dummy.msg_flags;
20387eeba170SArjun Roy 	}
20397eeba170SArjun Roy }
20407eeba170SArjun Roy 
20417a7f0946SArjun Roy static struct vm_area_struct *find_tcp_vma(struct mm_struct *mm,
20427a7f0946SArjun Roy 					   unsigned long address,
20437a7f0946SArjun Roy 					   bool *mmap_locked)
20447a7f0946SArjun Roy {
20457a7f0946SArjun Roy 	struct vm_area_struct *vma = NULL;
20467a7f0946SArjun Roy 
20477a7f0946SArjun Roy #ifdef CONFIG_PER_VMA_LOCK
20487a7f0946SArjun Roy 	vma = lock_vma_under_rcu(mm, address);
20497a7f0946SArjun Roy #endif
20507a7f0946SArjun Roy 	if (vma) {
20517a7f0946SArjun Roy 		if (!vma_is_tcp(vma)) {
20527a7f0946SArjun Roy 			vma_end_read(vma);
20537a7f0946SArjun Roy 			return NULL;
20547a7f0946SArjun Roy 		}
20557a7f0946SArjun Roy 		*mmap_locked = false;
20567a7f0946SArjun Roy 		return vma;
20577a7f0946SArjun Roy 	}
20587a7f0946SArjun Roy 
20597a7f0946SArjun Roy 	mmap_read_lock(mm);
20607a7f0946SArjun Roy 	vma = vma_lookup(mm, address);
20617a7f0946SArjun Roy 	if (!vma || !vma_is_tcp(vma)) {
20627a7f0946SArjun Roy 		mmap_read_unlock(mm);
20637a7f0946SArjun Roy 		return NULL;
20647a7f0946SArjun Roy 	}
20657a7f0946SArjun Roy 	*mmap_locked = true;
20667a7f0946SArjun Roy 	return vma;
20677a7f0946SArjun Roy }
20687a7f0946SArjun Roy 
206994ab9eb9SArjun Roy #define TCP_ZEROCOPY_PAGE_BATCH_SIZE 32
207005255b82SEric Dumazet static int tcp_zerocopy_receive(struct sock *sk,
20717eeba170SArjun Roy 				struct tcp_zerocopy_receive *zc,
20727eeba170SArjun Roy 				struct scm_timestamping_internal *tss)
207305255b82SEric Dumazet {
207494ab9eb9SArjun Roy 	u32 length = 0, offset, vma_len, avail_len, copylen = 0;
207505255b82SEric Dumazet 	unsigned long address = (unsigned long)zc->address;
207694ab9eb9SArjun Roy 	struct page *pages[TCP_ZEROCOPY_PAGE_BATCH_SIZE];
207718fb76edSArjun Roy 	s32 copybuf_len = zc->copybuf_len;
207818fb76edSArjun Roy 	struct tcp_sock *tp = tcp_sk(sk);
207905255b82SEric Dumazet 	const skb_frag_t *frags = NULL;
208094ab9eb9SArjun Roy 	unsigned int pages_to_map = 0;
208105255b82SEric Dumazet 	struct vm_area_struct *vma;
208205255b82SEric Dumazet 	struct sk_buff *skb = NULL;
208318fb76edSArjun Roy 	u32 seq = tp->copied_seq;
208494ab9eb9SArjun Roy 	u32 total_bytes_to_map;
208518fb76edSArjun Roy 	int inq = tcp_inq(sk);
20867a7f0946SArjun Roy 	bool mmap_locked;
208793ab6cc6SEric Dumazet 	int ret;
208893ab6cc6SEric Dumazet 
208918fb76edSArjun Roy 	zc->copybuf_len = 0;
20907eeba170SArjun Roy 	zc->msg_flags = 0;
209118fb76edSArjun Roy 
209205255b82SEric Dumazet 	if (address & (PAGE_SIZE - 1) || address != zc->address)
209393ab6cc6SEric Dumazet 		return -EINVAL;
209493ab6cc6SEric Dumazet 
209593ab6cc6SEric Dumazet 	if (sk->sk_state == TCP_LISTEN)
209605255b82SEric Dumazet 		return -ENOTCONN;
209793ab6cc6SEric Dumazet 
209893ab6cc6SEric Dumazet 	sock_rps_record_flow(sk);
209993ab6cc6SEric Dumazet 
2100f21a3c48SArjun Roy 	if (inq && inq <= copybuf_len)
21017eeba170SArjun Roy 		return receive_fallback_to_copy(sk, zc, inq, tss);
2102f21a3c48SArjun Roy 
2103936ced41SArjun Roy 	if (inq < PAGE_SIZE) {
2104936ced41SArjun Roy 		zc->length = 0;
2105936ced41SArjun Roy 		zc->recv_skip_hint = inq;
2106936ced41SArjun Roy 		if (!inq && sock_flag(sk, SOCK_DONE))
2107936ced41SArjun Roy 			return -EIO;
2108936ced41SArjun Roy 		return 0;
2109936ced41SArjun Roy 	}
2110936ced41SArjun Roy 
21117a7f0946SArjun Roy 	vma = find_tcp_vma(current->mm, address, &mmap_locked);
21127a7f0946SArjun Roy 	if (!vma)
2113e776af60SEric Dumazet 		return -EINVAL;
21147a7f0946SArjun Roy 
211518fb76edSArjun Roy 	vma_len = min_t(unsigned long, zc->length, vma->vm_end - address);
211618fb76edSArjun Roy 	avail_len = min_t(u32, vma_len, inq);
211794ab9eb9SArjun Roy 	total_bytes_to_map = avail_len & ~(PAGE_SIZE - 1);
211894ab9eb9SArjun Roy 	if (total_bytes_to_map) {
211994ab9eb9SArjun Roy 		if (!(zc->flags & TCP_RECEIVE_ZEROCOPY_FLAG_TLB_CLEAN_HINT))
2120e9adcfecSMike Kravetz 			zap_page_range_single(vma, address, total_bytes_to_map,
2121e9adcfecSMike Kravetz 					      NULL);
212294ab9eb9SArjun Roy 		zc->length = total_bytes_to_map;
212305255b82SEric Dumazet 		zc->recv_skip_hint = 0;
21248f2b0293SSoheil Hassas Yeganeh 	} else {
212518fb76edSArjun Roy 		zc->length = avail_len;
212618fb76edSArjun Roy 		zc->recv_skip_hint = avail_len;
21278f2b0293SSoheil Hassas Yeganeh 	}
212805255b82SEric Dumazet 	ret = 0;
212905255b82SEric Dumazet 	while (length + PAGE_SIZE <= zc->length) {
213098917cf0SArjun Roy 		int mappable_offset;
213194ab9eb9SArjun Roy 		struct page *page;
213298917cf0SArjun Roy 
213305255b82SEric Dumazet 		if (zc->recv_skip_hint < PAGE_SIZE) {
21347fba5309SArjun Roy 			u32 offset_frag;
21357fba5309SArjun Roy 
213605255b82SEric Dumazet 			if (skb) {
21370e627190SArjun Roy 				if (zc->recv_skip_hint > 0)
21380e627190SArjun Roy 					break;
213905255b82SEric Dumazet 				skb = skb->next;
214005255b82SEric Dumazet 				offset = seq - TCP_SKB_CB(skb)->seq;
214105255b82SEric Dumazet 			} else {
214293ab6cc6SEric Dumazet 				skb = tcp_recv_skb(sk, seq, &offset);
214305255b82SEric Dumazet 			}
21447eeba170SArjun Roy 
21457eeba170SArjun Roy 			if (TCP_SKB_CB(skb)->has_rxtstamp) {
21467eeba170SArjun Roy 				tcp_update_recv_tstamps(skb, tss);
21477eeba170SArjun Roy 				zc->msg_flags |= TCP_CMSG_TS;
21487eeba170SArjun Roy 			}
214905255b82SEric Dumazet 			zc->recv_skip_hint = skb->len - offset;
21507fba5309SArjun Roy 			frags = skb_advance_to_frag(skb, offset, &offset_frag);
21517fba5309SArjun Roy 			if (!frags || offset_frag)
215205255b82SEric Dumazet 				break;
215305255b82SEric Dumazet 		}
2154789762ceSSoheil Hassas Yeganeh 
215598917cf0SArjun Roy 		mappable_offset = find_next_mappable_frag(frags,
215698917cf0SArjun Roy 							  zc->recv_skip_hint);
215798917cf0SArjun Roy 		if (mappable_offset) {
215898917cf0SArjun Roy 			zc->recv_skip_hint = mappable_offset;
215905255b82SEric Dumazet 			break;
2160789762ceSSoheil Hassas Yeganeh 		}
216194ab9eb9SArjun Roy 		page = skb_frag_page(frags);
216294ab9eb9SArjun Roy 		prefetchw(page);
216394ab9eb9SArjun Roy 		pages[pages_to_map++] = page;
216405255b82SEric Dumazet 		length += PAGE_SIZE;
216505255b82SEric Dumazet 		zc->recv_skip_hint -= PAGE_SIZE;
216605255b82SEric Dumazet 		frags++;
216794ab9eb9SArjun Roy 		if (pages_to_map == TCP_ZEROCOPY_PAGE_BATCH_SIZE ||
216894ab9eb9SArjun Roy 		    zc->recv_skip_hint < PAGE_SIZE) {
216994ab9eb9SArjun Roy 			/* Either full batch, or we're about to go to next skb
217094ab9eb9SArjun Roy 			 * (and we cannot unroll failed ops across skbs).
217194ab9eb9SArjun Roy 			 */
217294ab9eb9SArjun Roy 			ret = tcp_zerocopy_vm_insert_batch(vma, pages,
217394ab9eb9SArjun Roy 							   pages_to_map,
217494ab9eb9SArjun Roy 							   &address, &length,
217594ab9eb9SArjun Roy 							   &seq, zc,
217694ab9eb9SArjun Roy 							   total_bytes_to_map);
21773763a24cSArjun Roy 			if (ret)
21783763a24cSArjun Roy 				goto out;
217994ab9eb9SArjun Roy 			pages_to_map = 0;
21803763a24cSArjun Roy 		}
21813763a24cSArjun Roy 	}
218294ab9eb9SArjun Roy 	if (pages_to_map) {
218394ab9eb9SArjun Roy 		ret = tcp_zerocopy_vm_insert_batch(vma, pages, pages_to_map,
218494ab9eb9SArjun Roy 						   &address, &length, &seq,
218594ab9eb9SArjun Roy 						   zc, total_bytes_to_map);
218693ab6cc6SEric Dumazet 	}
218705255b82SEric Dumazet out:
21887a7f0946SArjun Roy 	if (mmap_locked)
2189d8ed45c5SMichel Lespinasse 		mmap_read_unlock(current->mm);
21907a7f0946SArjun Roy 	else
21917a7f0946SArjun Roy 		vma_end_read(vma);
219218fb76edSArjun Roy 	/* Try to copy straggler data. */
219318fb76edSArjun Roy 	if (!ret)
21947eeba170SArjun Roy 		copylen = tcp_zc_handle_leftover(zc, sk, skb, &seq, copybuf_len, tss);
219518fb76edSArjun Roy 
219618fb76edSArjun Roy 	if (length + copylen) {
21977db48e98SEric Dumazet 		WRITE_ONCE(tp->copied_seq, seq);
219893ab6cc6SEric Dumazet 		tcp_rcv_space_adjust(sk);
219993ab6cc6SEric Dumazet 
220093ab6cc6SEric Dumazet 		/* Clean up data we have read: This will do ACK frames. */
220193ab6cc6SEric Dumazet 		tcp_recv_skb(sk, seq, &offset);
220218fb76edSArjun Roy 		tcp_cleanup_rbuf(sk, length + copylen);
220393ab6cc6SEric Dumazet 		ret = 0;
220405255b82SEric Dumazet 		if (length == zc->length)
220505255b82SEric Dumazet 			zc->recv_skip_hint = 0;
220605255b82SEric Dumazet 	} else {
220705255b82SEric Dumazet 		if (!zc->recv_skip_hint && sock_flag(sk, SOCK_DONE))
220805255b82SEric Dumazet 			ret = -EIO;
220905255b82SEric Dumazet 	}
221005255b82SEric Dumazet 	zc->length = length;
221193ab6cc6SEric Dumazet 	return ret;
221293ab6cc6SEric Dumazet }
221305255b82SEric Dumazet #endif
221493ab6cc6SEric Dumazet 
221598aaa913SMike Maloney /* Similar to __sock_recv_timestamp, but does not require an skb */
2216892bfd3dSFlorian Westphal void tcp_recv_timestamp(struct msghdr *msg, const struct sock *sk,
22179718475eSDeepa Dinamani 			struct scm_timestamping_internal *tss)
221898aaa913SMike Maloney {
2219887feae3SDeepa Dinamani 	int new_tstamp = sock_flag(sk, SOCK_TSTAMP_NEW);
222098aaa913SMike Maloney 	bool has_timestamping = false;
222198aaa913SMike Maloney 
222298aaa913SMike Maloney 	if (tss->ts[0].tv_sec || tss->ts[0].tv_nsec) {
222398aaa913SMike Maloney 		if (sock_flag(sk, SOCK_RCVTSTAMP)) {
222498aaa913SMike Maloney 			if (sock_flag(sk, SOCK_RCVTSTAMPNS)) {
2225887feae3SDeepa Dinamani 				if (new_tstamp) {
2226df1b4ba9SArnd Bergmann 					struct __kernel_timespec kts = {
2227df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
2228df1b4ba9SArnd Bergmann 						.tv_nsec = tss->ts[0].tv_nsec,
2229df1b4ba9SArnd Bergmann 					};
2230887feae3SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_NEW,
2231887feae3SDeepa Dinamani 						 sizeof(kts), &kts);
2232887feae3SDeepa Dinamani 				} else {
2233df1b4ba9SArnd Bergmann 					struct __kernel_old_timespec ts_old = {
2234df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
2235df1b4ba9SArnd Bergmann 						.tv_nsec = tss->ts[0].tv_nsec,
2236df1b4ba9SArnd Bergmann 					};
22377f1bc6e9SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_OLD,
22389718475eSDeepa Dinamani 						 sizeof(ts_old), &ts_old);
2239887feae3SDeepa Dinamani 				}
224098aaa913SMike Maloney 			} else {
2241887feae3SDeepa Dinamani 				if (new_tstamp) {
2242df1b4ba9SArnd Bergmann 					struct __kernel_sock_timeval stv = {
2243df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
2244df1b4ba9SArnd Bergmann 						.tv_usec = tss->ts[0].tv_nsec / 1000,
2245df1b4ba9SArnd Bergmann 					};
2246887feae3SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_NEW,
2247887feae3SDeepa Dinamani 						 sizeof(stv), &stv);
2248887feae3SDeepa Dinamani 				} else {
2249df1b4ba9SArnd Bergmann 					struct __kernel_old_timeval tv = {
2250df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
2251df1b4ba9SArnd Bergmann 						.tv_usec = tss->ts[0].tv_nsec / 1000,
2252df1b4ba9SArnd Bergmann 					};
22537f1bc6e9SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_OLD,
225498aaa913SMike Maloney 						 sizeof(tv), &tv);
225598aaa913SMike Maloney 				}
225698aaa913SMike Maloney 			}
2257887feae3SDeepa Dinamani 		}
225898aaa913SMike Maloney 
225998aaa913SMike Maloney 		if (sk->sk_tsflags & SOF_TIMESTAMPING_SOFTWARE)
226098aaa913SMike Maloney 			has_timestamping = true;
226198aaa913SMike Maloney 		else
22629718475eSDeepa Dinamani 			tss->ts[0] = (struct timespec64) {0};
226398aaa913SMike Maloney 	}
226498aaa913SMike Maloney 
226598aaa913SMike Maloney 	if (tss->ts[2].tv_sec || tss->ts[2].tv_nsec) {
226698aaa913SMike Maloney 		if (sk->sk_tsflags & SOF_TIMESTAMPING_RAW_HARDWARE)
226798aaa913SMike Maloney 			has_timestamping = true;
226898aaa913SMike Maloney 		else
22699718475eSDeepa Dinamani 			tss->ts[2] = (struct timespec64) {0};
227098aaa913SMike Maloney 	}
227198aaa913SMike Maloney 
227298aaa913SMike Maloney 	if (has_timestamping) {
22739718475eSDeepa Dinamani 		tss->ts[1] = (struct timespec64) {0};
22749718475eSDeepa Dinamani 		if (sock_flag(sk, SOCK_TSTAMP_NEW))
22759718475eSDeepa Dinamani 			put_cmsg_scm_timestamping64(msg, tss);
22769718475eSDeepa Dinamani 		else
22779718475eSDeepa Dinamani 			put_cmsg_scm_timestamping(msg, tss);
227898aaa913SMike Maloney 	}
227998aaa913SMike Maloney }
228098aaa913SMike Maloney 
2281b75eba76SSoheil Hassas Yeganeh static int tcp_inq_hint(struct sock *sk)
2282b75eba76SSoheil Hassas Yeganeh {
2283b75eba76SSoheil Hassas Yeganeh 	const struct tcp_sock *tp = tcp_sk(sk);
2284b75eba76SSoheil Hassas Yeganeh 	u32 copied_seq = READ_ONCE(tp->copied_seq);
2285b75eba76SSoheil Hassas Yeganeh 	u32 rcv_nxt = READ_ONCE(tp->rcv_nxt);
2286b75eba76SSoheil Hassas Yeganeh 	int inq;
2287b75eba76SSoheil Hassas Yeganeh 
2288b75eba76SSoheil Hassas Yeganeh 	inq = rcv_nxt - copied_seq;
2289b75eba76SSoheil Hassas Yeganeh 	if (unlikely(inq < 0 || copied_seq != READ_ONCE(tp->copied_seq))) {
2290b75eba76SSoheil Hassas Yeganeh 		lock_sock(sk);
2291b75eba76SSoheil Hassas Yeganeh 		inq = tp->rcv_nxt - tp->copied_seq;
2292b75eba76SSoheil Hassas Yeganeh 		release_sock(sk);
2293b75eba76SSoheil Hassas Yeganeh 	}
22946466e715SSoheil Hassas Yeganeh 	/* After receiving a FIN, tell the user-space to continue reading
22956466e715SSoheil Hassas Yeganeh 	 * by returning a non-zero inq.
22966466e715SSoheil Hassas Yeganeh 	 */
22976466e715SSoheil Hassas Yeganeh 	if (inq == 0 && sock_flag(sk, SOCK_DONE))
22986466e715SSoheil Hassas Yeganeh 		inq = 1;
2299b75eba76SSoheil Hassas Yeganeh 	return inq;
2300b75eba76SSoheil Hassas Yeganeh }
2301b75eba76SSoheil Hassas Yeganeh 
23021da177e4SLinus Torvalds /*
23031da177e4SLinus Torvalds  *	This routine copies from a sock struct into the user buffer.
23041da177e4SLinus Torvalds  *
23051da177e4SLinus Torvalds  *	Technical note: in 2.3 we work on _locked_ socket, so that
23061da177e4SLinus Torvalds  *	tricks with *seq access order and skb->users are not required.
23071da177e4SLinus Torvalds  *	Probably, code can be easily improved even more.
23081da177e4SLinus Torvalds  */
23091da177e4SLinus Torvalds 
23102cd81161SArjun Roy static int tcp_recvmsg_locked(struct sock *sk, struct msghdr *msg, size_t len,
2311ec095263SOliver Hartkopp 			      int flags, struct scm_timestamping_internal *tss,
23122cd81161SArjun Roy 			      int *cmsg_flags)
23131da177e4SLinus Torvalds {
23141da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
23151da177e4SLinus Torvalds 	int copied = 0;
23161da177e4SLinus Torvalds 	u32 peek_seq;
23171da177e4SLinus Torvalds 	u32 *seq;
23181da177e4SLinus Torvalds 	unsigned long used;
23192cd81161SArjun Roy 	int err;
23201da177e4SLinus Torvalds 	int target;		/* Read at least this many bytes */
23211da177e4SLinus Torvalds 	long timeo;
2322dfbafc99SSabrina Dubroca 	struct sk_buff *skb, *last;
232377527313SIlpo Järvinen 	u32 urg_hole = 0;
23241da177e4SLinus Torvalds 
23251da177e4SLinus Torvalds 	err = -ENOTCONN;
23261da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN)
23271da177e4SLinus Torvalds 		goto out;
23281da177e4SLinus Torvalds 
2329f94fd25cSJens Axboe 	if (tp->recvmsg_inq) {
2330925bba24SArjun Roy 		*cmsg_flags = TCP_CMSG_INQ;
2331f94fd25cSJens Axboe 		msg->msg_get_inq = 1;
2332f94fd25cSJens Axboe 	}
2333ec095263SOliver Hartkopp 	timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
23341da177e4SLinus Torvalds 
23351da177e4SLinus Torvalds 	/* Urgent data needs to be handled specially. */
23361da177e4SLinus Torvalds 	if (flags & MSG_OOB)
23371da177e4SLinus Torvalds 		goto recv_urg;
23381da177e4SLinus Torvalds 
2339c0e88ff0SPavel Emelyanov 	if (unlikely(tp->repair)) {
2340c0e88ff0SPavel Emelyanov 		err = -EPERM;
2341c0e88ff0SPavel Emelyanov 		if (!(flags & MSG_PEEK))
2342c0e88ff0SPavel Emelyanov 			goto out;
2343c0e88ff0SPavel Emelyanov 
2344c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_SEND_QUEUE)
2345c0e88ff0SPavel Emelyanov 			goto recv_sndq;
2346c0e88ff0SPavel Emelyanov 
2347c0e88ff0SPavel Emelyanov 		err = -EINVAL;
2348c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_NO_QUEUE)
2349c0e88ff0SPavel Emelyanov 			goto out;
2350c0e88ff0SPavel Emelyanov 
2351c0e88ff0SPavel Emelyanov 		/* 'common' recv queue MSG_PEEK-ing */
2352c0e88ff0SPavel Emelyanov 	}
2353c0e88ff0SPavel Emelyanov 
23541da177e4SLinus Torvalds 	seq = &tp->copied_seq;
23551da177e4SLinus Torvalds 	if (flags & MSG_PEEK) {
23561da177e4SLinus Torvalds 		peek_seq = tp->copied_seq;
23571da177e4SLinus Torvalds 		seq = &peek_seq;
23581da177e4SLinus Torvalds 	}
23591da177e4SLinus Torvalds 
23601da177e4SLinus Torvalds 	target = sock_rcvlowat(sk, flags & MSG_WAITALL, len);
23611da177e4SLinus Torvalds 
23621da177e4SLinus Torvalds 	do {
23631da177e4SLinus Torvalds 		u32 offset;
23641da177e4SLinus Torvalds 
23651da177e4SLinus Torvalds 		/* Are we at urgent data? Stop if we have read anything or have SIGURG pending. */
2366b96c51bdSEric Dumazet 		if (unlikely(tp->urg_data) && tp->urg_seq == *seq) {
23671da177e4SLinus Torvalds 			if (copied)
23681da177e4SLinus Torvalds 				break;
23691da177e4SLinus Torvalds 			if (signal_pending(current)) {
23701da177e4SLinus Torvalds 				copied = timeo ? sock_intr_errno(timeo) : -EAGAIN;
23711da177e4SLinus Torvalds 				break;
23721da177e4SLinus Torvalds 			}
23731da177e4SLinus Torvalds 		}
23741da177e4SLinus Torvalds 
23751da177e4SLinus Torvalds 		/* Next get a buffer. */
23761da177e4SLinus Torvalds 
2377dfbafc99SSabrina Dubroca 		last = skb_peek_tail(&sk->sk_receive_queue);
237891521944SDavid S. Miller 		skb_queue_walk(&sk->sk_receive_queue, skb) {
2379dfbafc99SSabrina Dubroca 			last = skb;
23801da177e4SLinus Torvalds 			/* Now that we have two receive queues this
23811da177e4SLinus Torvalds 			 * shouldn't happen.
23821da177e4SLinus Torvalds 			 */
2383d792c100SIlpo Järvinen 			if (WARN(before(*seq, TCP_SKB_CB(skb)->seq),
2384e56b8ce3SRandy Dunlap 				 "TCP recvmsg seq # bug: copied %X, seq %X, rcvnxt %X, fl %X\n",
23852af6fd8bSJoe Perches 				 *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt,
2386d792c100SIlpo Järvinen 				 flags))
23871da177e4SLinus Torvalds 				break;
2388d792c100SIlpo Järvinen 
23891da177e4SLinus Torvalds 			offset = *seq - TCP_SKB_CB(skb)->seq;
23909d691539SEric Dumazet 			if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) {
23919d691539SEric Dumazet 				pr_err_once("%s: found a SYN, please report !\n", __func__);
23921da177e4SLinus Torvalds 				offset--;
23939d691539SEric Dumazet 			}
23941da177e4SLinus Torvalds 			if (offset < skb->len)
23951da177e4SLinus Torvalds 				goto found_ok_skb;
2396e11ecddfSEric Dumazet 			if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
23971da177e4SLinus Torvalds 				goto found_fin_ok;
23982af6fd8bSJoe Perches 			WARN(!(flags & MSG_PEEK),
2399e56b8ce3SRandy Dunlap 			     "TCP recvmsg seq # bug 2: copied %X, seq %X, rcvnxt %X, fl %X\n",
24002af6fd8bSJoe Perches 			     *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt, flags);
240191521944SDavid S. Miller 		}
24021da177e4SLinus Torvalds 
24031da177e4SLinus Torvalds 		/* Well, if we have backlog, try to process it now yet. */
24041da177e4SLinus Torvalds 
24059ed498c6SEric Dumazet 		if (copied >= target && !READ_ONCE(sk->sk_backlog.tail))
24061da177e4SLinus Torvalds 			break;
24071da177e4SLinus Torvalds 
24081da177e4SLinus Torvalds 		if (copied) {
24098bd172b7SEric Dumazet 			if (!timeo ||
24108bd172b7SEric Dumazet 			    sk->sk_err ||
24111da177e4SLinus Torvalds 			    sk->sk_state == TCP_CLOSE ||
24121da177e4SLinus Torvalds 			    (sk->sk_shutdown & RCV_SHUTDOWN) ||
2413518a09efSDavid S. Miller 			    signal_pending(current))
24141da177e4SLinus Torvalds 				break;
24151da177e4SLinus Torvalds 		} else {
24161da177e4SLinus Torvalds 			if (sock_flag(sk, SOCK_DONE))
24171da177e4SLinus Torvalds 				break;
24181da177e4SLinus Torvalds 
24191da177e4SLinus Torvalds 			if (sk->sk_err) {
24201da177e4SLinus Torvalds 				copied = sock_error(sk);
24211da177e4SLinus Torvalds 				break;
24221da177e4SLinus Torvalds 			}
24231da177e4SLinus Torvalds 
24241da177e4SLinus Torvalds 			if (sk->sk_shutdown & RCV_SHUTDOWN)
24251da177e4SLinus Torvalds 				break;
24261da177e4SLinus Torvalds 
24271da177e4SLinus Torvalds 			if (sk->sk_state == TCP_CLOSE) {
24281da177e4SLinus Torvalds 				/* This occurs when user tries to read
24291da177e4SLinus Torvalds 				 * from never connected socket.
24301da177e4SLinus Torvalds 				 */
24311da177e4SLinus Torvalds 				copied = -ENOTCONN;
24321da177e4SLinus Torvalds 				break;
24331da177e4SLinus Torvalds 			}
24341da177e4SLinus Torvalds 
24351da177e4SLinus Torvalds 			if (!timeo) {
24361da177e4SLinus Torvalds 				copied = -EAGAIN;
24371da177e4SLinus Torvalds 				break;
24381da177e4SLinus Torvalds 			}
24391da177e4SLinus Torvalds 
24401da177e4SLinus Torvalds 			if (signal_pending(current)) {
24411da177e4SLinus Torvalds 				copied = sock_intr_errno(timeo);
24421da177e4SLinus Torvalds 				break;
24431da177e4SLinus Torvalds 			}
24441da177e4SLinus Torvalds 		}
24451da177e4SLinus Torvalds 
24461da177e4SLinus Torvalds 		if (copied >= target) {
24471da177e4SLinus Torvalds 			/* Do not sleep, just process backlog. */
244893afcfd1SEric Dumazet 			__sk_flush_backlog(sk);
2449dfbafc99SSabrina Dubroca 		} else {
245029fbc26eSEric Dumazet 			tcp_cleanup_rbuf(sk, copied);
2451dfbafc99SSabrina Dubroca 			sk_wait_data(sk, &timeo, last);
2452dfbafc99SSabrina Dubroca 		}
24531da177e4SLinus Torvalds 
245477527313SIlpo Järvinen 		if ((flags & MSG_PEEK) &&
245577527313SIlpo Järvinen 		    (peek_seq - copied - urg_hole != tp->copied_seq)) {
2456e87cc472SJoe Perches 			net_dbg_ratelimited("TCP(%s:%d): Application bug, race in MSG_PEEK\n",
2457e87cc472SJoe Perches 					    current->comm,
2458e87cc472SJoe Perches 					    task_pid_nr(current));
24591da177e4SLinus Torvalds 			peek_seq = tp->copied_seq;
24601da177e4SLinus Torvalds 		}
24611da177e4SLinus Torvalds 		continue;
24621da177e4SLinus Torvalds 
24631da177e4SLinus Torvalds found_ok_skb:
24641da177e4SLinus Torvalds 		/* Ok so how much can we use? */
24651da177e4SLinus Torvalds 		used = skb->len - offset;
24661da177e4SLinus Torvalds 		if (len < used)
24671da177e4SLinus Torvalds 			used = len;
24681da177e4SLinus Torvalds 
24691da177e4SLinus Torvalds 		/* Do we have urgent data here? */
2470b96c51bdSEric Dumazet 		if (unlikely(tp->urg_data)) {
24711da177e4SLinus Torvalds 			u32 urg_offset = tp->urg_seq - *seq;
24721da177e4SLinus Torvalds 			if (urg_offset < used) {
24731da177e4SLinus Torvalds 				if (!urg_offset) {
24741da177e4SLinus Torvalds 					if (!sock_flag(sk, SOCK_URGINLINE)) {
24757db48e98SEric Dumazet 						WRITE_ONCE(*seq, *seq + 1);
247677527313SIlpo Järvinen 						urg_hole++;
24771da177e4SLinus Torvalds 						offset++;
24781da177e4SLinus Torvalds 						used--;
24791da177e4SLinus Torvalds 						if (!used)
24801da177e4SLinus Torvalds 							goto skip_copy;
24811da177e4SLinus Torvalds 					}
24821da177e4SLinus Torvalds 				} else
24831da177e4SLinus Torvalds 					used = urg_offset;
24841da177e4SLinus Torvalds 			}
24851da177e4SLinus Torvalds 		}
24861da177e4SLinus Torvalds 
24871da177e4SLinus Torvalds 		if (!(flags & MSG_TRUNC)) {
248851f3d02bSDavid S. Miller 			err = skb_copy_datagram_msg(skb, offset, msg, used);
24891da177e4SLinus Torvalds 			if (err) {
24901da177e4SLinus Torvalds 				/* Exception. Bailout! */
24911da177e4SLinus Torvalds 				if (!copied)
24921da177e4SLinus Torvalds 					copied = -EFAULT;
24931da177e4SLinus Torvalds 				break;
24941da177e4SLinus Torvalds 			}
24951da177e4SLinus Torvalds 		}
24961da177e4SLinus Torvalds 
24977db48e98SEric Dumazet 		WRITE_ONCE(*seq, *seq + used);
24981da177e4SLinus Torvalds 		copied += used;
24991da177e4SLinus Torvalds 		len -= used;
25001da177e4SLinus Torvalds 
25011da177e4SLinus Torvalds 		tcp_rcv_space_adjust(sk);
25021da177e4SLinus Torvalds 
25031da177e4SLinus Torvalds skip_copy:
2504b96c51bdSEric Dumazet 		if (unlikely(tp->urg_data) && after(tp->copied_seq, tp->urg_seq)) {
25057b6a893aSEric Dumazet 			WRITE_ONCE(tp->urg_data, 0);
250631770e34SFlorian Westphal 			tcp_fast_path_check(sk);
250731770e34SFlorian Westphal 		}
25081da177e4SLinus Torvalds 
250998aaa913SMike Maloney 		if (TCP_SKB_CB(skb)->has_rxtstamp) {
25102cd81161SArjun Roy 			tcp_update_recv_tstamps(skb, tss);
2511925bba24SArjun Roy 			*cmsg_flags |= TCP_CMSG_TS;
251298aaa913SMike Maloney 		}
2513cc4de047SKelly Littlepage 
2514cc4de047SKelly Littlepage 		if (used + offset < skb->len)
2515cc4de047SKelly Littlepage 			continue;
2516cc4de047SKelly Littlepage 
2517e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
25181da177e4SLinus Torvalds 			goto found_fin_ok;
25197bced397SDan Williams 		if (!(flags & MSG_PEEK))
25203df684c1SEric Dumazet 			tcp_eat_recv_skb(sk, skb);
25211da177e4SLinus Torvalds 		continue;
25221da177e4SLinus Torvalds 
25231da177e4SLinus Torvalds found_fin_ok:
25241da177e4SLinus Torvalds 		/* Process the FIN. */
25257db48e98SEric Dumazet 		WRITE_ONCE(*seq, *seq + 1);
25267bced397SDan Williams 		if (!(flags & MSG_PEEK))
25273df684c1SEric Dumazet 			tcp_eat_recv_skb(sk, skb);
25281da177e4SLinus Torvalds 		break;
25291da177e4SLinus Torvalds 	} while (len > 0);
25301da177e4SLinus Torvalds 
25311da177e4SLinus Torvalds 	/* According to UNIX98, msg_name/msg_namelen are ignored
25321da177e4SLinus Torvalds 	 * on connected socket. I was just happy when found this 8) --ANK
25331da177e4SLinus Torvalds 	 */
25341da177e4SLinus Torvalds 
25351da177e4SLinus Torvalds 	/* Clean up data we have read: This will do ACK frames. */
25360e4b4992SChris Leech 	tcp_cleanup_rbuf(sk, copied);
25371da177e4SLinus Torvalds 	return copied;
25381da177e4SLinus Torvalds 
25391da177e4SLinus Torvalds out:
25401da177e4SLinus Torvalds 	return err;
25411da177e4SLinus Torvalds 
25421da177e4SLinus Torvalds recv_urg:
2543377f0a08SRami Rosen 	err = tcp_recv_urg(sk, msg, len, flags);
25441da177e4SLinus Torvalds 	goto out;
2545c0e88ff0SPavel Emelyanov 
2546c0e88ff0SPavel Emelyanov recv_sndq:
2547c0e88ff0SPavel Emelyanov 	err = tcp_peek_sndq(sk, msg, len);
2548c0e88ff0SPavel Emelyanov 	goto out;
25491da177e4SLinus Torvalds }
25502cd81161SArjun Roy 
2551ec095263SOliver Hartkopp int tcp_recvmsg(struct sock *sk, struct msghdr *msg, size_t len, int flags,
2552ec095263SOliver Hartkopp 		int *addr_len)
25532cd81161SArjun Roy {
2554f94fd25cSJens Axboe 	int cmsg_flags = 0, ret;
25552cd81161SArjun Roy 	struct scm_timestamping_internal tss;
25562cd81161SArjun Roy 
25572cd81161SArjun Roy 	if (unlikely(flags & MSG_ERRQUEUE))
25582cd81161SArjun Roy 		return inet_recv_error(sk, msg, len, addr_len);
25592cd81161SArjun Roy 
25602cd81161SArjun Roy 	if (sk_can_busy_loop(sk) &&
25612cd81161SArjun Roy 	    skb_queue_empty_lockless(&sk->sk_receive_queue) &&
25622cd81161SArjun Roy 	    sk->sk_state == TCP_ESTABLISHED)
2563ec095263SOliver Hartkopp 		sk_busy_loop(sk, flags & MSG_DONTWAIT);
25642cd81161SArjun Roy 
25652cd81161SArjun Roy 	lock_sock(sk);
2566ec095263SOliver Hartkopp 	ret = tcp_recvmsg_locked(sk, msg, len, flags, &tss, &cmsg_flags);
25672cd81161SArjun Roy 	release_sock(sk);
25682cd81161SArjun Roy 
2569f94fd25cSJens Axboe 	if ((cmsg_flags || msg->msg_get_inq) && ret >= 0) {
2570925bba24SArjun Roy 		if (cmsg_flags & TCP_CMSG_TS)
25712cd81161SArjun Roy 			tcp_recv_timestamp(msg, sk, &tss);
2572f94fd25cSJens Axboe 		if (msg->msg_get_inq) {
2573f94fd25cSJens Axboe 			msg->msg_inq = tcp_inq_hint(sk);
2574f94fd25cSJens Axboe 			if (cmsg_flags & TCP_CMSG_INQ)
2575f94fd25cSJens Axboe 				put_cmsg(msg, SOL_TCP, TCP_CM_INQ,
2576f94fd25cSJens Axboe 					 sizeof(msg->msg_inq), &msg->msg_inq);
25772cd81161SArjun Roy 		}
25782cd81161SArjun Roy 	}
25792cd81161SArjun Roy 	return ret;
25802cd81161SArjun Roy }
25814bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_recvmsg);
25821da177e4SLinus Torvalds 
2583490d5046SIlpo Järvinen void tcp_set_state(struct sock *sk, int state)
2584490d5046SIlpo Järvinen {
2585490d5046SIlpo Järvinen 	int oldstate = sk->sk_state;
2586490d5046SIlpo Järvinen 
2587d4487491SLawrence Brakmo 	/* We defined a new enum for TCP states that are exported in BPF
2588d4487491SLawrence Brakmo 	 * so as not force the internal TCP states to be frozen. The
2589d4487491SLawrence Brakmo 	 * following checks will detect if an internal state value ever
2590d4487491SLawrence Brakmo 	 * differs from the BPF value. If this ever happens, then we will
2591d4487491SLawrence Brakmo 	 * need to remap the internal value to the BPF value before calling
2592d4487491SLawrence Brakmo 	 * tcp_call_bpf_2arg.
2593d4487491SLawrence Brakmo 	 */
2594d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_ESTABLISHED != (int)TCP_ESTABLISHED);
2595d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_SYN_SENT != (int)TCP_SYN_SENT);
2596d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_SYN_RECV != (int)TCP_SYN_RECV);
2597d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT1 != (int)TCP_FIN_WAIT1);
2598d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT2 != (int)TCP_FIN_WAIT2);
2599d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_TIME_WAIT != (int)TCP_TIME_WAIT);
2600d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSE != (int)TCP_CLOSE);
2601d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSE_WAIT != (int)TCP_CLOSE_WAIT);
2602d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_LAST_ACK != (int)TCP_LAST_ACK);
2603d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_LISTEN != (int)TCP_LISTEN);
2604d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSING != (int)TCP_CLOSING);
2605d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_NEW_SYN_RECV != (int)TCP_NEW_SYN_RECV);
2606d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_MAX_STATES != (int)TCP_MAX_STATES);
2607d4487491SLawrence Brakmo 
260897a19cafSYonghong Song 	/* bpf uapi header bpf.h defines an anonymous enum with values
260997a19cafSYonghong Song 	 * BPF_TCP_* used by bpf programs. Currently gcc built vmlinux
261097a19cafSYonghong Song 	 * is able to emit this enum in DWARF due to the above BUILD_BUG_ON.
261197a19cafSYonghong Song 	 * But clang built vmlinux does not have this enum in DWARF
261297a19cafSYonghong Song 	 * since clang removes the above code before generating IR/debuginfo.
261397a19cafSYonghong Song 	 * Let us explicitly emit the type debuginfo to ensure the
261497a19cafSYonghong Song 	 * above-mentioned anonymous enum in the vmlinux DWARF and hence BTF
261597a19cafSYonghong Song 	 * regardless of which compiler is used.
261697a19cafSYonghong Song 	 */
261797a19cafSYonghong Song 	BTF_TYPE_EMIT_ENUM(BPF_TCP_ESTABLISHED);
261897a19cafSYonghong Song 
2619d4487491SLawrence Brakmo 	if (BPF_SOCK_OPS_TEST_FLAG(tcp_sk(sk), BPF_SOCK_OPS_STATE_CB_FLAG))
2620d4487491SLawrence Brakmo 		tcp_call_bpf_2arg(sk, BPF_SOCK_OPS_STATE_CB, oldstate, state);
2621e8fce239SSong Liu 
2622490d5046SIlpo Järvinen 	switch (state) {
2623490d5046SIlpo Järvinen 	case TCP_ESTABLISHED:
2624490d5046SIlpo Järvinen 		if (oldstate != TCP_ESTABLISHED)
262581cc8a75SPavel Emelyanov 			TCP_INC_STATS(sock_net(sk), TCP_MIB_CURRESTAB);
2626490d5046SIlpo Järvinen 		break;
2627490d5046SIlpo Järvinen 
2628490d5046SIlpo Järvinen 	case TCP_CLOSE:
2629490d5046SIlpo Järvinen 		if (oldstate == TCP_CLOSE_WAIT || oldstate == TCP_ESTABLISHED)
263081cc8a75SPavel Emelyanov 			TCP_INC_STATS(sock_net(sk), TCP_MIB_ESTABRESETS);
2631490d5046SIlpo Järvinen 
2632490d5046SIlpo Järvinen 		sk->sk_prot->unhash(sk);
2633490d5046SIlpo Järvinen 		if (inet_csk(sk)->icsk_bind_hash &&
2634490d5046SIlpo Järvinen 		    !(sk->sk_userlocks & SOCK_BINDPORT_LOCK))
2635ab1e0a13SArnaldo Carvalho de Melo 			inet_put_port(sk);
2636a8eceea8SJoe Perches 		fallthrough;
2637490d5046SIlpo Järvinen 	default:
2638490d5046SIlpo Järvinen 		if (oldstate == TCP_ESTABLISHED)
263974688e48SPavel Emelyanov 			TCP_DEC_STATS(sock_net(sk), TCP_MIB_CURRESTAB);
2640490d5046SIlpo Järvinen 	}
2641490d5046SIlpo Järvinen 
2642490d5046SIlpo Järvinen 	/* Change state AFTER socket is unhashed to avoid closed
2643490d5046SIlpo Järvinen 	 * socket sitting in hash tables.
2644490d5046SIlpo Järvinen 	 */
2645563e0bb0SYafang Shao 	inet_sk_state_store(sk, state);
2646490d5046SIlpo Järvinen }
2647490d5046SIlpo Järvinen EXPORT_SYMBOL_GPL(tcp_set_state);
2648490d5046SIlpo Järvinen 
26491da177e4SLinus Torvalds /*
26501da177e4SLinus Torvalds  *	State processing on a close. This implements the state shift for
26511da177e4SLinus Torvalds  *	sending our FIN frame. Note that we only send a FIN for some
26521da177e4SLinus Torvalds  *	states. A shutdown() may have already sent the FIN, or we may be
26531da177e4SLinus Torvalds  *	closed.
26541da177e4SLinus Torvalds  */
26551da177e4SLinus Torvalds 
26569b5b5cffSArjan van de Ven static const unsigned char new_state[16] = {
26571da177e4SLinus Torvalds   /* current state:        new state:      action:	*/
26580980c1e3SEric Dumazet   [0 /* (Invalid) */]	= TCP_CLOSE,
26590980c1e3SEric Dumazet   [TCP_ESTABLISHED]	= TCP_FIN_WAIT1 | TCP_ACTION_FIN,
26600980c1e3SEric Dumazet   [TCP_SYN_SENT]	= TCP_CLOSE,
26610980c1e3SEric Dumazet   [TCP_SYN_RECV]	= TCP_FIN_WAIT1 | TCP_ACTION_FIN,
26620980c1e3SEric Dumazet   [TCP_FIN_WAIT1]	= TCP_FIN_WAIT1,
26630980c1e3SEric Dumazet   [TCP_FIN_WAIT2]	= TCP_FIN_WAIT2,
26640980c1e3SEric Dumazet   [TCP_TIME_WAIT]	= TCP_CLOSE,
26650980c1e3SEric Dumazet   [TCP_CLOSE]		= TCP_CLOSE,
26660980c1e3SEric Dumazet   [TCP_CLOSE_WAIT]	= TCP_LAST_ACK  | TCP_ACTION_FIN,
26670980c1e3SEric Dumazet   [TCP_LAST_ACK]	= TCP_LAST_ACK,
26680980c1e3SEric Dumazet   [TCP_LISTEN]		= TCP_CLOSE,
26690980c1e3SEric Dumazet   [TCP_CLOSING]		= TCP_CLOSING,
26700980c1e3SEric Dumazet   [TCP_NEW_SYN_RECV]	= TCP_CLOSE,	/* should not happen ! */
26711da177e4SLinus Torvalds };
26721da177e4SLinus Torvalds 
26731da177e4SLinus Torvalds static int tcp_close_state(struct sock *sk)
26741da177e4SLinus Torvalds {
26751da177e4SLinus Torvalds 	int next = (int)new_state[sk->sk_state];
26761da177e4SLinus Torvalds 	int ns = next & TCP_STATE_MASK;
26771da177e4SLinus Torvalds 
26781da177e4SLinus Torvalds 	tcp_set_state(sk, ns);
26791da177e4SLinus Torvalds 
26801da177e4SLinus Torvalds 	return next & TCP_ACTION_FIN;
26811da177e4SLinus Torvalds }
26821da177e4SLinus Torvalds 
26831da177e4SLinus Torvalds /*
26841da177e4SLinus Torvalds  *	Shutdown the sending side of a connection. Much like close except
26851f29b058SSatoru SATOH  *	that we don't receive shut down or sock_set_flag(sk, SOCK_DEAD).
26861da177e4SLinus Torvalds  */
26871da177e4SLinus Torvalds 
26881da177e4SLinus Torvalds void tcp_shutdown(struct sock *sk, int how)
26891da177e4SLinus Torvalds {
26901da177e4SLinus Torvalds 	/*	We need to grab some memory, and put together a FIN,
26911da177e4SLinus Torvalds 	 *	and then put it into the queue to be sent.
26921da177e4SLinus Torvalds 	 *		Tim MacKenzie(tym@dibbler.cs.monash.edu.au) 4 Dec '92.
26931da177e4SLinus Torvalds 	 */
26941da177e4SLinus Torvalds 	if (!(how & SEND_SHUTDOWN))
26951da177e4SLinus Torvalds 		return;
26961da177e4SLinus Torvalds 
26971da177e4SLinus Torvalds 	/* If we've already sent a FIN, or it's a closed state, skip this. */
26981da177e4SLinus Torvalds 	if ((1 << sk->sk_state) &
26991da177e4SLinus Torvalds 	    (TCPF_ESTABLISHED | TCPF_SYN_SENT |
27001da177e4SLinus Torvalds 	     TCPF_SYN_RECV | TCPF_CLOSE_WAIT)) {
27011da177e4SLinus Torvalds 		/* Clear out any half completed packets.  FIN if needed. */
27021da177e4SLinus Torvalds 		if (tcp_close_state(sk))
27031da177e4SLinus Torvalds 			tcp_send_fin(sk);
27041da177e4SLinus Torvalds 	}
27051da177e4SLinus Torvalds }
27064bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_shutdown);
27071da177e4SLinus Torvalds 
270819757cebSEric Dumazet int tcp_orphan_count_sum(void)
270919757cebSEric Dumazet {
271019757cebSEric Dumazet 	int i, total = 0;
271119757cebSEric Dumazet 
271219757cebSEric Dumazet 	for_each_possible_cpu(i)
271319757cebSEric Dumazet 		total += per_cpu(tcp_orphan_count, i);
271419757cebSEric Dumazet 
271519757cebSEric Dumazet 	return max(total, 0);
271619757cebSEric Dumazet }
271719757cebSEric Dumazet 
271819757cebSEric Dumazet static int tcp_orphan_cache;
271919757cebSEric Dumazet static struct timer_list tcp_orphan_timer;
272019757cebSEric Dumazet #define TCP_ORPHAN_TIMER_PERIOD msecs_to_jiffies(100)
272119757cebSEric Dumazet 
272219757cebSEric Dumazet static void tcp_orphan_update(struct timer_list *unused)
272319757cebSEric Dumazet {
272419757cebSEric Dumazet 	WRITE_ONCE(tcp_orphan_cache, tcp_orphan_count_sum());
272519757cebSEric Dumazet 	mod_timer(&tcp_orphan_timer, jiffies + TCP_ORPHAN_TIMER_PERIOD);
272619757cebSEric Dumazet }
272719757cebSEric Dumazet 
272819757cebSEric Dumazet static bool tcp_too_many_orphans(int shift)
272919757cebSEric Dumazet {
273047e6ab24SKuniyuki Iwashima 	return READ_ONCE(tcp_orphan_cache) << shift >
273147e6ab24SKuniyuki Iwashima 		READ_ONCE(sysctl_tcp_max_orphans);
273219757cebSEric Dumazet }
273319757cebSEric Dumazet 
2734efcdbf24SArun Sharma bool tcp_check_oom(struct sock *sk, int shift)
2735efcdbf24SArun Sharma {
2736efcdbf24SArun Sharma 	bool too_many_orphans, out_of_socket_memory;
2737efcdbf24SArun Sharma 
273819757cebSEric Dumazet 	too_many_orphans = tcp_too_many_orphans(shift);
2739efcdbf24SArun Sharma 	out_of_socket_memory = tcp_out_of_memory(sk);
2740efcdbf24SArun Sharma 
2741e87cc472SJoe Perches 	if (too_many_orphans)
2742e87cc472SJoe Perches 		net_info_ratelimited("too many orphaned sockets\n");
2743e87cc472SJoe Perches 	if (out_of_socket_memory)
2744e87cc472SJoe Perches 		net_info_ratelimited("out of memory -- consider tuning tcp_mem\n");
2745efcdbf24SArun Sharma 	return too_many_orphans || out_of_socket_memory;
2746efcdbf24SArun Sharma }
2747efcdbf24SArun Sharma 
274877c3c956SPaolo Abeni void __tcp_close(struct sock *sk, long timeout)
27491da177e4SLinus Torvalds {
27501da177e4SLinus Torvalds 	struct sk_buff *skb;
27511da177e4SLinus Torvalds 	int data_was_unread = 0;
275275c2d907SHerbert Xu 	int state;
27531da177e4SLinus Torvalds 
2754e14cadfdSEric Dumazet 	WRITE_ONCE(sk->sk_shutdown, SHUTDOWN_MASK);
27551da177e4SLinus Torvalds 
27561da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN) {
27571da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
27581da177e4SLinus Torvalds 
27591da177e4SLinus Torvalds 		/* Special case. */
27600a5578cfSArnaldo Carvalho de Melo 		inet_csk_listen_stop(sk);
27611da177e4SLinus Torvalds 
27621da177e4SLinus Torvalds 		goto adjudge_to_death;
27631da177e4SLinus Torvalds 	}
27641da177e4SLinus Torvalds 
27651da177e4SLinus Torvalds 	/*  We need to flush the recv. buffs.  We do this only on the
27661da177e4SLinus Torvalds 	 *  descriptor close, not protocol-sourced closes, because the
27671da177e4SLinus Torvalds 	 *  reader process may not have drained the data yet!
27681da177e4SLinus Torvalds 	 */
27691da177e4SLinus Torvalds 	while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
2770e11ecddfSEric Dumazet 		u32 len = TCP_SKB_CB(skb)->end_seq - TCP_SKB_CB(skb)->seq;
2771e11ecddfSEric Dumazet 
2772e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
2773e11ecddfSEric Dumazet 			len--;
27741da177e4SLinus Torvalds 		data_was_unread += len;
27751da177e4SLinus Torvalds 		__kfree_skb(skb);
27761da177e4SLinus Torvalds 	}
27771da177e4SLinus Torvalds 
2778565b7b2dSKonstantin Khorenko 	/* If socket has been already reset (e.g. in tcp_reset()) - kill it. */
2779565b7b2dSKonstantin Khorenko 	if (sk->sk_state == TCP_CLOSE)
2780565b7b2dSKonstantin Khorenko 		goto adjudge_to_death;
2781565b7b2dSKonstantin Khorenko 
278265bb723cSGerrit Renker 	/* As outlined in RFC 2525, section 2.17, we send a RST here because
278365bb723cSGerrit Renker 	 * data was lost. To witness the awful effects of the old behavior of
278465bb723cSGerrit Renker 	 * always doing a FIN, run an older 2.1.x kernel or 2.0.x, start a bulk
278565bb723cSGerrit Renker 	 * GET in an FTP client, suspend the process, wait for the client to
278665bb723cSGerrit Renker 	 * advertise a zero window, then kill -9 the FTP client, wheee...
278765bb723cSGerrit Renker 	 * Note: timeout is always zero in such a case.
27881da177e4SLinus Torvalds 	 */
2789ee995283SPavel Emelyanov 	if (unlikely(tcp_sk(sk)->repair)) {
2790ee995283SPavel Emelyanov 		sk->sk_prot->disconnect(sk, 0);
2791ee995283SPavel Emelyanov 	} else if (data_was_unread) {
27921da177e4SLinus Torvalds 		/* Unread data was tossed, zap the connection. */
27936aef70a8SEric Dumazet 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONCLOSE);
27941da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
2795aa133076SWu Fengguang 		tcp_send_active_reset(sk, sk->sk_allocation);
27961da177e4SLinus Torvalds 	} else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
27971da177e4SLinus Torvalds 		/* Check zero linger _after_ checking for unread data. */
27981da177e4SLinus Torvalds 		sk->sk_prot->disconnect(sk, 0);
27996aef70a8SEric Dumazet 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONDATA);
28001da177e4SLinus Torvalds 	} else if (tcp_close_state(sk)) {
28011da177e4SLinus Torvalds 		/* We FIN if the application ate all the data before
28021da177e4SLinus Torvalds 		 * zapping the connection.
28031da177e4SLinus Torvalds 		 */
28041da177e4SLinus Torvalds 
28051da177e4SLinus Torvalds 		/* RED-PEN. Formally speaking, we have broken TCP state
28061da177e4SLinus Torvalds 		 * machine. State transitions:
28071da177e4SLinus Torvalds 		 *
28081da177e4SLinus Torvalds 		 * TCP_ESTABLISHED -> TCP_FIN_WAIT1
28091da177e4SLinus Torvalds 		 * TCP_SYN_RECV	-> TCP_FIN_WAIT1 (forget it, it's impossible)
28101da177e4SLinus Torvalds 		 * TCP_CLOSE_WAIT -> TCP_LAST_ACK
28111da177e4SLinus Torvalds 		 *
28121da177e4SLinus Torvalds 		 * are legal only when FIN has been sent (i.e. in window),
28131da177e4SLinus Torvalds 		 * rather than queued out of window. Purists blame.
28141da177e4SLinus Torvalds 		 *
28151da177e4SLinus Torvalds 		 * F.e. "RFC state" is ESTABLISHED,
28161da177e4SLinus Torvalds 		 * if Linux state is FIN-WAIT-1, but FIN is still not sent.
28171da177e4SLinus Torvalds 		 *
28181da177e4SLinus Torvalds 		 * The visible declinations are that sometimes
28191da177e4SLinus Torvalds 		 * we enter time-wait state, when it is not required really
28201da177e4SLinus Torvalds 		 * (harmless), do not send active resets, when they are
28211da177e4SLinus Torvalds 		 * required by specs (TCP_ESTABLISHED, TCP_CLOSE_WAIT, when
28221da177e4SLinus Torvalds 		 * they look as CLOSING or LAST_ACK for Linux)
28231da177e4SLinus Torvalds 		 * Probably, I missed some more holelets.
28241da177e4SLinus Torvalds 		 * 						--ANK
28258336886fSJerry Chu 		 * XXX (TFO) - To start off we don't support SYN+ACK+FIN
28268336886fSJerry Chu 		 * in a single packet! (May consider it later but will
28278336886fSJerry Chu 		 * probably need API support or TCP_CORK SYN-ACK until
28288336886fSJerry Chu 		 * data is written and socket is closed.)
28291da177e4SLinus Torvalds 		 */
28301da177e4SLinus Torvalds 		tcp_send_fin(sk);
28311da177e4SLinus Torvalds 	}
28321da177e4SLinus Torvalds 
28331da177e4SLinus Torvalds 	sk_stream_wait_close(sk, timeout);
28341da177e4SLinus Torvalds 
28351da177e4SLinus Torvalds adjudge_to_death:
283675c2d907SHerbert Xu 	state = sk->sk_state;
283775c2d907SHerbert Xu 	sock_hold(sk);
283875c2d907SHerbert Xu 	sock_orphan(sk);
283975c2d907SHerbert Xu 
28401da177e4SLinus Torvalds 	local_bh_disable();
28411da177e4SLinus Torvalds 	bh_lock_sock(sk);
28428873c064SEric Dumazet 	/* remove backlog if any, without releasing ownership. */
28438873c064SEric Dumazet 	__release_sock(sk);
28441da177e4SLinus Torvalds 
284519757cebSEric Dumazet 	this_cpu_inc(tcp_orphan_count);
2846eb4dea58SHerbert Xu 
284775c2d907SHerbert Xu 	/* Have we already been destroyed by a softirq or backlog? */
284875c2d907SHerbert Xu 	if (state != TCP_CLOSE && sk->sk_state == TCP_CLOSE)
284975c2d907SHerbert Xu 		goto out;
28501da177e4SLinus Torvalds 
28511da177e4SLinus Torvalds 	/*	This is a (useful) BSD violating of the RFC. There is a
28521da177e4SLinus Torvalds 	 *	problem with TCP as specified in that the other end could
28531da177e4SLinus Torvalds 	 *	keep a socket open forever with no application left this end.
2854b10bd54cSJesper Juhl 	 *	We use a 1 minute timeout (about the same as BSD) then kill
28551da177e4SLinus Torvalds 	 *	our end. If they send after that then tough - BUT: long enough
28561da177e4SLinus Torvalds 	 *	that we won't make the old 4*rto = almost no time - whoops
28571da177e4SLinus Torvalds 	 *	reset mistake.
28581da177e4SLinus Torvalds 	 *
28591da177e4SLinus Torvalds 	 *	Nope, it was not mistake. It is really desired behaviour
28601da177e4SLinus Torvalds 	 *	f.e. on http servers, when such sockets are useless, but
28611da177e4SLinus Torvalds 	 *	consume significant resources. Let's do it with special
28621da177e4SLinus Torvalds 	 *	linger2	option.					--ANK
28631da177e4SLinus Torvalds 	 */
28641da177e4SLinus Torvalds 
28651da177e4SLinus Torvalds 	if (sk->sk_state == TCP_FIN_WAIT2) {
28661da177e4SLinus Torvalds 		struct tcp_sock *tp = tcp_sk(sk);
28671da177e4SLinus Torvalds 		if (tp->linger2 < 0) {
28681da177e4SLinus Torvalds 			tcp_set_state(sk, TCP_CLOSE);
28691da177e4SLinus Torvalds 			tcp_send_active_reset(sk, GFP_ATOMIC);
287002a1d6e7SEric Dumazet 			__NET_INC_STATS(sock_net(sk),
2871de0744afSPavel Emelyanov 					LINUX_MIB_TCPABORTONLINGER);
28721da177e4SLinus Torvalds 		} else {
2873463c84b9SArnaldo Carvalho de Melo 			const int tmo = tcp_fin_time(sk);
28741da177e4SLinus Torvalds 
28751da177e4SLinus Torvalds 			if (tmo > TCP_TIMEWAIT_LEN) {
287652499afeSDavid S. Miller 				inet_csk_reset_keepalive_timer(sk,
287752499afeSDavid S. Miller 						tmo - TCP_TIMEWAIT_LEN);
28781da177e4SLinus Torvalds 			} else {
28791da177e4SLinus Torvalds 				tcp_time_wait(sk, TCP_FIN_WAIT2, tmo);
28801da177e4SLinus Torvalds 				goto out;
28811da177e4SLinus Torvalds 			}
28821da177e4SLinus Torvalds 		}
28831da177e4SLinus Torvalds 	}
28841da177e4SLinus Torvalds 	if (sk->sk_state != TCP_CLOSE) {
2885efcdbf24SArun Sharma 		if (tcp_check_oom(sk, 0)) {
28861da177e4SLinus Torvalds 			tcp_set_state(sk, TCP_CLOSE);
28871da177e4SLinus Torvalds 			tcp_send_active_reset(sk, GFP_ATOMIC);
288802a1d6e7SEric Dumazet 			__NET_INC_STATS(sock_net(sk),
2889de0744afSPavel Emelyanov 					LINUX_MIB_TCPABORTONMEMORY);
28904ee806d5SDan Streetman 		} else if (!check_net(sock_net(sk))) {
28914ee806d5SDan Streetman 			/* Not possible to send reset; just close */
28924ee806d5SDan Streetman 			tcp_set_state(sk, TCP_CLOSE);
28931da177e4SLinus Torvalds 		}
28941da177e4SLinus Torvalds 	}
28951da177e4SLinus Torvalds 
28968336886fSJerry Chu 	if (sk->sk_state == TCP_CLOSE) {
2897d983ea6fSEric Dumazet 		struct request_sock *req;
2898d983ea6fSEric Dumazet 
2899d983ea6fSEric Dumazet 		req = rcu_dereference_protected(tcp_sk(sk)->fastopen_rsk,
2900d983ea6fSEric Dumazet 						lockdep_sock_is_held(sk));
29018336886fSJerry Chu 		/* We could get here with a non-NULL req if the socket is
29028336886fSJerry Chu 		 * aborted (e.g., closed with unread data) before 3WHS
29038336886fSJerry Chu 		 * finishes.
29048336886fSJerry Chu 		 */
290500db4124SIan Morris 		if (req)
29068336886fSJerry Chu 			reqsk_fastopen_remove(sk, req, false);
29070a5578cfSArnaldo Carvalho de Melo 		inet_csk_destroy_sock(sk);
29088336886fSJerry Chu 	}
29091da177e4SLinus Torvalds 	/* Otherwise, socket is reprieved until protocol close. */
29101da177e4SLinus Torvalds 
29111da177e4SLinus Torvalds out:
29121da177e4SLinus Torvalds 	bh_unlock_sock(sk);
29131da177e4SLinus Torvalds 	local_bh_enable();
291477c3c956SPaolo Abeni }
291577c3c956SPaolo Abeni 
291677c3c956SPaolo Abeni void tcp_close(struct sock *sk, long timeout)
291777c3c956SPaolo Abeni {
291877c3c956SPaolo Abeni 	lock_sock(sk);
291977c3c956SPaolo Abeni 	__tcp_close(sk, timeout);
29208873c064SEric Dumazet 	release_sock(sk);
29211da177e4SLinus Torvalds 	sock_put(sk);
29221da177e4SLinus Torvalds }
29234bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_close);
29241da177e4SLinus Torvalds 
29251da177e4SLinus Torvalds /* These states need RST on ABORT according to RFC793 */
29261da177e4SLinus Torvalds 
2927a2a385d6SEric Dumazet static inline bool tcp_need_reset(int state)
29281da177e4SLinus Torvalds {
29291da177e4SLinus Torvalds 	return (1 << state) &
29301da177e4SLinus Torvalds 	       (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT | TCPF_FIN_WAIT1 |
2931a7150e38SEric Dumazet 		TCPF_FIN_WAIT2 | TCPF_SYN_RECV);
29321da177e4SLinus Torvalds }
29331da177e4SLinus Torvalds 
293475c119afSEric Dumazet static void tcp_rtx_queue_purge(struct sock *sk)
293575c119afSEric Dumazet {
293675c119afSEric Dumazet 	struct rb_node *p = rb_first(&sk->tcp_rtx_queue);
293775c119afSEric Dumazet 
29382bec445fSEric Dumazet 	tcp_sk(sk)->highest_sack = NULL;
293975c119afSEric Dumazet 	while (p) {
294075c119afSEric Dumazet 		struct sk_buff *skb = rb_to_skb(p);
294175c119afSEric Dumazet 
294275c119afSEric Dumazet 		p = rb_next(p);
294375c119afSEric Dumazet 		/* Since we are deleting whole queue, no need to
294475c119afSEric Dumazet 		 * list_del(&skb->tcp_tsorted_anchor)
294575c119afSEric Dumazet 		 */
294675c119afSEric Dumazet 		tcp_rtx_queue_unlink(skb, sk);
294703271f3aSTalal Ahmad 		tcp_wmem_free_skb(sk, skb);
294875c119afSEric Dumazet 	}
294975c119afSEric Dumazet }
295075c119afSEric Dumazet 
2951ac3f09baSEric Dumazet void tcp_write_queue_purge(struct sock *sk)
2952ac3f09baSEric Dumazet {
2953ac3f09baSEric Dumazet 	struct sk_buff *skb;
2954ac3f09baSEric Dumazet 
2955ac3f09baSEric Dumazet 	tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
2956ac3f09baSEric Dumazet 	while ((skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) {
2957ac3f09baSEric Dumazet 		tcp_skb_tsorted_anchor_cleanup(skb);
295803271f3aSTalal Ahmad 		tcp_wmem_free_skb(sk, skb);
2959ac3f09baSEric Dumazet 	}
296075c119afSEric Dumazet 	tcp_rtx_queue_purge(sk);
2961ac3f09baSEric Dumazet 	INIT_LIST_HEAD(&tcp_sk(sk)->tsorted_sent_queue);
2962ac3f09baSEric Dumazet 	tcp_clear_all_retrans_hints(tcp_sk(sk));
2963bffd168cSSoheil Hassas Yeganeh 	tcp_sk(sk)->packets_out = 0;
296404c03114SEric Dumazet 	inet_csk(sk)->icsk_backoff = 0;
2965ac3f09baSEric Dumazet }
2966ac3f09baSEric Dumazet 
29671da177e4SLinus Torvalds int tcp_disconnect(struct sock *sk, int flags)
29681da177e4SLinus Torvalds {
29691da177e4SLinus Torvalds 	struct inet_sock *inet = inet_sk(sk);
2970463c84b9SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
29711da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
29721da177e4SLinus Torvalds 	int old_state = sk->sk_state;
29730f317464SEric Dumazet 	u32 seq;
29741da177e4SLinus Torvalds 
29754faeee0cSEric Dumazet 	/* Deny disconnect if other threads are blocked in sk_wait_event()
29764faeee0cSEric Dumazet 	 * or inet_wait_for_connect().
29774faeee0cSEric Dumazet 	 */
29784faeee0cSEric Dumazet 	if (sk->sk_wait_pending)
29794faeee0cSEric Dumazet 		return -EBUSY;
29804faeee0cSEric Dumazet 
29811da177e4SLinus Torvalds 	if (old_state != TCP_CLOSE)
29821da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
29831da177e4SLinus Torvalds 
29841da177e4SLinus Torvalds 	/* ABORT function of RFC793 */
29851da177e4SLinus Torvalds 	if (old_state == TCP_LISTEN) {
29860a5578cfSArnaldo Carvalho de Melo 		inet_csk_listen_stop(sk);
2987ee995283SPavel Emelyanov 	} else if (unlikely(tp->repair)) {
2988e13ec3daSEric Dumazet 		WRITE_ONCE(sk->sk_err, ECONNABORTED);
29891da177e4SLinus Torvalds 	} else if (tcp_need_reset(old_state) ||
29901da177e4SLinus Torvalds 		   (tp->snd_nxt != tp->write_seq &&
29911da177e4SLinus Torvalds 		    (1 << old_state) & (TCPF_CLOSING | TCPF_LAST_ACK))) {
2992caa20d9aSStephen Hemminger 		/* The last check adjusts for discrepancy of Linux wrt. RFC
29931da177e4SLinus Torvalds 		 * states
29941da177e4SLinus Torvalds 		 */
29951da177e4SLinus Torvalds 		tcp_send_active_reset(sk, gfp_any());
2996e13ec3daSEric Dumazet 		WRITE_ONCE(sk->sk_err, ECONNRESET);
2997a7150e38SEric Dumazet 	} else if (old_state == TCP_SYN_SENT)
2998e13ec3daSEric Dumazet 		WRITE_ONCE(sk->sk_err, ECONNRESET);
29991da177e4SLinus Torvalds 
30001da177e4SLinus Torvalds 	tcp_clear_xmit_timers(sk);
30011da177e4SLinus Torvalds 	__skb_queue_purge(&sk->sk_receive_queue);
30027db48e98SEric Dumazet 	WRITE_ONCE(tp->copied_seq, tp->rcv_nxt);
30037b6a893aSEric Dumazet 	WRITE_ONCE(tp->urg_data, 0);
3004fe067e8aSDavid S. Miller 	tcp_write_queue_purge(sk);
3005cf1ef3f0SWei Wang 	tcp_fastopen_active_disable_ofo_check(sk);
30069f5afeaeSYaogong Wang 	skb_rbtree_purge(&tp->out_of_order_queue);
30071da177e4SLinus Torvalds 
3008c720c7e8SEric Dumazet 	inet->inet_dport = 0;
30091da177e4SLinus Torvalds 
3010e0833d1fSKuniyuki Iwashima 	inet_bhash2_reset_saddr(sk);
30111da177e4SLinus Torvalds 
3012e14cadfdSEric Dumazet 	WRITE_ONCE(sk->sk_shutdown, 0);
30131da177e4SLinus Torvalds 	sock_reset_flag(sk, SOCK_DONE);
3014740b0f18SEric Dumazet 	tp->srtt_us = 0;
3015b9e2e689SEric Dumazet 	tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT);
30163f6c65d6SWei Wang 	tp->rcv_rtt_last_tsecr = 0;
30170f317464SEric Dumazet 
30180f317464SEric Dumazet 	seq = tp->write_seq + tp->max_window + 2;
30190f317464SEric Dumazet 	if (!seq)
30200f317464SEric Dumazet 		seq = 1;
30210f317464SEric Dumazet 	WRITE_ONCE(tp->write_seq, seq);
30220f317464SEric Dumazet 
3023463c84b9SArnaldo Carvalho de Melo 	icsk->icsk_backoff = 0;
30246687e988SArnaldo Carvalho de Melo 	icsk->icsk_probes_out = 0;
30259d9b1ee0SEnke Chen 	icsk->icsk_probes_tstamp = 0;
30266a408147SEric Dumazet 	icsk->icsk_rto = TCP_TIMEOUT_INIT;
3027ca584ba0SMartin KaFai Lau 	icsk->icsk_rto_min = TCP_RTO_MIN;
30282b8ee4f0SMartin KaFai Lau 	icsk->icsk_delack_max = TCP_DELACK_MAX;
30290b6a05c1SIlpo Järvinen 	tp->snd_ssthresh = TCP_INFINITE_SSTHRESH;
303040570375SEric Dumazet 	tcp_snd_cwnd_set(tp, TCP_INIT_CWND);
30311da177e4SLinus Torvalds 	tp->snd_cwnd_cnt = 0;
3032f4ce91ceSNeal Cardwell 	tp->is_cwnd_limited = 0;
3033f4ce91ceSNeal Cardwell 	tp->max_packets_out = 0;
30341fdf475aSEric Dumazet 	tp->window_clamp = 0;
30352fbdd562SEric Dumazet 	tp->delivered = 0;
3036e21db6f6SYuchung Cheng 	tp->delivered_ce = 0;
3037ce69e563SChristoph Paasch 	if (icsk->icsk_ca_ops->release)
3038ce69e563SChristoph Paasch 		icsk->icsk_ca_ops->release(sk);
3039ce69e563SChristoph Paasch 	memset(icsk->icsk_ca_priv, 0, sizeof(icsk->icsk_ca_priv));
30408919a9b3SNeal Cardwell 	icsk->icsk_ca_initialized = 0;
30416687e988SArnaldo Carvalho de Melo 	tcp_set_ca_state(sk, TCP_CA_Open);
3042d4761754SYousuk Seung 	tp->is_sack_reneg = 0;
30431da177e4SLinus Torvalds 	tcp_clear_retrans(tp);
3044c13c48c0SEric Dumazet 	tp->total_retrans = 0;
3045463c84b9SArnaldo Carvalho de Melo 	inet_csk_delack_init(sk);
3046499350a5SWei Wang 	/* Initialize rcv_mss to TCP_MIN_MSS to avoid division by 0
3047499350a5SWei Wang 	 * issue in __tcp_select_window()
3048499350a5SWei Wang 	 */
3049499350a5SWei Wang 	icsk->icsk_ack.rcv_mss = TCP_MIN_MSS;
3050b40b4f79SSrinivas Aji 	memset(&tp->rx_opt, 0, sizeof(tp->rx_opt));
30511da177e4SLinus Torvalds 	__sk_dst_reset(sk);
30528f905c0eSEric Dumazet 	dst_release(xchg((__force struct dst_entry **)&sk->sk_rx_dst, NULL));
305317c3060bSEric Dumazet 	tcp_saved_syn_free(tp);
30545d9f4262SEric Dumazet 	tp->compressed_ack = 0;
3055784f8344SEric Dumazet 	tp->segs_in = 0;
3056784f8344SEric Dumazet 	tp->segs_out = 0;
3057ba113c3aSWei Wang 	tp->bytes_sent = 0;
3058e858faf5SChristoph Paasch 	tp->bytes_acked = 0;
3059e858faf5SChristoph Paasch 	tp->bytes_received = 0;
3060fb31c9b9SWei Wang 	tp->bytes_retrans = 0;
3061db7ffee6SEric Dumazet 	tp->data_segs_in = 0;
3062db7ffee6SEric Dumazet 	tp->data_segs_out = 0;
30637788174eSYuchung Cheng 	tp->duplicate_sack[0].start_seq = 0;
30647788174eSYuchung Cheng 	tp->duplicate_sack[0].end_seq = 0;
30657e10b655SWei Wang 	tp->dsack_dups = 0;
30667ec65372SWei Wang 	tp->reord_seen = 0;
30675c701549SEric Dumazet 	tp->retrans_out = 0;
30685c701549SEric Dumazet 	tp->sacked_out = 0;
30695c701549SEric Dumazet 	tp->tlp_high_seq = 0;
30705c701549SEric Dumazet 	tp->last_oow_ack_time = 0;
307129c1c446SMubashir Adnan Qureshi 	tp->plb_rehash = 0;
30726cda8b74SEric Dumazet 	/* There's a bubble in the pipe until at least the first ACK. */
30736cda8b74SEric Dumazet 	tp->app_limited = ~0U;
3074300b655dSDavid Morley 	tp->rate_app_limited = 1;
3075792c4354SEric Dumazet 	tp->rack.mstamp = 0;
3076792c4354SEric Dumazet 	tp->rack.advanced = 0;
3077792c4354SEric Dumazet 	tp->rack.reo_wnd_steps = 1;
3078792c4354SEric Dumazet 	tp->rack.last_delivered = 0;
3079792c4354SEric Dumazet 	tp->rack.reo_wnd_persist = 0;
3080792c4354SEric Dumazet 	tp->rack.dsack_seen = 0;
30816bcdc40dSEric Dumazet 	tp->syn_data_acked = 0;
30826bcdc40dSEric Dumazet 	tp->rx_opt.saw_tstamp = 0;
30836bcdc40dSEric Dumazet 	tp->rx_opt.dsack = 0;
30846bcdc40dSEric Dumazet 	tp->rx_opt.num_sacks = 0;
3085f9af2dbbSThomas Higdon 	tp->rcv_ooopack = 0;
30866cda8b74SEric Dumazet 
30871da177e4SLinus Torvalds 
30887db92362SWei Wang 	/* Clean up fastopen related fields */
30897db92362SWei Wang 	tcp_free_fastopen_req(tp);
30907db92362SWei Wang 	inet->defer_connect = 0;
309148027478SJason Baron 	tp->fastopen_client_fail = 0;
30927db92362SWei Wang 
3093c720c7e8SEric Dumazet 	WARN_ON(inet->inet_num && !icsk->icsk_bind_hash);
30941da177e4SLinus Torvalds 
30959b42d55aSLi RongQing 	if (sk->sk_frag.page) {
30969b42d55aSLi RongQing 		put_page(sk->sk_frag.page);
30979b42d55aSLi RongQing 		sk->sk_frag.page = NULL;
30989b42d55aSLi RongQing 		sk->sk_frag.offset = 0;
30999b42d55aSLi RongQing 	}
3100e3ae2365SAlexander Aring 	sk_error_report(sk);
3101a01512b1SYueHaibing 	return 0;
31021da177e4SLinus Torvalds }
31034bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_disconnect);
31041da177e4SLinus Torvalds 
3105a2a385d6SEric Dumazet static inline bool tcp_can_repair_sock(const struct sock *sk)
3106ee995283SPavel Emelyanov {
3107cb388e7eSMartin KaFai Lau 	return sockopt_ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN) &&
3108319b0534SAndrey Vagin 		(sk->sk_state != TCP_LISTEN);
3109ee995283SPavel Emelyanov }
3110ee995283SPavel Emelyanov 
3111d38d2b00SChristoph Hellwig static int tcp_repair_set_window(struct tcp_sock *tp, sockptr_t optbuf, int len)
3112b1ed4c4fSAndrey Vagin {
3113b1ed4c4fSAndrey Vagin 	struct tcp_repair_window opt;
3114b1ed4c4fSAndrey Vagin 
3115b1ed4c4fSAndrey Vagin 	if (!tp->repair)
3116b1ed4c4fSAndrey Vagin 		return -EPERM;
3117b1ed4c4fSAndrey Vagin 
3118b1ed4c4fSAndrey Vagin 	if (len != sizeof(opt))
3119b1ed4c4fSAndrey Vagin 		return -EINVAL;
3120b1ed4c4fSAndrey Vagin 
3121d38d2b00SChristoph Hellwig 	if (copy_from_sockptr(&opt, optbuf, sizeof(opt)))
3122b1ed4c4fSAndrey Vagin 		return -EFAULT;
3123b1ed4c4fSAndrey Vagin 
3124b1ed4c4fSAndrey Vagin 	if (opt.max_window < opt.snd_wnd)
3125b1ed4c4fSAndrey Vagin 		return -EINVAL;
3126b1ed4c4fSAndrey Vagin 
3127b1ed4c4fSAndrey Vagin 	if (after(opt.snd_wl1, tp->rcv_nxt + opt.rcv_wnd))
3128b1ed4c4fSAndrey Vagin 		return -EINVAL;
3129b1ed4c4fSAndrey Vagin 
3130b1ed4c4fSAndrey Vagin 	if (after(opt.rcv_wup, tp->rcv_nxt))
3131b1ed4c4fSAndrey Vagin 		return -EINVAL;
3132b1ed4c4fSAndrey Vagin 
3133b1ed4c4fSAndrey Vagin 	tp->snd_wl1	= opt.snd_wl1;
3134b1ed4c4fSAndrey Vagin 	tp->snd_wnd	= opt.snd_wnd;
3135b1ed4c4fSAndrey Vagin 	tp->max_window	= opt.max_window;
3136b1ed4c4fSAndrey Vagin 
3137b1ed4c4fSAndrey Vagin 	tp->rcv_wnd	= opt.rcv_wnd;
3138b1ed4c4fSAndrey Vagin 	tp->rcv_wup	= opt.rcv_wup;
3139b1ed4c4fSAndrey Vagin 
3140b1ed4c4fSAndrey Vagin 	return 0;
3141b1ed4c4fSAndrey Vagin }
3142b1ed4c4fSAndrey Vagin 
3143d38d2b00SChristoph Hellwig static int tcp_repair_options_est(struct sock *sk, sockptr_t optbuf,
3144d38d2b00SChristoph Hellwig 		unsigned int len)
3145b139ba4eSPavel Emelyanov {
314615e56515SDouglas Caetano dos Santos 	struct tcp_sock *tp = tcp_sk(sk);
3147de248a75SPavel Emelyanov 	struct tcp_repair_opt opt;
3148d3c48151SChristoph Hellwig 	size_t offset = 0;
3149b139ba4eSPavel Emelyanov 
3150de248a75SPavel Emelyanov 	while (len >= sizeof(opt)) {
3151d3c48151SChristoph Hellwig 		if (copy_from_sockptr_offset(&opt, optbuf, offset, sizeof(opt)))
3152b139ba4eSPavel Emelyanov 			return -EFAULT;
3153b139ba4eSPavel Emelyanov 
3154d3c48151SChristoph Hellwig 		offset += sizeof(opt);
3155de248a75SPavel Emelyanov 		len -= sizeof(opt);
3156b139ba4eSPavel Emelyanov 
3157de248a75SPavel Emelyanov 		switch (opt.opt_code) {
3158de248a75SPavel Emelyanov 		case TCPOPT_MSS:
3159de248a75SPavel Emelyanov 			tp->rx_opt.mss_clamp = opt.opt_val;
316015e56515SDouglas Caetano dos Santos 			tcp_mtup_init(sk);
3161b139ba4eSPavel Emelyanov 			break;
3162de248a75SPavel Emelyanov 		case TCPOPT_WINDOW:
3163bc26ccd8SAndrey Vagin 			{
3164bc26ccd8SAndrey Vagin 				u16 snd_wscale = opt.opt_val & 0xFFFF;
3165bc26ccd8SAndrey Vagin 				u16 rcv_wscale = opt.opt_val >> 16;
3166bc26ccd8SAndrey Vagin 
3167589c49cbSGao Feng 				if (snd_wscale > TCP_MAX_WSCALE || rcv_wscale > TCP_MAX_WSCALE)
3168b139ba4eSPavel Emelyanov 					return -EFBIG;
3169b139ba4eSPavel Emelyanov 
3170bc26ccd8SAndrey Vagin 				tp->rx_opt.snd_wscale = snd_wscale;
3171bc26ccd8SAndrey Vagin 				tp->rx_opt.rcv_wscale = rcv_wscale;
3172bc26ccd8SAndrey Vagin 				tp->rx_opt.wscale_ok = 1;
3173bc26ccd8SAndrey Vagin 			}
3174b139ba4eSPavel Emelyanov 			break;
3175b139ba4eSPavel Emelyanov 		case TCPOPT_SACK_PERM:
3176de248a75SPavel Emelyanov 			if (opt.opt_val != 0)
3177de248a75SPavel Emelyanov 				return -EINVAL;
3178de248a75SPavel Emelyanov 
3179b139ba4eSPavel Emelyanov 			tp->rx_opt.sack_ok |= TCP_SACK_SEEN;
3180b139ba4eSPavel Emelyanov 			break;
3181b139ba4eSPavel Emelyanov 		case TCPOPT_TIMESTAMP:
3182de248a75SPavel Emelyanov 			if (opt.opt_val != 0)
3183de248a75SPavel Emelyanov 				return -EINVAL;
3184de248a75SPavel Emelyanov 
3185b139ba4eSPavel Emelyanov 			tp->rx_opt.tstamp_ok = 1;
3186b139ba4eSPavel Emelyanov 			break;
3187b139ba4eSPavel Emelyanov 		}
3188b139ba4eSPavel Emelyanov 	}
3189b139ba4eSPavel Emelyanov 
3190b139ba4eSPavel Emelyanov 	return 0;
3191b139ba4eSPavel Emelyanov }
3192b139ba4eSPavel Emelyanov 
3193a842fe14SEric Dumazet DEFINE_STATIC_KEY_FALSE(tcp_tx_delay_enabled);
3194a842fe14SEric Dumazet EXPORT_SYMBOL(tcp_tx_delay_enabled);
3195a842fe14SEric Dumazet 
3196a842fe14SEric Dumazet static void tcp_enable_tx_delay(void)
3197a842fe14SEric Dumazet {
3198a842fe14SEric Dumazet 	if (!static_branch_unlikely(&tcp_tx_delay_enabled)) {
3199a842fe14SEric Dumazet 		static int __tcp_tx_delay_enabled = 0;
3200a842fe14SEric Dumazet 
3201a842fe14SEric Dumazet 		if (cmpxchg(&__tcp_tx_delay_enabled, 0, 1) == 0) {
3202a842fe14SEric Dumazet 			static_branch_enable(&tcp_tx_delay_enabled);
3203a842fe14SEric Dumazet 			pr_info("TCP_TX_DELAY enabled\n");
3204a842fe14SEric Dumazet 		}
3205a842fe14SEric Dumazet 	}
3206a842fe14SEric Dumazet }
3207a842fe14SEric Dumazet 
3208db10538aSChristoph Hellwig /* When set indicates to always queue non-full frames.  Later the user clears
3209db10538aSChristoph Hellwig  * this option and we transmit any pending partial frames in the queue.  This is
3210db10538aSChristoph Hellwig  * meant to be used alongside sendfile() to get properly filled frames when the
3211db10538aSChristoph Hellwig  * user (for example) must write out headers with a write() call first and then
3212db10538aSChristoph Hellwig  * use sendfile to send out the data parts.
3213db10538aSChristoph Hellwig  *
3214db10538aSChristoph Hellwig  * TCP_CORK can be set together with TCP_NODELAY and it is stronger than
3215db10538aSChristoph Hellwig  * TCP_NODELAY.
3216db10538aSChristoph Hellwig  */
32176fadaa56SMaxim Galaganov void __tcp_sock_set_cork(struct sock *sk, bool on)
3218db10538aSChristoph Hellwig {
3219db10538aSChristoph Hellwig 	struct tcp_sock *tp = tcp_sk(sk);
3220db10538aSChristoph Hellwig 
3221db10538aSChristoph Hellwig 	if (on) {
3222db10538aSChristoph Hellwig 		tp->nonagle |= TCP_NAGLE_CORK;
3223db10538aSChristoph Hellwig 	} else {
3224db10538aSChristoph Hellwig 		tp->nonagle &= ~TCP_NAGLE_CORK;
3225db10538aSChristoph Hellwig 		if (tp->nonagle & TCP_NAGLE_OFF)
3226db10538aSChristoph Hellwig 			tp->nonagle |= TCP_NAGLE_PUSH;
3227db10538aSChristoph Hellwig 		tcp_push_pending_frames(sk);
3228db10538aSChristoph Hellwig 	}
3229db10538aSChristoph Hellwig }
3230db10538aSChristoph Hellwig 
3231db10538aSChristoph Hellwig void tcp_sock_set_cork(struct sock *sk, bool on)
3232db10538aSChristoph Hellwig {
3233db10538aSChristoph Hellwig 	lock_sock(sk);
3234db10538aSChristoph Hellwig 	__tcp_sock_set_cork(sk, on);
3235db10538aSChristoph Hellwig 	release_sock(sk);
3236db10538aSChristoph Hellwig }
3237db10538aSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_cork);
3238db10538aSChristoph Hellwig 
323912abc5eeSChristoph Hellwig /* TCP_NODELAY is weaker than TCP_CORK, so that this option on corked socket is
324012abc5eeSChristoph Hellwig  * remembered, but it is not activated until cork is cleared.
324112abc5eeSChristoph Hellwig  *
324212abc5eeSChristoph Hellwig  * However, when TCP_NODELAY is set we make an explicit push, which overrides
324312abc5eeSChristoph Hellwig  * even TCP_CORK for currently queued segments.
324412abc5eeSChristoph Hellwig  */
32456fadaa56SMaxim Galaganov void __tcp_sock_set_nodelay(struct sock *sk, bool on)
324612abc5eeSChristoph Hellwig {
324712abc5eeSChristoph Hellwig 	if (on) {
324812abc5eeSChristoph Hellwig 		tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF|TCP_NAGLE_PUSH;
324912abc5eeSChristoph Hellwig 		tcp_push_pending_frames(sk);
325012abc5eeSChristoph Hellwig 	} else {
325112abc5eeSChristoph Hellwig 		tcp_sk(sk)->nonagle &= ~TCP_NAGLE_OFF;
325212abc5eeSChristoph Hellwig 	}
325312abc5eeSChristoph Hellwig }
325412abc5eeSChristoph Hellwig 
325512abc5eeSChristoph Hellwig void tcp_sock_set_nodelay(struct sock *sk)
325612abc5eeSChristoph Hellwig {
325712abc5eeSChristoph Hellwig 	lock_sock(sk);
325812abc5eeSChristoph Hellwig 	__tcp_sock_set_nodelay(sk, true);
325912abc5eeSChristoph Hellwig 	release_sock(sk);
326012abc5eeSChristoph Hellwig }
326112abc5eeSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_nodelay);
326212abc5eeSChristoph Hellwig 
3263ddd061b8SChristoph Hellwig static void __tcp_sock_set_quickack(struct sock *sk, int val)
3264ddd061b8SChristoph Hellwig {
3265ddd061b8SChristoph Hellwig 	if (!val) {
3266ddd061b8SChristoph Hellwig 		inet_csk_enter_pingpong_mode(sk);
3267ddd061b8SChristoph Hellwig 		return;
3268ddd061b8SChristoph Hellwig 	}
3269ddd061b8SChristoph Hellwig 
3270ddd061b8SChristoph Hellwig 	inet_csk_exit_pingpong_mode(sk);
3271ddd061b8SChristoph Hellwig 	if ((1 << sk->sk_state) & (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT) &&
3272ddd061b8SChristoph Hellwig 	    inet_csk_ack_scheduled(sk)) {
3273ddd061b8SChristoph Hellwig 		inet_csk(sk)->icsk_ack.pending |= ICSK_ACK_PUSHED;
3274ddd061b8SChristoph Hellwig 		tcp_cleanup_rbuf(sk, 1);
3275ddd061b8SChristoph Hellwig 		if (!(val & 1))
3276ddd061b8SChristoph Hellwig 			inet_csk_enter_pingpong_mode(sk);
3277ddd061b8SChristoph Hellwig 	}
3278ddd061b8SChristoph Hellwig }
3279ddd061b8SChristoph Hellwig 
3280ddd061b8SChristoph Hellwig void tcp_sock_set_quickack(struct sock *sk, int val)
3281ddd061b8SChristoph Hellwig {
3282ddd061b8SChristoph Hellwig 	lock_sock(sk);
3283ddd061b8SChristoph Hellwig 	__tcp_sock_set_quickack(sk, val);
3284ddd061b8SChristoph Hellwig 	release_sock(sk);
3285ddd061b8SChristoph Hellwig }
3286ddd061b8SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_quickack);
3287ddd061b8SChristoph Hellwig 
3288557eadfcSChristoph Hellwig int tcp_sock_set_syncnt(struct sock *sk, int val)
3289557eadfcSChristoph Hellwig {
3290557eadfcSChristoph Hellwig 	if (val < 1 || val > MAX_TCP_SYNCNT)
3291557eadfcSChristoph Hellwig 		return -EINVAL;
3292557eadfcSChristoph Hellwig 
3293557eadfcSChristoph Hellwig 	lock_sock(sk);
3294557eadfcSChristoph Hellwig 	inet_csk(sk)->icsk_syn_retries = val;
3295557eadfcSChristoph Hellwig 	release_sock(sk);
3296557eadfcSChristoph Hellwig 	return 0;
3297557eadfcSChristoph Hellwig }
3298557eadfcSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_syncnt);
3299557eadfcSChristoph Hellwig 
3300c488aeadSChristoph Hellwig void tcp_sock_set_user_timeout(struct sock *sk, u32 val)
3301c488aeadSChristoph Hellwig {
3302c488aeadSChristoph Hellwig 	lock_sock(sk);
3303c488aeadSChristoph Hellwig 	inet_csk(sk)->icsk_user_timeout = val;
3304c488aeadSChristoph Hellwig 	release_sock(sk);
3305c488aeadSChristoph Hellwig }
3306c488aeadSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_user_timeout);
3307c488aeadSChristoph Hellwig 
3308aad4a0a9SDmitry Yakunin int tcp_sock_set_keepidle_locked(struct sock *sk, int val)
330971c48eb8SChristoph Hellwig {
331071c48eb8SChristoph Hellwig 	struct tcp_sock *tp = tcp_sk(sk);
331171c48eb8SChristoph Hellwig 
331271c48eb8SChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPIDLE)
331371c48eb8SChristoph Hellwig 		return -EINVAL;
331471c48eb8SChristoph Hellwig 
33154164245cSEric Dumazet 	/* Paired with WRITE_ONCE() in keepalive_time_when() */
33164164245cSEric Dumazet 	WRITE_ONCE(tp->keepalive_time, val * HZ);
331771c48eb8SChristoph Hellwig 	if (sock_flag(sk, SOCK_KEEPOPEN) &&
331871c48eb8SChristoph Hellwig 	    !((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN))) {
331971c48eb8SChristoph Hellwig 		u32 elapsed = keepalive_time_elapsed(tp);
332071c48eb8SChristoph Hellwig 
332171c48eb8SChristoph Hellwig 		if (tp->keepalive_time > elapsed)
332271c48eb8SChristoph Hellwig 			elapsed = tp->keepalive_time - elapsed;
332371c48eb8SChristoph Hellwig 		else
332471c48eb8SChristoph Hellwig 			elapsed = 0;
332571c48eb8SChristoph Hellwig 		inet_csk_reset_keepalive_timer(sk, elapsed);
332671c48eb8SChristoph Hellwig 	}
332771c48eb8SChristoph Hellwig 
332871c48eb8SChristoph Hellwig 	return 0;
332971c48eb8SChristoph Hellwig }
333071c48eb8SChristoph Hellwig 
333171c48eb8SChristoph Hellwig int tcp_sock_set_keepidle(struct sock *sk, int val)
333271c48eb8SChristoph Hellwig {
333371c48eb8SChristoph Hellwig 	int err;
333471c48eb8SChristoph Hellwig 
333571c48eb8SChristoph Hellwig 	lock_sock(sk);
3336aad4a0a9SDmitry Yakunin 	err = tcp_sock_set_keepidle_locked(sk, val);
333771c48eb8SChristoph Hellwig 	release_sock(sk);
333871c48eb8SChristoph Hellwig 	return err;
333971c48eb8SChristoph Hellwig }
334071c48eb8SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepidle);
334171c48eb8SChristoph Hellwig 
3342d41ecaacSChristoph Hellwig int tcp_sock_set_keepintvl(struct sock *sk, int val)
3343d41ecaacSChristoph Hellwig {
3344d41ecaacSChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPINTVL)
3345d41ecaacSChristoph Hellwig 		return -EINVAL;
3346d41ecaacSChristoph Hellwig 
3347d41ecaacSChristoph Hellwig 	lock_sock(sk);
33485ecf9d4fSEric Dumazet 	WRITE_ONCE(tcp_sk(sk)->keepalive_intvl, val * HZ);
3349d41ecaacSChristoph Hellwig 	release_sock(sk);
3350d41ecaacSChristoph Hellwig 	return 0;
3351d41ecaacSChristoph Hellwig }
3352d41ecaacSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepintvl);
3353d41ecaacSChristoph Hellwig 
3354480aeb96SChristoph Hellwig int tcp_sock_set_keepcnt(struct sock *sk, int val)
3355480aeb96SChristoph Hellwig {
3356480aeb96SChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPCNT)
3357480aeb96SChristoph Hellwig 		return -EINVAL;
3358480aeb96SChristoph Hellwig 
3359480aeb96SChristoph Hellwig 	lock_sock(sk);
3360*6e5e1de6SEric Dumazet 	/* Paired with READ_ONCE() in keepalive_probes() */
3361*6e5e1de6SEric Dumazet 	WRITE_ONCE(tcp_sk(sk)->keepalive_probes, val);
3362480aeb96SChristoph Hellwig 	release_sock(sk);
3363480aeb96SChristoph Hellwig 	return 0;
3364480aeb96SChristoph Hellwig }
3365480aeb96SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepcnt);
3366480aeb96SChristoph Hellwig 
3367cb811109SPrankur gupta int tcp_set_window_clamp(struct sock *sk, int val)
3368cb811109SPrankur gupta {
3369cb811109SPrankur gupta 	struct tcp_sock *tp = tcp_sk(sk);
3370cb811109SPrankur gupta 
3371cb811109SPrankur gupta 	if (!val) {
3372cb811109SPrankur gupta 		if (sk->sk_state != TCP_CLOSE)
3373cb811109SPrankur gupta 			return -EINVAL;
3374cb811109SPrankur gupta 		tp->window_clamp = 0;
3375cb811109SPrankur gupta 	} else {
3376cb811109SPrankur gupta 		tp->window_clamp = val < SOCK_MIN_RCVBUF / 2 ?
3377cb811109SPrankur gupta 			SOCK_MIN_RCVBUF / 2 : val;
33783aa7857fSNeil Spring 		tp->rcv_ssthresh = min(tp->rcv_wnd, tp->window_clamp);
3379cb811109SPrankur gupta 	}
3380cb811109SPrankur gupta 	return 0;
3381cb811109SPrankur gupta }
3382cb811109SPrankur gupta 
33831da177e4SLinus Torvalds /*
33841da177e4SLinus Torvalds  *	Socket option code for TCP.
33851da177e4SLinus Torvalds  */
33860c751f70SMartin KaFai Lau int do_tcp_setsockopt(struct sock *sk, int level, int optname,
3387d38d2b00SChristoph Hellwig 		      sockptr_t optval, unsigned int optlen)
33881da177e4SLinus Torvalds {
33891da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
3390463c84b9SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
33911e579caaSNikolay Borisov 	struct net *net = sock_net(sk);
33921da177e4SLinus Torvalds 	int val;
33931da177e4SLinus Torvalds 	int err = 0;
33941da177e4SLinus Torvalds 
3395e56fb50fSWilliam Allen Simpson 	/* These are data/string values, all the others are ints */
3396e56fb50fSWilliam Allen Simpson 	switch (optname) {
3397e56fb50fSWilliam Allen Simpson 	case TCP_CONGESTION: {
33985f8ef48dSStephen Hemminger 		char name[TCP_CA_NAME_MAX];
33995f8ef48dSStephen Hemminger 
34005f8ef48dSStephen Hemminger 		if (optlen < 1)
34015f8ef48dSStephen Hemminger 			return -EINVAL;
34025f8ef48dSStephen Hemminger 
3403d38d2b00SChristoph Hellwig 		val = strncpy_from_sockptr(name, optval,
34044fdb78d3SAndrew Morton 					min_t(long, TCP_CA_NAME_MAX-1, optlen));
34055f8ef48dSStephen Hemminger 		if (val < 0)
34065f8ef48dSStephen Hemminger 			return -EFAULT;
34075f8ef48dSStephen Hemminger 		name[val] = 0;
34085f8ef48dSStephen Hemminger 
3409cb388e7eSMartin KaFai Lau 		sockopt_lock_sock(sk);
341084e5a0f2SMartin KaFai Lau 		err = tcp_set_congestion_control(sk, name, !has_current_bpf_ctx(),
3411cb388e7eSMartin KaFai Lau 						 sockopt_ns_capable(sock_net(sk)->user_ns,
34128d650cdeSEric Dumazet 								    CAP_NET_ADMIN));
3413cb388e7eSMartin KaFai Lau 		sockopt_release_sock(sk);
34145f8ef48dSStephen Hemminger 		return err;
34155f8ef48dSStephen Hemminger 	}
3416734942ccSDave Watson 	case TCP_ULP: {
3417734942ccSDave Watson 		char name[TCP_ULP_NAME_MAX];
3418734942ccSDave Watson 
3419734942ccSDave Watson 		if (optlen < 1)
3420734942ccSDave Watson 			return -EINVAL;
3421734942ccSDave Watson 
3422d38d2b00SChristoph Hellwig 		val = strncpy_from_sockptr(name, optval,
3423734942ccSDave Watson 					min_t(long, TCP_ULP_NAME_MAX - 1,
3424734942ccSDave Watson 					      optlen));
3425734942ccSDave Watson 		if (val < 0)
3426734942ccSDave Watson 			return -EFAULT;
3427734942ccSDave Watson 		name[val] = 0;
3428734942ccSDave Watson 
3429cb388e7eSMartin KaFai Lau 		sockopt_lock_sock(sk);
3430734942ccSDave Watson 		err = tcp_set_ulp(sk, name);
3431cb388e7eSMartin KaFai Lau 		sockopt_release_sock(sk);
3432734942ccSDave Watson 		return err;
3433734942ccSDave Watson 	}
34341fba70e5SYuchung Cheng 	case TCP_FASTOPEN_KEY: {
34350f1ce023SJason Baron 		__u8 key[TCP_FASTOPEN_KEY_BUF_LENGTH];
34360f1ce023SJason Baron 		__u8 *backup_key = NULL;
34371fba70e5SYuchung Cheng 
34380f1ce023SJason Baron 		/* Allow a backup key as well to facilitate key rotation
34390f1ce023SJason Baron 		 * First key is the active one.
34400f1ce023SJason Baron 		 */
34410f1ce023SJason Baron 		if (optlen != TCP_FASTOPEN_KEY_LENGTH &&
34420f1ce023SJason Baron 		    optlen != TCP_FASTOPEN_KEY_BUF_LENGTH)
34431fba70e5SYuchung Cheng 			return -EINVAL;
34441fba70e5SYuchung Cheng 
3445d38d2b00SChristoph Hellwig 		if (copy_from_sockptr(key, optval, optlen))
34461fba70e5SYuchung Cheng 			return -EFAULT;
34471fba70e5SYuchung Cheng 
34480f1ce023SJason Baron 		if (optlen == TCP_FASTOPEN_KEY_BUF_LENGTH)
34490f1ce023SJason Baron 			backup_key = key + TCP_FASTOPEN_KEY_LENGTH;
34500f1ce023SJason Baron 
3451438ac880SArd Biesheuvel 		return tcp_fastopen_reset_cipher(net, sk, key, backup_key);
34521fba70e5SYuchung Cheng 	}
3453e56fb50fSWilliam Allen Simpson 	default:
3454e56fb50fSWilliam Allen Simpson 		/* fallthru */
3455e56fb50fSWilliam Allen Simpson 		break;
3456ccbd6a5aSJoe Perches 	}
34575f8ef48dSStephen Hemminger 
34581da177e4SLinus Torvalds 	if (optlen < sizeof(int))
34591da177e4SLinus Torvalds 		return -EINVAL;
34601da177e4SLinus Torvalds 
3461d38d2b00SChristoph Hellwig 	if (copy_from_sockptr(&val, optval, sizeof(val)))
34621da177e4SLinus Torvalds 		return -EFAULT;
34631da177e4SLinus Torvalds 
3464cb388e7eSMartin KaFai Lau 	sockopt_lock_sock(sk);
34651da177e4SLinus Torvalds 
34661da177e4SLinus Torvalds 	switch (optname) {
34671da177e4SLinus Torvalds 	case TCP_MAXSEG:
34681da177e4SLinus Torvalds 		/* Values greater than interface MTU won't take effect. However
34691da177e4SLinus Torvalds 		 * at the point when this call is done we typically don't yet
3470a777f715SRohit Chavan 		 * know which interface is going to be used
3471a777f715SRohit Chavan 		 */
3472cfc62d87SGao Feng 		if (val && (val < TCP_MIN_MSS || val > MAX_TCP_WINDOW)) {
34731da177e4SLinus Torvalds 			err = -EINVAL;
34741da177e4SLinus Torvalds 			break;
34751da177e4SLinus Torvalds 		}
34761da177e4SLinus Torvalds 		tp->rx_opt.user_mss = val;
34771da177e4SLinus Torvalds 		break;
34781da177e4SLinus Torvalds 
34791da177e4SLinus Torvalds 	case TCP_NODELAY:
348012abc5eeSChristoph Hellwig 		__tcp_sock_set_nodelay(sk, val);
34811da177e4SLinus Torvalds 		break;
34821da177e4SLinus Torvalds 
348336e31b0aSAndreas Petlund 	case TCP_THIN_LINEAR_TIMEOUTS:
348436e31b0aSAndreas Petlund 		if (val < 0 || val > 1)
348536e31b0aSAndreas Petlund 			err = -EINVAL;
348636e31b0aSAndreas Petlund 		else
348736e31b0aSAndreas Petlund 			tp->thin_lto = val;
348836e31b0aSAndreas Petlund 		break;
348936e31b0aSAndreas Petlund 
34907e380175SAndreas Petlund 	case TCP_THIN_DUPACK:
34917e380175SAndreas Petlund 		if (val < 0 || val > 1)
34927e380175SAndreas Petlund 			err = -EINVAL;
34937e380175SAndreas Petlund 		break;
34947e380175SAndreas Petlund 
3495ee995283SPavel Emelyanov 	case TCP_REPAIR:
3496ee995283SPavel Emelyanov 		if (!tcp_can_repair_sock(sk))
3497ee995283SPavel Emelyanov 			err = -EPERM;
349831048d7aSStefan Baranoff 		else if (val == TCP_REPAIR_ON) {
3499ee995283SPavel Emelyanov 			tp->repair = 1;
3500ee995283SPavel Emelyanov 			sk->sk_reuse = SK_FORCE_REUSE;
3501ee995283SPavel Emelyanov 			tp->repair_queue = TCP_NO_QUEUE;
350231048d7aSStefan Baranoff 		} else if (val == TCP_REPAIR_OFF) {
3503ee995283SPavel Emelyanov 			tp->repair = 0;
3504ee995283SPavel Emelyanov 			sk->sk_reuse = SK_NO_REUSE;
3505ee995283SPavel Emelyanov 			tcp_send_window_probe(sk);
350631048d7aSStefan Baranoff 		} else if (val == TCP_REPAIR_OFF_NO_WP) {
350731048d7aSStefan Baranoff 			tp->repair = 0;
350831048d7aSStefan Baranoff 			sk->sk_reuse = SK_NO_REUSE;
3509ee995283SPavel Emelyanov 		} else
3510ee995283SPavel Emelyanov 			err = -EINVAL;
3511ee995283SPavel Emelyanov 
3512ee995283SPavel Emelyanov 		break;
3513ee995283SPavel Emelyanov 
3514ee995283SPavel Emelyanov 	case TCP_REPAIR_QUEUE:
3515ee995283SPavel Emelyanov 		if (!tp->repair)
3516ee995283SPavel Emelyanov 			err = -EPERM;
3517bf2acc94SEric Dumazet 		else if ((unsigned int)val < TCP_QUEUES_NR)
3518ee995283SPavel Emelyanov 			tp->repair_queue = val;
3519ee995283SPavel Emelyanov 		else
3520ee995283SPavel Emelyanov 			err = -EINVAL;
3521ee995283SPavel Emelyanov 		break;
3522ee995283SPavel Emelyanov 
3523ee995283SPavel Emelyanov 	case TCP_QUEUE_SEQ:
35248811f4a9SEric Dumazet 		if (sk->sk_state != TCP_CLOSE) {
3525ee995283SPavel Emelyanov 			err = -EPERM;
35268811f4a9SEric Dumazet 		} else if (tp->repair_queue == TCP_SEND_QUEUE) {
35278811f4a9SEric Dumazet 			if (!tcp_rtx_queue_empty(sk))
35288811f4a9SEric Dumazet 				err = -EPERM;
35298811f4a9SEric Dumazet 			else
35300f317464SEric Dumazet 				WRITE_ONCE(tp->write_seq, val);
35318811f4a9SEric Dumazet 		} else if (tp->repair_queue == TCP_RECV_QUEUE) {
35328811f4a9SEric Dumazet 			if (tp->rcv_nxt != tp->copied_seq) {
35338811f4a9SEric Dumazet 				err = -EPERM;
35348811f4a9SEric Dumazet 			} else {
3535dba7d9b8SEric Dumazet 				WRITE_ONCE(tp->rcv_nxt, val);
35366cd6cbf5SEric Dumazet 				WRITE_ONCE(tp->copied_seq, val);
35376cd6cbf5SEric Dumazet 			}
35388811f4a9SEric Dumazet 		} else {
3539ee995283SPavel Emelyanov 			err = -EINVAL;
35408811f4a9SEric Dumazet 		}
3541ee995283SPavel Emelyanov 		break;
3542ee995283SPavel Emelyanov 
3543b139ba4eSPavel Emelyanov 	case TCP_REPAIR_OPTIONS:
3544b139ba4eSPavel Emelyanov 		if (!tp->repair)
3545b139ba4eSPavel Emelyanov 			err = -EINVAL;
35460c175da7SLu Wei 		else if (sk->sk_state == TCP_ESTABLISHED && !tp->bytes_sent)
3547d38d2b00SChristoph Hellwig 			err = tcp_repair_options_est(sk, optval, optlen);
3548b139ba4eSPavel Emelyanov 		else
3549b139ba4eSPavel Emelyanov 			err = -EPERM;
3550b139ba4eSPavel Emelyanov 		break;
3551b139ba4eSPavel Emelyanov 
35521da177e4SLinus Torvalds 	case TCP_CORK:
3553db10538aSChristoph Hellwig 		__tcp_sock_set_cork(sk, val);
35541da177e4SLinus Torvalds 		break;
35551da177e4SLinus Torvalds 
35561da177e4SLinus Torvalds 	case TCP_KEEPIDLE:
3557aad4a0a9SDmitry Yakunin 		err = tcp_sock_set_keepidle_locked(sk, val);
35581da177e4SLinus Torvalds 		break;
35591da177e4SLinus Torvalds 	case TCP_KEEPINTVL:
35601da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_KEEPINTVL)
35611da177e4SLinus Torvalds 			err = -EINVAL;
35621da177e4SLinus Torvalds 		else
35635ecf9d4fSEric Dumazet 			WRITE_ONCE(tp->keepalive_intvl, val * HZ);
35641da177e4SLinus Torvalds 		break;
35651da177e4SLinus Torvalds 	case TCP_KEEPCNT:
35661da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_KEEPCNT)
35671da177e4SLinus Torvalds 			err = -EINVAL;
35681da177e4SLinus Torvalds 		else
3569*6e5e1de6SEric Dumazet 			WRITE_ONCE(tp->keepalive_probes, val);
35701da177e4SLinus Torvalds 		break;
35711da177e4SLinus Torvalds 	case TCP_SYNCNT:
35721da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_SYNCNT)
35731da177e4SLinus Torvalds 			err = -EINVAL;
35741da177e4SLinus Torvalds 		else
3575463c84b9SArnaldo Carvalho de Melo 			icsk->icsk_syn_retries = val;
35761da177e4SLinus Torvalds 		break;
35771da177e4SLinus Torvalds 
3578cd8ae852SEric Dumazet 	case TCP_SAVE_SYN:
3579267cf9faSMartin KaFai Lau 		/* 0: disable, 1: enable, 2: start from ether_header */
3580267cf9faSMartin KaFai Lau 		if (val < 0 || val > 2)
3581cd8ae852SEric Dumazet 			err = -EINVAL;
3582cd8ae852SEric Dumazet 		else
3583cd8ae852SEric Dumazet 			tp->save_syn = val;
3584cd8ae852SEric Dumazet 		break;
3585cd8ae852SEric Dumazet 
35861da177e4SLinus Torvalds 	case TCP_LINGER2:
35871da177e4SLinus Torvalds 		if (val < 0)
35881da177e4SLinus Torvalds 			tp->linger2 = -1;
3589f0628c52SCambda Zhu 		else if (val > TCP_FIN_TIMEOUT_MAX / HZ)
3590f0628c52SCambda Zhu 			tp->linger2 = TCP_FIN_TIMEOUT_MAX;
35911da177e4SLinus Torvalds 		else
35921da177e4SLinus Torvalds 			tp->linger2 = val * HZ;
35931da177e4SLinus Torvalds 		break;
35941da177e4SLinus Torvalds 
35951da177e4SLinus Torvalds 	case TCP_DEFER_ACCEPT:
3596b103cf34SJulian Anastasov 		/* Translate value in seconds to number of retransmits */
3597b103cf34SJulian Anastasov 		icsk->icsk_accept_queue.rskq_defer_accept =
3598b103cf34SJulian Anastasov 			secs_to_retrans(val, TCP_TIMEOUT_INIT / HZ,
3599b103cf34SJulian Anastasov 					TCP_RTO_MAX / HZ);
36001da177e4SLinus Torvalds 		break;
36011da177e4SLinus Torvalds 
36021da177e4SLinus Torvalds 	case TCP_WINDOW_CLAMP:
3603cb811109SPrankur gupta 		err = tcp_set_window_clamp(sk, val);
36041da177e4SLinus Torvalds 		break;
36051da177e4SLinus Torvalds 
36061da177e4SLinus Torvalds 	case TCP_QUICKACK:
3607ddd061b8SChristoph Hellwig 		__tcp_sock_set_quickack(sk, val);
36081da177e4SLinus Torvalds 		break;
36091da177e4SLinus Torvalds 
3610cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG
3611cfb6eeb4SYOSHIFUJI Hideaki 	case TCP_MD5SIG:
36128917a777SIvan Delalande 	case TCP_MD5SIG_EXT:
3613d38d2b00SChristoph Hellwig 		err = tp->af_specific->md5_parse(sk, optname, optval, optlen);
3614cfb6eeb4SYOSHIFUJI Hideaki 		break;
3615cfb6eeb4SYOSHIFUJI Hideaki #endif
3616dca43c75SJerry Chu 	case TCP_USER_TIMEOUT:
3617b248230cSYuchung Cheng 		/* Cap the max time in ms TCP will retry or probe the window
3618dca43c75SJerry Chu 		 * before giving up and aborting (ETIMEDOUT) a connection.
3619dca43c75SJerry Chu 		 */
362042493570SHangbin Liu 		if (val < 0)
362142493570SHangbin Liu 			err = -EINVAL;
362242493570SHangbin Liu 		else
36239bcc66e1SJon Maxwell 			icsk->icsk_user_timeout = val;
3624dca43c75SJerry Chu 		break;
36258336886fSJerry Chu 
36268336886fSJerry Chu 	case TCP_FASTOPEN:
36278336886fSJerry Chu 		if (val >= 0 && ((1 << sk->sk_state) & (TCPF_CLOSE |
3628dfea2aa6SChristoph Paasch 		    TCPF_LISTEN))) {
362943713848SHaishuang Yan 			tcp_fastopen_init_key_once(net);
3630dfea2aa6SChristoph Paasch 
36310536fcc0SEric Dumazet 			fastopen_queue_tune(sk, val);
3632dfea2aa6SChristoph Paasch 		} else {
36338336886fSJerry Chu 			err = -EINVAL;
3634dfea2aa6SChristoph Paasch 		}
36358336886fSJerry Chu 		break;
363619f6d3f3SWei Wang 	case TCP_FASTOPEN_CONNECT:
363719f6d3f3SWei Wang 		if (val > 1 || val < 0) {
363819f6d3f3SWei Wang 			err = -EINVAL;
36395a542133SKuniyuki Iwashima 		} else if (READ_ONCE(net->ipv4.sysctl_tcp_fastopen) &
36405a542133SKuniyuki Iwashima 			   TFO_CLIENT_ENABLE) {
364119f6d3f3SWei Wang 			if (sk->sk_state == TCP_CLOSE)
364219f6d3f3SWei Wang 				tp->fastopen_connect = val;
364319f6d3f3SWei Wang 			else
364419f6d3f3SWei Wang 				err = -EINVAL;
364519f6d3f3SWei Wang 		} else {
364619f6d3f3SWei Wang 			err = -EOPNOTSUPP;
364719f6d3f3SWei Wang 		}
364819f6d3f3SWei Wang 		break;
364971c02379SChristoph Paasch 	case TCP_FASTOPEN_NO_COOKIE:
365071c02379SChristoph Paasch 		if (val > 1 || val < 0)
365171c02379SChristoph Paasch 			err = -EINVAL;
365271c02379SChristoph Paasch 		else if (!((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)))
365371c02379SChristoph Paasch 			err = -EINVAL;
365471c02379SChristoph Paasch 		else
365571c02379SChristoph Paasch 			tp->fastopen_no_cookie = val;
365671c02379SChristoph Paasch 		break;
365793be6ce0SAndrey Vagin 	case TCP_TIMESTAMP:
365893be6ce0SAndrey Vagin 		if (!tp->repair)
365993be6ce0SAndrey Vagin 			err = -EPERM;
366093be6ce0SAndrey Vagin 		else
3661dd23c9f1SEric Dumazet 			WRITE_ONCE(tp->tsoffset, val - tcp_time_stamp_raw());
366293be6ce0SAndrey Vagin 		break;
3663b1ed4c4fSAndrey Vagin 	case TCP_REPAIR_WINDOW:
3664b1ed4c4fSAndrey Vagin 		err = tcp_repair_set_window(tp, optval, optlen);
3665b1ed4c4fSAndrey Vagin 		break;
3666c9bee3b7SEric Dumazet 	case TCP_NOTSENT_LOWAT:
3667c9bee3b7SEric Dumazet 		tp->notsent_lowat = val;
3668c9bee3b7SEric Dumazet 		sk->sk_write_space(sk);
3669c9bee3b7SEric Dumazet 		break;
3670b75eba76SSoheil Hassas Yeganeh 	case TCP_INQ:
3671b75eba76SSoheil Hassas Yeganeh 		if (val > 1 || val < 0)
3672b75eba76SSoheil Hassas Yeganeh 			err = -EINVAL;
3673b75eba76SSoheil Hassas Yeganeh 		else
3674b75eba76SSoheil Hassas Yeganeh 			tp->recvmsg_inq = val;
3675b75eba76SSoheil Hassas Yeganeh 		break;
3676a842fe14SEric Dumazet 	case TCP_TX_DELAY:
3677a842fe14SEric Dumazet 		if (val)
3678a842fe14SEric Dumazet 			tcp_enable_tx_delay();
3679348b81b6SEric Dumazet 		WRITE_ONCE(tp->tcp_tx_delay, val);
3680a842fe14SEric Dumazet 		break;
36811da177e4SLinus Torvalds 	default:
36821da177e4SLinus Torvalds 		err = -ENOPROTOOPT;
36831da177e4SLinus Torvalds 		break;
36843ff50b79SStephen Hemminger 	}
36853ff50b79SStephen Hemminger 
3686cb388e7eSMartin KaFai Lau 	sockopt_release_sock(sk);
36871da177e4SLinus Torvalds 	return err;
36881da177e4SLinus Torvalds }
36891da177e4SLinus Torvalds 
3690a7b75c5aSChristoph Hellwig int tcp_setsockopt(struct sock *sk, int level, int optname, sockptr_t optval,
3691b7058842SDavid S. Miller 		   unsigned int optlen)
36923fdadf7dSDmitry Mishin {
3693cf533ea5SEric Dumazet 	const struct inet_connection_sock *icsk = inet_csk(sk);
36943fdadf7dSDmitry Mishin 
36953fdadf7dSDmitry Mishin 	if (level != SOL_TCP)
3696f49cd2f4SKuniyuki Iwashima 		/* Paired with WRITE_ONCE() in do_ipv6_setsockopt() and tcp_v6_connect() */
3697f49cd2f4SKuniyuki Iwashima 		return READ_ONCE(icsk->icsk_af_ops)->setsockopt(sk, level, optname,
36983fdadf7dSDmitry Mishin 								optval, optlen);
3699a7b75c5aSChristoph Hellwig 	return do_tcp_setsockopt(sk, level, optname, optval, optlen);
37003fdadf7dSDmitry Mishin }
37014bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_setsockopt);
37023fdadf7dSDmitry Mishin 
3703efd90174SFrancis Yan static void tcp_get_info_chrono_stats(const struct tcp_sock *tp,
3704efd90174SFrancis Yan 				      struct tcp_info *info)
3705efd90174SFrancis Yan {
3706efd90174SFrancis Yan 	u64 stats[__TCP_CHRONO_MAX], total = 0;
3707efd90174SFrancis Yan 	enum tcp_chrono i;
3708efd90174SFrancis Yan 
3709efd90174SFrancis Yan 	for (i = TCP_CHRONO_BUSY; i < __TCP_CHRONO_MAX; ++i) {
3710efd90174SFrancis Yan 		stats[i] = tp->chrono_stat[i - 1];
3711efd90174SFrancis Yan 		if (i == tp->chrono_type)
3712628174ccSEric Dumazet 			stats[i] += tcp_jiffies32 - tp->chrono_start;
3713efd90174SFrancis Yan 		stats[i] *= USEC_PER_SEC / HZ;
3714efd90174SFrancis Yan 		total += stats[i];
3715efd90174SFrancis Yan 	}
3716efd90174SFrancis Yan 
3717efd90174SFrancis Yan 	info->tcpi_busy_time = total;
3718efd90174SFrancis Yan 	info->tcpi_rwnd_limited = stats[TCP_CHRONO_RWND_LIMITED];
3719efd90174SFrancis Yan 	info->tcpi_sndbuf_limited = stats[TCP_CHRONO_SNDBUF_LIMITED];
3720efd90174SFrancis Yan }
3721efd90174SFrancis Yan 
37221da177e4SLinus Torvalds /* Return information about state of tcp endpoint in API format. */
37230df48c26SEric Dumazet void tcp_get_info(struct sock *sk, struct tcp_info *info)
37241da177e4SLinus Torvalds {
372535ac838aSCraig Gallek 	const struct tcp_sock *tp = tcp_sk(sk); /* iff sk_type == SOCK_STREAM */
3726463c84b9SArnaldo Carvalho de Melo 	const struct inet_connection_sock *icsk = inet_csk(sk);
372776a9ebe8SEric Dumazet 	unsigned long rate;
37280263598cSWei Wang 	u32 now;
3729ff5d7497SEric Dumazet 	u64 rate64;
373067db3e4bSEric Dumazet 	bool slow;
37311da177e4SLinus Torvalds 
37321da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
373335ac838aSCraig Gallek 	if (sk->sk_type != SOCK_STREAM)
373435ac838aSCraig Gallek 		return;
37351da177e4SLinus Torvalds 
3736986ffdfdSYafang Shao 	info->tcpi_state = inet_sk_state_load(sk);
373700fd38d9SEric Dumazet 
3738ccbf3bfaSEric Dumazet 	/* Report meaningful fields for all TCP states, including listeners */
3739ccbf3bfaSEric Dumazet 	rate = READ_ONCE(sk->sk_pacing_rate);
374076a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3741f522a5fcSEric Dumazet 	info->tcpi_pacing_rate = rate64;
3742ccbf3bfaSEric Dumazet 
3743ccbf3bfaSEric Dumazet 	rate = READ_ONCE(sk->sk_max_pacing_rate);
374476a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3745f522a5fcSEric Dumazet 	info->tcpi_max_pacing_rate = rate64;
3746ccbf3bfaSEric Dumazet 
3747ccbf3bfaSEric Dumazet 	info->tcpi_reordering = tp->reordering;
374840570375SEric Dumazet 	info->tcpi_snd_cwnd = tcp_snd_cwnd(tp);
3749ccbf3bfaSEric Dumazet 
3750ccbf3bfaSEric Dumazet 	if (info->tcpi_state == TCP_LISTEN) {
3751ccbf3bfaSEric Dumazet 		/* listeners aliased fields :
3752ccbf3bfaSEric Dumazet 		 * tcpi_unacked -> Number of children ready for accept()
3753ccbf3bfaSEric Dumazet 		 * tcpi_sacked  -> max backlog
3754ccbf3bfaSEric Dumazet 		 */
3755288efe86SEric Dumazet 		info->tcpi_unacked = READ_ONCE(sk->sk_ack_backlog);
3756099ecf59SEric Dumazet 		info->tcpi_sacked = READ_ONCE(sk->sk_max_ack_backlog);
3757ccbf3bfaSEric Dumazet 		return;
3758ccbf3bfaSEric Dumazet 	}
3759b369e7fdSEric Dumazet 
3760b369e7fdSEric Dumazet 	slow = lock_sock_fast(sk);
3761b369e7fdSEric Dumazet 
37626687e988SArnaldo Carvalho de Melo 	info->tcpi_ca_state = icsk->icsk_ca_state;
3763463c84b9SArnaldo Carvalho de Melo 	info->tcpi_retransmits = icsk->icsk_retransmits;
37646687e988SArnaldo Carvalho de Melo 	info->tcpi_probes = icsk->icsk_probes_out;
3765463c84b9SArnaldo Carvalho de Melo 	info->tcpi_backoff = icsk->icsk_backoff;
37661da177e4SLinus Torvalds 
37671da177e4SLinus Torvalds 	if (tp->rx_opt.tstamp_ok)
37681da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_TIMESTAMPS;
3769e60402d0SIlpo Järvinen 	if (tcp_is_sack(tp))
37701da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_SACK;
37711da177e4SLinus Torvalds 	if (tp->rx_opt.wscale_ok) {
37721da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_WSCALE;
37731da177e4SLinus Torvalds 		info->tcpi_snd_wscale = tp->rx_opt.snd_wscale;
37741da177e4SLinus Torvalds 		info->tcpi_rcv_wscale = tp->rx_opt.rcv_wscale;
37751da177e4SLinus Torvalds 	}
37761da177e4SLinus Torvalds 
37771da177e4SLinus Torvalds 	if (tp->ecn_flags & TCP_ECN_OK)
37781da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_ECN;
3779b5c5693bSEric Dumazet 	if (tp->ecn_flags & TCP_ECN_SEEN)
3780b5c5693bSEric Dumazet 		info->tcpi_options |= TCPI_OPT_ECN_SEEN;
37816f73601eSYuchung Cheng 	if (tp->syn_data_acked)
37826f73601eSYuchung Cheng 		info->tcpi_options |= TCPI_OPT_SYN_DATA;
37831da177e4SLinus Torvalds 
3784463c84b9SArnaldo Carvalho de Melo 	info->tcpi_rto = jiffies_to_usecs(icsk->icsk_rto);
3785463c84b9SArnaldo Carvalho de Melo 	info->tcpi_ato = jiffies_to_usecs(icsk->icsk_ack.ato);
3786c1b4a7e6SDavid S. Miller 	info->tcpi_snd_mss = tp->mss_cache;
3787463c84b9SArnaldo Carvalho de Melo 	info->tcpi_rcv_mss = icsk->icsk_ack.rcv_mss;
37881da177e4SLinus Torvalds 
37891da177e4SLinus Torvalds 	info->tcpi_unacked = tp->packets_out;
37901da177e4SLinus Torvalds 	info->tcpi_sacked = tp->sacked_out;
3791ccbf3bfaSEric Dumazet 
37921da177e4SLinus Torvalds 	info->tcpi_lost = tp->lost_out;
37931da177e4SLinus Torvalds 	info->tcpi_retrans = tp->retrans_out;
37941da177e4SLinus Torvalds 
3795d635fbe2SEric Dumazet 	now = tcp_jiffies32;
37961da177e4SLinus Torvalds 	info->tcpi_last_data_sent = jiffies_to_msecs(now - tp->lsndtime);
3797463c84b9SArnaldo Carvalho de Melo 	info->tcpi_last_data_recv = jiffies_to_msecs(now - icsk->icsk_ack.lrcvtime);
37981da177e4SLinus Torvalds 	info->tcpi_last_ack_recv = jiffies_to_msecs(now - tp->rcv_tstamp);
37991da177e4SLinus Torvalds 
3800d83d8461SArnaldo Carvalho de Melo 	info->tcpi_pmtu = icsk->icsk_pmtu_cookie;
38011da177e4SLinus Torvalds 	info->tcpi_rcv_ssthresh = tp->rcv_ssthresh;
3802740b0f18SEric Dumazet 	info->tcpi_rtt = tp->srtt_us >> 3;
3803740b0f18SEric Dumazet 	info->tcpi_rttvar = tp->mdev_us >> 2;
38041da177e4SLinus Torvalds 	info->tcpi_snd_ssthresh = tp->snd_ssthresh;
38051da177e4SLinus Torvalds 	info->tcpi_advmss = tp->advmss;
38061da177e4SLinus Torvalds 
3807645f4c6fSEric Dumazet 	info->tcpi_rcv_rtt = tp->rcv_rtt_est.rtt_us >> 3;
38081da177e4SLinus Torvalds 	info->tcpi_rcv_space = tp->rcvq_space.space;
38091da177e4SLinus Torvalds 
38101da177e4SLinus Torvalds 	info->tcpi_total_retrans = tp->total_retrans;
3811977cb0ecSEric Dumazet 
3812f522a5fcSEric Dumazet 	info->tcpi_bytes_acked = tp->bytes_acked;
3813f522a5fcSEric Dumazet 	info->tcpi_bytes_received = tp->bytes_received;
381467db3e4bSEric Dumazet 	info->tcpi_notsent_bytes = max_t(int, 0, tp->write_seq - tp->snd_nxt);
3815efd90174SFrancis Yan 	tcp_get_info_chrono_stats(tp, info);
381667db3e4bSEric Dumazet 
38172efd055cSMarcelo Ricardo Leitner 	info->tcpi_segs_out = tp->segs_out;
38180307a0b7SEric Dumazet 
38190307a0b7SEric Dumazet 	/* segs_in and data_segs_in can be updated from tcp_segs_in() from BH */
38200307a0b7SEric Dumazet 	info->tcpi_segs_in = READ_ONCE(tp->segs_in);
38210307a0b7SEric Dumazet 	info->tcpi_data_segs_in = READ_ONCE(tp->data_segs_in);
3822cd9b2660SEric Dumazet 
3823cd9b2660SEric Dumazet 	info->tcpi_min_rtt = tcp_min_rtt(tp);
3824a44d6eacSMartin KaFai Lau 	info->tcpi_data_segs_out = tp->data_segs_out;
3825eb8329e0SYuchung Cheng 
3826eb8329e0SYuchung Cheng 	info->tcpi_delivery_rate_app_limited = tp->rate_app_limited ? 1 : 0;
38270263598cSWei Wang 	rate64 = tcp_compute_delivery_rate(tp);
38280263598cSWei Wang 	if (rate64)
3829f522a5fcSEric Dumazet 		info->tcpi_delivery_rate = rate64;
3830feb5f2ecSYuchung Cheng 	info->tcpi_delivered = tp->delivered;
3831feb5f2ecSYuchung Cheng 	info->tcpi_delivered_ce = tp->delivered_ce;
3832ba113c3aSWei Wang 	info->tcpi_bytes_sent = tp->bytes_sent;
3833fb31c9b9SWei Wang 	info->tcpi_bytes_retrans = tp->bytes_retrans;
38347e10b655SWei Wang 	info->tcpi_dsack_dups = tp->dsack_dups;
38357ec65372SWei Wang 	info->tcpi_reord_seen = tp->reord_seen;
3836f9af2dbbSThomas Higdon 	info->tcpi_rcv_ooopack = tp->rcv_ooopack;
38378f7baad7SThomas Higdon 	info->tcpi_snd_wnd = tp->snd_wnd;
383871fc7047SMubashir Adnan Qureshi 	info->tcpi_rcv_wnd = tp->rcv_wnd;
383971fc7047SMubashir Adnan Qureshi 	info->tcpi_rehash = tp->plb_rehash + tp->timeout_rehash;
384048027478SJason Baron 	info->tcpi_fastopen_client_fail = tp->fastopen_client_fail;
3841b369e7fdSEric Dumazet 	unlock_sock_fast(sk, slow);
38421da177e4SLinus Torvalds }
38431da177e4SLinus Torvalds EXPORT_SYMBOL_GPL(tcp_get_info);
38441da177e4SLinus Torvalds 
3845984988aaSWei Wang static size_t tcp_opt_stats_get_size(void)
3846984988aaSWei Wang {
3847984988aaSWei Wang 	return
3848984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BUSY */
3849984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_RWND_LIMITED */
3850984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_SNDBUF_LIMITED */
3851984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DATA_SEGS_OUT */
3852984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_TOTAL_RETRANS */
3853984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_PACING_RATE */
3854984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DELIVERY_RATE */
3855984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_CWND */
3856984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_REORDERING */
3857984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_MIN_RTT */
3858984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_RECUR_RETRANS */
3859984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_DELIVERY_RATE_APP_LMT */
3860984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SNDQ_SIZE */
3861984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_CA_STATE */
3862984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_SSTHRESH */
3863984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED */
3864984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED_CE */
3865ba113c3aSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_SENT */
3866fb31c9b9SWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_RETRANS */
38677e10b655SWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DSACK_DUPS */
38687ec65372SWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_REORD_SEEN */
3869e8bd8fcaSYousuk Seung 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SRTT */
387032efcc06SAbdul Kabbani 		nla_total_size(sizeof(u16)) + /* TCP_NLA_TIMEOUT_REHASH */
3871e08ab0b3SYousuk Seung 		nla_total_size(sizeof(u32)) + /* TCP_NLA_BYTES_NOTSENT */
387248040793SYousuk Seung 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_EDT */
3873e7ed11eeSYousuk Seung 		nla_total_size(sizeof(u8)) + /* TCP_NLA_TTL */
387429c1c446SMubashir Adnan Qureshi 		nla_total_size(sizeof(u32)) + /* TCP_NLA_REHASH */
3875984988aaSWei Wang 		0;
3876984988aaSWei Wang }
3877984988aaSWei Wang 
3878e7ed11eeSYousuk Seung /* Returns TTL or hop limit of an incoming packet from skb. */
3879e7ed11eeSYousuk Seung static u8 tcp_skb_ttl_or_hop_limit(const struct sk_buff *skb)
3880e7ed11eeSYousuk Seung {
3881e7ed11eeSYousuk Seung 	if (skb->protocol == htons(ETH_P_IP))
3882e7ed11eeSYousuk Seung 		return ip_hdr(skb)->ttl;
3883e7ed11eeSYousuk Seung 	else if (skb->protocol == htons(ETH_P_IPV6))
3884e7ed11eeSYousuk Seung 		return ipv6_hdr(skb)->hop_limit;
3885e7ed11eeSYousuk Seung 	else
3886e7ed11eeSYousuk Seung 		return 0;
3887e7ed11eeSYousuk Seung }
3888e7ed11eeSYousuk Seung 
388948040793SYousuk Seung struct sk_buff *tcp_get_timestamping_opt_stats(const struct sock *sk,
3890e7ed11eeSYousuk Seung 					       const struct sk_buff *orig_skb,
3891e7ed11eeSYousuk Seung 					       const struct sk_buff *ack_skb)
38921c885808SFrancis Yan {
38931c885808SFrancis Yan 	const struct tcp_sock *tp = tcp_sk(sk);
38941c885808SFrancis Yan 	struct sk_buff *stats;
38951c885808SFrancis Yan 	struct tcp_info info;
389676a9ebe8SEric Dumazet 	unsigned long rate;
3897bb7c19f9SWei Wang 	u64 rate64;
38981c885808SFrancis Yan 
3899984988aaSWei Wang 	stats = alloc_skb(tcp_opt_stats_get_size(), GFP_ATOMIC);
39001c885808SFrancis Yan 	if (!stats)
39011c885808SFrancis Yan 		return NULL;
39021c885808SFrancis Yan 
39031c885808SFrancis Yan 	tcp_get_info_chrono_stats(tp, &info);
39041c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_BUSY,
39051c885808SFrancis Yan 			  info.tcpi_busy_time, TCP_NLA_PAD);
39061c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_RWND_LIMITED,
39071c885808SFrancis Yan 			  info.tcpi_rwnd_limited, TCP_NLA_PAD);
39081c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_SNDBUF_LIMITED,
39091c885808SFrancis Yan 			  info.tcpi_sndbuf_limited, TCP_NLA_PAD);
39107e98102fSYuchung Cheng 	nla_put_u64_64bit(stats, TCP_NLA_DATA_SEGS_OUT,
39117e98102fSYuchung Cheng 			  tp->data_segs_out, TCP_NLA_PAD);
39127e98102fSYuchung Cheng 	nla_put_u64_64bit(stats, TCP_NLA_TOTAL_RETRANS,
39137e98102fSYuchung Cheng 			  tp->total_retrans, TCP_NLA_PAD);
3914bb7c19f9SWei Wang 
3915bb7c19f9SWei Wang 	rate = READ_ONCE(sk->sk_pacing_rate);
391676a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3917bb7c19f9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_PACING_RATE, rate64, TCP_NLA_PAD);
3918bb7c19f9SWei Wang 
3919bb7c19f9SWei Wang 	rate64 = tcp_compute_delivery_rate(tp);
3920bb7c19f9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_DELIVERY_RATE, rate64, TCP_NLA_PAD);
3921bb7c19f9SWei Wang 
392240570375SEric Dumazet 	nla_put_u32(stats, TCP_NLA_SND_CWND, tcp_snd_cwnd(tp));
3923bb7c19f9SWei Wang 	nla_put_u32(stats, TCP_NLA_REORDERING, tp->reordering);
3924bb7c19f9SWei Wang 	nla_put_u32(stats, TCP_NLA_MIN_RTT, tcp_min_rtt(tp));
3925bb7c19f9SWei Wang 
3926bb7c19f9SWei Wang 	nla_put_u8(stats, TCP_NLA_RECUR_RETRANS, inet_csk(sk)->icsk_retransmits);
3927bb7c19f9SWei Wang 	nla_put_u8(stats, TCP_NLA_DELIVERY_RATE_APP_LMT, !!tp->rate_app_limited);
39287156d194SYousuk Seung 	nla_put_u32(stats, TCP_NLA_SND_SSTHRESH, tp->snd_ssthresh);
3929feb5f2ecSYuchung Cheng 	nla_put_u32(stats, TCP_NLA_DELIVERED, tp->delivered);
3930feb5f2ecSYuchung Cheng 	nla_put_u32(stats, TCP_NLA_DELIVERED_CE, tp->delivered_ce);
393187ecc95dSPriyaranjan Jha 
393287ecc95dSPriyaranjan Jha 	nla_put_u32(stats, TCP_NLA_SNDQ_SIZE, tp->write_seq - tp->snd_una);
3933be631892SPriyaranjan Jha 	nla_put_u8(stats, TCP_NLA_CA_STATE, inet_csk(sk)->icsk_ca_state);
3934feb5f2ecSYuchung Cheng 
3935ba113c3aSWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_BYTES_SENT, tp->bytes_sent,
3936ba113c3aSWei Wang 			  TCP_NLA_PAD);
3937fb31c9b9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_BYTES_RETRANS, tp->bytes_retrans,
3938fb31c9b9SWei Wang 			  TCP_NLA_PAD);
39397e10b655SWei Wang 	nla_put_u32(stats, TCP_NLA_DSACK_DUPS, tp->dsack_dups);
39407ec65372SWei Wang 	nla_put_u32(stats, TCP_NLA_REORD_SEEN, tp->reord_seen);
3941e8bd8fcaSYousuk Seung 	nla_put_u32(stats, TCP_NLA_SRTT, tp->srtt_us >> 3);
394232efcc06SAbdul Kabbani 	nla_put_u16(stats, TCP_NLA_TIMEOUT_REHASH, tp->timeout_rehash);
3943e08ab0b3SYousuk Seung 	nla_put_u32(stats, TCP_NLA_BYTES_NOTSENT,
3944e08ab0b3SYousuk Seung 		    max_t(int, 0, tp->write_seq - tp->snd_nxt));
394548040793SYousuk Seung 	nla_put_u64_64bit(stats, TCP_NLA_EDT, orig_skb->skb_mstamp_ns,
394648040793SYousuk Seung 			  TCP_NLA_PAD);
3947e7ed11eeSYousuk Seung 	if (ack_skb)
3948e7ed11eeSYousuk Seung 		nla_put_u8(stats, TCP_NLA_TTL,
3949e7ed11eeSYousuk Seung 			   tcp_skb_ttl_or_hop_limit(ack_skb));
3950ba113c3aSWei Wang 
395129c1c446SMubashir Adnan Qureshi 	nla_put_u32(stats, TCP_NLA_REHASH, tp->plb_rehash + tp->timeout_rehash);
39521c885808SFrancis Yan 	return stats;
39531c885808SFrancis Yan }
39541c885808SFrancis Yan 
3955273b7f0fSMartin KaFai Lau int do_tcp_getsockopt(struct sock *sk, int level,
395634704ef0SMartin KaFai Lau 		      int optname, sockptr_t optval, sockptr_t optlen)
39571da177e4SLinus Torvalds {
3958295f7324SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
39591da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
39606fa25166SNikolay Borisov 	struct net *net = sock_net(sk);
39611da177e4SLinus Torvalds 	int val, len;
39621da177e4SLinus Torvalds 
396334704ef0SMartin KaFai Lau 	if (copy_from_sockptr(&len, optlen, sizeof(int)))
39641da177e4SLinus Torvalds 		return -EFAULT;
39651da177e4SLinus Torvalds 
39661da177e4SLinus Torvalds 	len = min_t(unsigned int, len, sizeof(int));
39671da177e4SLinus Torvalds 
39681da177e4SLinus Torvalds 	if (len < 0)
39691da177e4SLinus Torvalds 		return -EINVAL;
39701da177e4SLinus Torvalds 
39711da177e4SLinus Torvalds 	switch (optname) {
39721da177e4SLinus Torvalds 	case TCP_MAXSEG:
3973c1b4a7e6SDavid S. Miller 		val = tp->mss_cache;
397434dfde4aSCambda Zhu 		if (tp->rx_opt.user_mss &&
397534dfde4aSCambda Zhu 		    ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)))
39761da177e4SLinus Torvalds 			val = tp->rx_opt.user_mss;
39775e6a3ce6SPavel Emelyanov 		if (tp->repair)
39785e6a3ce6SPavel Emelyanov 			val = tp->rx_opt.mss_clamp;
39791da177e4SLinus Torvalds 		break;
39801da177e4SLinus Torvalds 	case TCP_NODELAY:
39811da177e4SLinus Torvalds 		val = !!(tp->nonagle&TCP_NAGLE_OFF);
39821da177e4SLinus Torvalds 		break;
39831da177e4SLinus Torvalds 	case TCP_CORK:
39841da177e4SLinus Torvalds 		val = !!(tp->nonagle&TCP_NAGLE_CORK);
39851da177e4SLinus Torvalds 		break;
39861da177e4SLinus Torvalds 	case TCP_KEEPIDLE:
3987df19a626SEric Dumazet 		val = keepalive_time_when(tp) / HZ;
39881da177e4SLinus Torvalds 		break;
39891da177e4SLinus Torvalds 	case TCP_KEEPINTVL:
3990df19a626SEric Dumazet 		val = keepalive_intvl_when(tp) / HZ;
39911da177e4SLinus Torvalds 		break;
39921da177e4SLinus Torvalds 	case TCP_KEEPCNT:
3993df19a626SEric Dumazet 		val = keepalive_probes(tp);
39941da177e4SLinus Torvalds 		break;
39951da177e4SLinus Torvalds 	case TCP_SYNCNT:
399620a3b1c0SKuniyuki Iwashima 		val = icsk->icsk_syn_retries ? :
399720a3b1c0SKuniyuki Iwashima 			READ_ONCE(net->ipv4.sysctl_tcp_syn_retries);
39981da177e4SLinus Torvalds 		break;
39991da177e4SLinus Torvalds 	case TCP_LINGER2:
40001da177e4SLinus Torvalds 		val = tp->linger2;
40011da177e4SLinus Torvalds 		if (val >= 0)
400239e24435SKuniyuki Iwashima 			val = (val ? : READ_ONCE(net->ipv4.sysctl_tcp_fin_timeout)) / HZ;
40031da177e4SLinus Torvalds 		break;
40041da177e4SLinus Torvalds 	case TCP_DEFER_ACCEPT:
4005b103cf34SJulian Anastasov 		val = retrans_to_secs(icsk->icsk_accept_queue.rskq_defer_accept,
4006b103cf34SJulian Anastasov 				      TCP_TIMEOUT_INIT / HZ, TCP_RTO_MAX / HZ);
40071da177e4SLinus Torvalds 		break;
40081da177e4SLinus Torvalds 	case TCP_WINDOW_CLAMP:
40091da177e4SLinus Torvalds 		val = tp->window_clamp;
40101da177e4SLinus Torvalds 		break;
40111da177e4SLinus Torvalds 	case TCP_INFO: {
40121da177e4SLinus Torvalds 		struct tcp_info info;
40131da177e4SLinus Torvalds 
401434704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
40151da177e4SLinus Torvalds 			return -EFAULT;
40161da177e4SLinus Torvalds 
40171da177e4SLinus Torvalds 		tcp_get_info(sk, &info);
40181da177e4SLinus Torvalds 
40191da177e4SLinus Torvalds 		len = min_t(unsigned int, len, sizeof(info));
402034704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optlen, &len, sizeof(int)))
40211da177e4SLinus Torvalds 			return -EFAULT;
402234704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, &info, len))
40231da177e4SLinus Torvalds 			return -EFAULT;
40241da177e4SLinus Torvalds 		return 0;
40251da177e4SLinus Torvalds 	}
40266e9250f5SEric Dumazet 	case TCP_CC_INFO: {
40276e9250f5SEric Dumazet 		const struct tcp_congestion_ops *ca_ops;
40286e9250f5SEric Dumazet 		union tcp_cc_info info;
40296e9250f5SEric Dumazet 		size_t sz = 0;
40306e9250f5SEric Dumazet 		int attr;
40316e9250f5SEric Dumazet 
403234704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
40336e9250f5SEric Dumazet 			return -EFAULT;
40346e9250f5SEric Dumazet 
40356e9250f5SEric Dumazet 		ca_ops = icsk->icsk_ca_ops;
40366e9250f5SEric Dumazet 		if (ca_ops && ca_ops->get_info)
40376e9250f5SEric Dumazet 			sz = ca_ops->get_info(sk, ~0U, &attr, &info);
40386e9250f5SEric Dumazet 
40396e9250f5SEric Dumazet 		len = min_t(unsigned int, len, sz);
404034704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optlen, &len, sizeof(int)))
40416e9250f5SEric Dumazet 			return -EFAULT;
404234704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, &info, len))
40436e9250f5SEric Dumazet 			return -EFAULT;
40446e9250f5SEric Dumazet 		return 0;
40456e9250f5SEric Dumazet 	}
40461da177e4SLinus Torvalds 	case TCP_QUICKACK:
404731954cd8SWei Wang 		val = !inet_csk_in_pingpong_mode(sk);
40481da177e4SLinus Torvalds 		break;
40495f8ef48dSStephen Hemminger 
40505f8ef48dSStephen Hemminger 	case TCP_CONGESTION:
405134704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
40525f8ef48dSStephen Hemminger 			return -EFAULT;
40535f8ef48dSStephen Hemminger 		len = min_t(unsigned int, len, TCP_CA_NAME_MAX);
405434704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optlen, &len, sizeof(int)))
40555f8ef48dSStephen Hemminger 			return -EFAULT;
405634704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, icsk->icsk_ca_ops->name, len))
40575f8ef48dSStephen Hemminger 			return -EFAULT;
40585f8ef48dSStephen Hemminger 		return 0;
4059e56fb50fSWilliam Allen Simpson 
4060734942ccSDave Watson 	case TCP_ULP:
406134704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
4062734942ccSDave Watson 			return -EFAULT;
4063734942ccSDave Watson 		len = min_t(unsigned int, len, TCP_ULP_NAME_MAX);
4064d97af30fSDave Watson 		if (!icsk->icsk_ulp_ops) {
406534704ef0SMartin KaFai Lau 			len = 0;
406634704ef0SMartin KaFai Lau 			if (copy_to_sockptr(optlen, &len, sizeof(int)))
4067d97af30fSDave Watson 				return -EFAULT;
4068d97af30fSDave Watson 			return 0;
4069d97af30fSDave Watson 		}
407034704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optlen, &len, sizeof(int)))
4071734942ccSDave Watson 			return -EFAULT;
407234704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, icsk->icsk_ulp_ops->name, len))
4073734942ccSDave Watson 			return -EFAULT;
4074734942ccSDave Watson 		return 0;
4075734942ccSDave Watson 
40761fba70e5SYuchung Cheng 	case TCP_FASTOPEN_KEY: {
4077f19008e6SJason Baron 		u64 key[TCP_FASTOPEN_KEY_BUF_LENGTH / sizeof(u64)];
4078f19008e6SJason Baron 		unsigned int key_len;
40791fba70e5SYuchung Cheng 
408034704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
40811fba70e5SYuchung Cheng 			return -EFAULT;
40821fba70e5SYuchung Cheng 
4083f19008e6SJason Baron 		key_len = tcp_fastopen_get_cipher(net, icsk, key) *
40840f1ce023SJason Baron 				TCP_FASTOPEN_KEY_LENGTH;
40850f1ce023SJason Baron 		len = min_t(unsigned int, len, key_len);
408634704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optlen, &len, sizeof(int)))
40871fba70e5SYuchung Cheng 			return -EFAULT;
408834704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, key, len))
40891fba70e5SYuchung Cheng 			return -EFAULT;
40901fba70e5SYuchung Cheng 		return 0;
40911fba70e5SYuchung Cheng 	}
40923c0fef0bSJosh Hunt 	case TCP_THIN_LINEAR_TIMEOUTS:
40933c0fef0bSJosh Hunt 		val = tp->thin_lto;
40943c0fef0bSJosh Hunt 		break;
40954a7f6009SYuchung Cheng 
40963c0fef0bSJosh Hunt 	case TCP_THIN_DUPACK:
40974a7f6009SYuchung Cheng 		val = 0;
40983c0fef0bSJosh Hunt 		break;
4099dca43c75SJerry Chu 
4100ee995283SPavel Emelyanov 	case TCP_REPAIR:
4101ee995283SPavel Emelyanov 		val = tp->repair;
4102ee995283SPavel Emelyanov 		break;
4103ee995283SPavel Emelyanov 
4104ee995283SPavel Emelyanov 	case TCP_REPAIR_QUEUE:
4105ee995283SPavel Emelyanov 		if (tp->repair)
4106ee995283SPavel Emelyanov 			val = tp->repair_queue;
4107ee995283SPavel Emelyanov 		else
4108ee995283SPavel Emelyanov 			return -EINVAL;
4109ee995283SPavel Emelyanov 		break;
4110ee995283SPavel Emelyanov 
4111b1ed4c4fSAndrey Vagin 	case TCP_REPAIR_WINDOW: {
4112b1ed4c4fSAndrey Vagin 		struct tcp_repair_window opt;
4113b1ed4c4fSAndrey Vagin 
411434704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
4115b1ed4c4fSAndrey Vagin 			return -EFAULT;
4116b1ed4c4fSAndrey Vagin 
4117b1ed4c4fSAndrey Vagin 		if (len != sizeof(opt))
4118b1ed4c4fSAndrey Vagin 			return -EINVAL;
4119b1ed4c4fSAndrey Vagin 
4120b1ed4c4fSAndrey Vagin 		if (!tp->repair)
4121b1ed4c4fSAndrey Vagin 			return -EPERM;
4122b1ed4c4fSAndrey Vagin 
4123b1ed4c4fSAndrey Vagin 		opt.snd_wl1	= tp->snd_wl1;
4124b1ed4c4fSAndrey Vagin 		opt.snd_wnd	= tp->snd_wnd;
4125b1ed4c4fSAndrey Vagin 		opt.max_window	= tp->max_window;
4126b1ed4c4fSAndrey Vagin 		opt.rcv_wnd	= tp->rcv_wnd;
4127b1ed4c4fSAndrey Vagin 		opt.rcv_wup	= tp->rcv_wup;
4128b1ed4c4fSAndrey Vagin 
412934704ef0SMartin KaFai Lau 		if (copy_to_sockptr(optval, &opt, len))
4130b1ed4c4fSAndrey Vagin 			return -EFAULT;
4131b1ed4c4fSAndrey Vagin 		return 0;
4132b1ed4c4fSAndrey Vagin 	}
4133ee995283SPavel Emelyanov 	case TCP_QUEUE_SEQ:
4134ee995283SPavel Emelyanov 		if (tp->repair_queue == TCP_SEND_QUEUE)
4135ee995283SPavel Emelyanov 			val = tp->write_seq;
4136ee995283SPavel Emelyanov 		else if (tp->repair_queue == TCP_RECV_QUEUE)
4137ee995283SPavel Emelyanov 			val = tp->rcv_nxt;
4138ee995283SPavel Emelyanov 		else
4139ee995283SPavel Emelyanov 			return -EINVAL;
4140ee995283SPavel Emelyanov 		break;
4141ee995283SPavel Emelyanov 
4142dca43c75SJerry Chu 	case TCP_USER_TIMEOUT:
41439bcc66e1SJon Maxwell 		val = icsk->icsk_user_timeout;
4144dca43c75SJerry Chu 		break;
41451536e285SKenjiro Nakayama 
41461536e285SKenjiro Nakayama 	case TCP_FASTOPEN:
41470536fcc0SEric Dumazet 		val = icsk->icsk_accept_queue.fastopenq.max_qlen;
41481536e285SKenjiro Nakayama 		break;
41491536e285SKenjiro Nakayama 
415019f6d3f3SWei Wang 	case TCP_FASTOPEN_CONNECT:
415119f6d3f3SWei Wang 		val = tp->fastopen_connect;
415219f6d3f3SWei Wang 		break;
415319f6d3f3SWei Wang 
415471c02379SChristoph Paasch 	case TCP_FASTOPEN_NO_COOKIE:
415571c02379SChristoph Paasch 		val = tp->fastopen_no_cookie;
415671c02379SChristoph Paasch 		break;
415771c02379SChristoph Paasch 
4158a842fe14SEric Dumazet 	case TCP_TX_DELAY:
4159348b81b6SEric Dumazet 		val = READ_ONCE(tp->tcp_tx_delay);
4160a842fe14SEric Dumazet 		break;
4161a842fe14SEric Dumazet 
416293be6ce0SAndrey Vagin 	case TCP_TIMESTAMP:
4163dd23c9f1SEric Dumazet 		val = tcp_time_stamp_raw() + READ_ONCE(tp->tsoffset);
416493be6ce0SAndrey Vagin 		break;
4165c9bee3b7SEric Dumazet 	case TCP_NOTSENT_LOWAT:
4166c9bee3b7SEric Dumazet 		val = tp->notsent_lowat;
4167c9bee3b7SEric Dumazet 		break;
4168b75eba76SSoheil Hassas Yeganeh 	case TCP_INQ:
4169b75eba76SSoheil Hassas Yeganeh 		val = tp->recvmsg_inq;
4170b75eba76SSoheil Hassas Yeganeh 		break;
4171cd8ae852SEric Dumazet 	case TCP_SAVE_SYN:
4172cd8ae852SEric Dumazet 		val = tp->save_syn;
4173cd8ae852SEric Dumazet 		break;
4174cd8ae852SEric Dumazet 	case TCP_SAVED_SYN: {
417534704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
4176cd8ae852SEric Dumazet 			return -EFAULT;
4177cd8ae852SEric Dumazet 
4178d51bbff2SMartin KaFai Lau 		sockopt_lock_sock(sk);
4179cd8ae852SEric Dumazet 		if (tp->saved_syn) {
418070a217f1SMartin KaFai Lau 			if (len < tcp_saved_syn_len(tp->saved_syn)) {
418134704ef0SMartin KaFai Lau 				len = tcp_saved_syn_len(tp->saved_syn);
418234704ef0SMartin KaFai Lau 				if (copy_to_sockptr(optlen, &len, sizeof(int))) {
4183d51bbff2SMartin KaFai Lau 					sockopt_release_sock(sk);
4184aea0929eSEric B Munson 					return -EFAULT;
4185aea0929eSEric B Munson 				}
4186d51bbff2SMartin KaFai Lau 				sockopt_release_sock(sk);
4187aea0929eSEric B Munson 				return -EINVAL;
4188aea0929eSEric B Munson 			}
418970a217f1SMartin KaFai Lau 			len = tcp_saved_syn_len(tp->saved_syn);
419034704ef0SMartin KaFai Lau 			if (copy_to_sockptr(optlen, &len, sizeof(int))) {
4191d51bbff2SMartin KaFai Lau 				sockopt_release_sock(sk);
4192cd8ae852SEric Dumazet 				return -EFAULT;
4193cd8ae852SEric Dumazet 			}
419434704ef0SMartin KaFai Lau 			if (copy_to_sockptr(optval, tp->saved_syn->data, len)) {
4195d51bbff2SMartin KaFai Lau 				sockopt_release_sock(sk);
4196cd8ae852SEric Dumazet 				return -EFAULT;
4197cd8ae852SEric Dumazet 			}
4198cd8ae852SEric Dumazet 			tcp_saved_syn_free(tp);
4199d51bbff2SMartin KaFai Lau 			sockopt_release_sock(sk);
4200cd8ae852SEric Dumazet 		} else {
4201d51bbff2SMartin KaFai Lau 			sockopt_release_sock(sk);
4202cd8ae852SEric Dumazet 			len = 0;
420334704ef0SMartin KaFai Lau 			if (copy_to_sockptr(optlen, &len, sizeof(int)))
4204cd8ae852SEric Dumazet 				return -EFAULT;
4205cd8ae852SEric Dumazet 		}
4206cd8ae852SEric Dumazet 		return 0;
4207cd8ae852SEric Dumazet 	}
420805255b82SEric Dumazet #ifdef CONFIG_MMU
420905255b82SEric Dumazet 	case TCP_ZEROCOPY_RECEIVE: {
42107eeba170SArjun Roy 		struct scm_timestamping_internal tss;
4211e0fecb28SArjun Roy 		struct tcp_zerocopy_receive zc = {};
421205255b82SEric Dumazet 		int err;
421305255b82SEric Dumazet 
421434704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&len, optlen, sizeof(int)))
421505255b82SEric Dumazet 			return -EFAULT;
42162107d45fSArjun Roy 		if (len < 0 ||
42172107d45fSArjun Roy 		    len < offsetofend(struct tcp_zerocopy_receive, length))
421805255b82SEric Dumazet 			return -EINVAL;
42193c5a2fd0SArjun Roy 		if (unlikely(len > sizeof(zc))) {
422034704ef0SMartin KaFai Lau 			err = check_zeroed_sockptr(optval, sizeof(zc),
42213c5a2fd0SArjun Roy 						   len - sizeof(zc));
42223c5a2fd0SArjun Roy 			if (err < 1)
42233c5a2fd0SArjun Roy 				return err == 0 ? -EINVAL : err;
4224c8856c05SArjun Roy 			len = sizeof(zc);
422534704ef0SMartin KaFai Lau 			if (copy_to_sockptr(optlen, &len, sizeof(int)))
42260b7f41f6SArjun Roy 				return -EFAULT;
42270b7f41f6SArjun Roy 		}
422834704ef0SMartin KaFai Lau 		if (copy_from_sockptr(&zc, optval, len))
422905255b82SEric Dumazet 			return -EFAULT;
42303c5a2fd0SArjun Roy 		if (zc.reserved)
42313c5a2fd0SArjun Roy 			return -EINVAL;
42323c5a2fd0SArjun Roy 		if (zc.msg_flags &  ~(TCP_VALID_ZC_MSG_FLAGS))
42333c5a2fd0SArjun Roy 			return -EINVAL;
4234d51bbff2SMartin KaFai Lau 		sockopt_lock_sock(sk);
42357eeba170SArjun Roy 		err = tcp_zerocopy_receive(sk, &zc, &tss);
42369cacf81fSStanislav Fomichev 		err = BPF_CGROUP_RUN_PROG_GETSOCKOPT_KERN(sk, level, optname,
42379cacf81fSStanislav Fomichev 							  &zc, &len, err);
4238d51bbff2SMartin KaFai Lau 		sockopt_release_sock(sk);
42397eeba170SArjun Roy 		if (len >= offsetofend(struct tcp_zerocopy_receive, msg_flags))
42407eeba170SArjun Roy 			goto zerocopy_rcv_cmsg;
4241c8856c05SArjun Roy 		switch (len) {
42427eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, msg_flags):
42437eeba170SArjun Roy 			goto zerocopy_rcv_cmsg;
42447eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, msg_controllen):
42457eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, msg_control):
42467eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, flags):
42477eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, copybuf_len):
42487eeba170SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, copybuf_address):
424933946518SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, err):
425033946518SArjun Roy 			goto zerocopy_rcv_sk_err;
4251c8856c05SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, inq):
4252c8856c05SArjun Roy 			goto zerocopy_rcv_inq;
4253c8856c05SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, length):
4254c8856c05SArjun Roy 		default:
4255c8856c05SArjun Roy 			goto zerocopy_rcv_out;
4256c8856c05SArjun Roy 		}
42577eeba170SArjun Roy zerocopy_rcv_cmsg:
42587eeba170SArjun Roy 		if (zc.msg_flags & TCP_CMSG_TS)
42597eeba170SArjun Roy 			tcp_zc_finalize_rx_tstamp(sk, &zc, &tss);
42607eeba170SArjun Roy 		else
42617eeba170SArjun Roy 			zc.msg_flags = 0;
426233946518SArjun Roy zerocopy_rcv_sk_err:
426333946518SArjun Roy 		if (!err)
426433946518SArjun Roy 			zc.err = sock_error(sk);
4265c8856c05SArjun Roy zerocopy_rcv_inq:
4266c8856c05SArjun Roy 		zc.inq = tcp_inq_hint(sk);
4267c8856c05SArjun Roy zerocopy_rcv_out:
426834704ef0SMartin KaFai Lau 		if (!err && copy_to_sockptr(optval, &zc, len))
426905255b82SEric Dumazet 			err = -EFAULT;
427005255b82SEric Dumazet 		return err;
427105255b82SEric Dumazet 	}
427205255b82SEric Dumazet #endif
42731da177e4SLinus Torvalds 	default:
42741da177e4SLinus Torvalds 		return -ENOPROTOOPT;
42753ff50b79SStephen Hemminger 	}
42761da177e4SLinus Torvalds 
427734704ef0SMartin KaFai Lau 	if (copy_to_sockptr(optlen, &len, sizeof(int)))
42781da177e4SLinus Torvalds 		return -EFAULT;
427934704ef0SMartin KaFai Lau 	if (copy_to_sockptr(optval, &val, len))
42801da177e4SLinus Torvalds 		return -EFAULT;
42811da177e4SLinus Torvalds 	return 0;
42821da177e4SLinus Torvalds }
42831da177e4SLinus Torvalds 
42849cacf81fSStanislav Fomichev bool tcp_bpf_bypass_getsockopt(int level, int optname)
42859cacf81fSStanislav Fomichev {
42869cacf81fSStanislav Fomichev 	/* TCP do_tcp_getsockopt has optimized getsockopt implementation
42879cacf81fSStanislav Fomichev 	 * to avoid extra socket lock for TCP_ZEROCOPY_RECEIVE.
42889cacf81fSStanislav Fomichev 	 */
42899cacf81fSStanislav Fomichev 	if (level == SOL_TCP && optname == TCP_ZEROCOPY_RECEIVE)
42909cacf81fSStanislav Fomichev 		return true;
42919cacf81fSStanislav Fomichev 
42929cacf81fSStanislav Fomichev 	return false;
42939cacf81fSStanislav Fomichev }
42949cacf81fSStanislav Fomichev EXPORT_SYMBOL(tcp_bpf_bypass_getsockopt);
42959cacf81fSStanislav Fomichev 
42963fdadf7dSDmitry Mishin int tcp_getsockopt(struct sock *sk, int level, int optname, char __user *optval,
42973fdadf7dSDmitry Mishin 		   int __user *optlen)
42983fdadf7dSDmitry Mishin {
42993fdadf7dSDmitry Mishin 	struct inet_connection_sock *icsk = inet_csk(sk);
43003fdadf7dSDmitry Mishin 
43013fdadf7dSDmitry Mishin 	if (level != SOL_TCP)
4302f49cd2f4SKuniyuki Iwashima 		/* Paired with WRITE_ONCE() in do_ipv6_setsockopt() and tcp_v6_connect() */
4303f49cd2f4SKuniyuki Iwashima 		return READ_ONCE(icsk->icsk_af_ops)->getsockopt(sk, level, optname,
43043fdadf7dSDmitry Mishin 								optval, optlen);
430534704ef0SMartin KaFai Lau 	return do_tcp_getsockopt(sk, level, optname, USER_SOCKPTR(optval),
430634704ef0SMartin KaFai Lau 				 USER_SOCKPTR(optlen));
43073fdadf7dSDmitry Mishin }
43084bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_getsockopt);
43093fdadf7dSDmitry Mishin 
4310cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG
4311349ce993SEric Dumazet static DEFINE_PER_CPU(struct tcp_md5sig_pool, tcp_md5sig_pool);
431271cea17eSEric Dumazet static DEFINE_MUTEX(tcp_md5sig_mutex);
4313349ce993SEric Dumazet static bool tcp_md5sig_pool_populated = false;
4314cfb6eeb4SYOSHIFUJI Hideaki 
431571cea17eSEric Dumazet static void __tcp_alloc_md5sig_pool(void)
4316cfb6eeb4SYOSHIFUJI Hideaki {
4317cf80e0e4SHerbert Xu 	struct crypto_ahash *hash;
4318cfb6eeb4SYOSHIFUJI Hideaki 	int cpu;
4319cfb6eeb4SYOSHIFUJI Hideaki 
4320cf80e0e4SHerbert Xu 	hash = crypto_alloc_ahash("md5", 0, CRYPTO_ALG_ASYNC);
43211eea84b7SInsu Yun 	if (IS_ERR(hash))
4322349ce993SEric Dumazet 		return;
4323cf80e0e4SHerbert Xu 
4324cf80e0e4SHerbert Xu 	for_each_possible_cpu(cpu) {
432519689e38SEric Dumazet 		void *scratch = per_cpu(tcp_md5sig_pool, cpu).scratch;
4326cf80e0e4SHerbert Xu 		struct ahash_request *req;
4327cf80e0e4SHerbert Xu 
432819689e38SEric Dumazet 		if (!scratch) {
432919689e38SEric Dumazet 			scratch = kmalloc_node(sizeof(union tcp_md5sum_block) +
433019689e38SEric Dumazet 					       sizeof(struct tcphdr),
433119689e38SEric Dumazet 					       GFP_KERNEL,
433219689e38SEric Dumazet 					       cpu_to_node(cpu));
433319689e38SEric Dumazet 			if (!scratch)
433419689e38SEric Dumazet 				return;
433519689e38SEric Dumazet 			per_cpu(tcp_md5sig_pool, cpu).scratch = scratch;
433619689e38SEric Dumazet 		}
4337cf80e0e4SHerbert Xu 		if (per_cpu(tcp_md5sig_pool, cpu).md5_req)
4338cf80e0e4SHerbert Xu 			continue;
4339cf80e0e4SHerbert Xu 
4340cf80e0e4SHerbert Xu 		req = ahash_request_alloc(hash, GFP_KERNEL);
4341cf80e0e4SHerbert Xu 		if (!req)
4342cf80e0e4SHerbert Xu 			return;
4343cf80e0e4SHerbert Xu 
4344cf80e0e4SHerbert Xu 		ahash_request_set_callback(req, 0, NULL, NULL);
4345cf80e0e4SHerbert Xu 
4346cf80e0e4SHerbert Xu 		per_cpu(tcp_md5sig_pool, cpu).md5_req = req;
4347349ce993SEric Dumazet 	}
4348349ce993SEric Dumazet 	/* before setting tcp_md5sig_pool_populated, we must commit all writes
4349349ce993SEric Dumazet 	 * to memory. See smp_rmb() in tcp_get_md5sig_pool()
435071cea17eSEric Dumazet 	 */
435171cea17eSEric Dumazet 	smp_wmb();
4352aacd467cSEric Dumazet 	/* Paired with READ_ONCE() from tcp_alloc_md5sig_pool()
4353aacd467cSEric Dumazet 	 * and tcp_get_md5sig_pool().
4354aacd467cSEric Dumazet 	*/
4355aacd467cSEric Dumazet 	WRITE_ONCE(tcp_md5sig_pool_populated, true);
4356cfb6eeb4SYOSHIFUJI Hideaki }
4357cfb6eeb4SYOSHIFUJI Hideaki 
435871cea17eSEric Dumazet bool tcp_alloc_md5sig_pool(void)
4359cfb6eeb4SYOSHIFUJI Hideaki {
4360aacd467cSEric Dumazet 	/* Paired with WRITE_ONCE() from __tcp_alloc_md5sig_pool() */
4361aacd467cSEric Dumazet 	if (unlikely(!READ_ONCE(tcp_md5sig_pool_populated))) {
436271cea17eSEric Dumazet 		mutex_lock(&tcp_md5sig_mutex);
4363cfb6eeb4SYOSHIFUJI Hideaki 
4364459837b5SDmitry Safonov 		if (!tcp_md5sig_pool_populated)
436571cea17eSEric Dumazet 			__tcp_alloc_md5sig_pool();
4366cfb6eeb4SYOSHIFUJI Hideaki 
436771cea17eSEric Dumazet 		mutex_unlock(&tcp_md5sig_mutex);
4368cfb6eeb4SYOSHIFUJI Hideaki 	}
4369aacd467cSEric Dumazet 	/* Paired with WRITE_ONCE() from __tcp_alloc_md5sig_pool() */
4370aacd467cSEric Dumazet 	return READ_ONCE(tcp_md5sig_pool_populated);
4371cfb6eeb4SYOSHIFUJI Hideaki }
4372cfb6eeb4SYOSHIFUJI Hideaki EXPORT_SYMBOL(tcp_alloc_md5sig_pool);
4373cfb6eeb4SYOSHIFUJI Hideaki 
437435790c04SEric Dumazet 
437535790c04SEric Dumazet /**
437635790c04SEric Dumazet  *	tcp_get_md5sig_pool - get md5sig_pool for this user
437735790c04SEric Dumazet  *
437835790c04SEric Dumazet  *	We use percpu structure, so if we succeed, we exit with preemption
437935790c04SEric Dumazet  *	and BH disabled, to make sure another thread or softirq handling
438035790c04SEric Dumazet  *	wont try to get same context.
438135790c04SEric Dumazet  */
438235790c04SEric Dumazet struct tcp_md5sig_pool *tcp_get_md5sig_pool(void)
4383cfb6eeb4SYOSHIFUJI Hideaki {
438435790c04SEric Dumazet 	local_bh_disable();
438535790c04SEric Dumazet 
4386aacd467cSEric Dumazet 	/* Paired with WRITE_ONCE() from __tcp_alloc_md5sig_pool() */
4387aacd467cSEric Dumazet 	if (READ_ONCE(tcp_md5sig_pool_populated)) {
4388349ce993SEric Dumazet 		/* coupled with smp_wmb() in __tcp_alloc_md5sig_pool() */
4389349ce993SEric Dumazet 		smp_rmb();
4390349ce993SEric Dumazet 		return this_cpu_ptr(&tcp_md5sig_pool);
4391349ce993SEric Dumazet 	}
439235790c04SEric Dumazet 	local_bh_enable();
439335790c04SEric Dumazet 	return NULL;
4394cfb6eeb4SYOSHIFUJI Hideaki }
439535790c04SEric Dumazet EXPORT_SYMBOL(tcp_get_md5sig_pool);
4396cfb6eeb4SYOSHIFUJI Hideaki 
439749a72dfbSAdam Langley int tcp_md5_hash_skb_data(struct tcp_md5sig_pool *hp,
4398cf533ea5SEric Dumazet 			  const struct sk_buff *skb, unsigned int header_len)
439949a72dfbSAdam Langley {
440049a72dfbSAdam Langley 	struct scatterlist sg;
440149a72dfbSAdam Langley 	const struct tcphdr *tp = tcp_hdr(skb);
4402cf80e0e4SHerbert Xu 	struct ahash_request *req = hp->md5_req;
440395c96174SEric Dumazet 	unsigned int i;
440495c96174SEric Dumazet 	const unsigned int head_data_len = skb_headlen(skb) > header_len ?
440549a72dfbSAdam Langley 					   skb_headlen(skb) - header_len : 0;
440649a72dfbSAdam Langley 	const struct skb_shared_info *shi = skb_shinfo(skb);
4407d7fd1b57SEric Dumazet 	struct sk_buff *frag_iter;
440849a72dfbSAdam Langley 
440949a72dfbSAdam Langley 	sg_init_table(&sg, 1);
441049a72dfbSAdam Langley 
441149a72dfbSAdam Langley 	sg_set_buf(&sg, ((u8 *) tp) + header_len, head_data_len);
4412cf80e0e4SHerbert Xu 	ahash_request_set_crypt(req, &sg, NULL, head_data_len);
4413cf80e0e4SHerbert Xu 	if (crypto_ahash_update(req))
441449a72dfbSAdam Langley 		return 1;
441549a72dfbSAdam Langley 
441649a72dfbSAdam Langley 	for (i = 0; i < shi->nr_frags; ++i) {
4417d8e18a51SMatthew Wilcox (Oracle) 		const skb_frag_t *f = &shi->frags[i];
4418b54c9d5bSJonathan Lemon 		unsigned int offset = skb_frag_off(f);
441954d27fcbSEric Dumazet 		struct page *page = skb_frag_page(f) + (offset >> PAGE_SHIFT);
442054d27fcbSEric Dumazet 
442154d27fcbSEric Dumazet 		sg_set_page(&sg, page, skb_frag_size(f),
442254d27fcbSEric Dumazet 			    offset_in_page(offset));
4423cf80e0e4SHerbert Xu 		ahash_request_set_crypt(req, &sg, NULL, skb_frag_size(f));
4424cf80e0e4SHerbert Xu 		if (crypto_ahash_update(req))
442549a72dfbSAdam Langley 			return 1;
442649a72dfbSAdam Langley 	}
442749a72dfbSAdam Langley 
4428d7fd1b57SEric Dumazet 	skb_walk_frags(skb, frag_iter)
4429d7fd1b57SEric Dumazet 		if (tcp_md5_hash_skb_data(hp, frag_iter, 0))
4430d7fd1b57SEric Dumazet 			return 1;
4431d7fd1b57SEric Dumazet 
443249a72dfbSAdam Langley 	return 0;
443349a72dfbSAdam Langley }
443449a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_skb_data);
443549a72dfbSAdam Langley 
4436cf533ea5SEric Dumazet int tcp_md5_hash_key(struct tcp_md5sig_pool *hp, const struct tcp_md5sig_key *key)
443749a72dfbSAdam Langley {
4438e6ced831SEric Dumazet 	u8 keylen = READ_ONCE(key->keylen); /* paired with WRITE_ONCE() in tcp_md5_do_add */
443949a72dfbSAdam Langley 	struct scatterlist sg;
444049a72dfbSAdam Langley 
44416a2febecSEric Dumazet 	sg_init_one(&sg, key->key, keylen);
44426a2febecSEric Dumazet 	ahash_request_set_crypt(hp->md5_req, &sg, NULL, keylen);
4443e6ced831SEric Dumazet 
4444e6ced831SEric Dumazet 	/* We use data_race() because tcp_md5_do_add() might change key->key under us */
4445e6ced831SEric Dumazet 	return data_race(crypto_ahash_update(hp->md5_req));
444649a72dfbSAdam Langley }
444749a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_key);
444849a72dfbSAdam Langley 
44497bbb765bSDmitry Safonov /* Called with rcu_read_lock() */
44501330b6efSJakub Kicinski enum skb_drop_reason
44511330b6efSJakub Kicinski tcp_inbound_md5_hash(const struct sock *sk, const struct sk_buff *skb,
44527bbb765bSDmitry Safonov 		     const void *saddr, const void *daddr,
44537bbb765bSDmitry Safonov 		     int family, int dif, int sdif)
44547bbb765bSDmitry Safonov {
44557bbb765bSDmitry Safonov 	/*
44567bbb765bSDmitry Safonov 	 * This gets called for each TCP segment that arrives
44577bbb765bSDmitry Safonov 	 * so we want to be efficient.
44587bbb765bSDmitry Safonov 	 * We have 3 drop cases:
44597bbb765bSDmitry Safonov 	 * o No MD5 hash and one expected.
44607bbb765bSDmitry Safonov 	 * o MD5 hash and we're not expecting one.
44617bbb765bSDmitry Safonov 	 * o MD5 hash and its wrong.
44627bbb765bSDmitry Safonov 	 */
44637bbb765bSDmitry Safonov 	const __u8 *hash_location = NULL;
44647bbb765bSDmitry Safonov 	struct tcp_md5sig_key *hash_expected;
44657bbb765bSDmitry Safonov 	const struct tcphdr *th = tcp_hdr(skb);
4466e9d9da91SEric Dumazet 	const struct tcp_sock *tp = tcp_sk(sk);
44677bbb765bSDmitry Safonov 	int genhash, l3index;
44687bbb765bSDmitry Safonov 	u8 newhash[16];
44697bbb765bSDmitry Safonov 
44707bbb765bSDmitry Safonov 	/* sdif set, means packet ingressed via a device
44717bbb765bSDmitry Safonov 	 * in an L3 domain and dif is set to the l3mdev
44727bbb765bSDmitry Safonov 	 */
44737bbb765bSDmitry Safonov 	l3index = sdif ? dif : 0;
44747bbb765bSDmitry Safonov 
44757bbb765bSDmitry Safonov 	hash_expected = tcp_md5_do_lookup(sk, l3index, saddr, family);
44767bbb765bSDmitry Safonov 	hash_location = tcp_parse_md5sig_option(th);
44777bbb765bSDmitry Safonov 
44787bbb765bSDmitry Safonov 	/* We've parsed the options - do we have a hash? */
44797bbb765bSDmitry Safonov 	if (!hash_expected && !hash_location)
44801330b6efSJakub Kicinski 		return SKB_NOT_DROPPED_YET;
44817bbb765bSDmitry Safonov 
44827bbb765bSDmitry Safonov 	if (hash_expected && !hash_location) {
44837bbb765bSDmitry Safonov 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMD5NOTFOUND);
44841330b6efSJakub Kicinski 		return SKB_DROP_REASON_TCP_MD5NOTFOUND;
44857bbb765bSDmitry Safonov 	}
44867bbb765bSDmitry Safonov 
44877bbb765bSDmitry Safonov 	if (!hash_expected && hash_location) {
44887bbb765bSDmitry Safonov 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMD5UNEXPECTED);
44891330b6efSJakub Kicinski 		return SKB_DROP_REASON_TCP_MD5UNEXPECTED;
44907bbb765bSDmitry Safonov 	}
44917bbb765bSDmitry Safonov 
4492e62d2e11SEric Dumazet 	/* Check the signature.
4493e62d2e11SEric Dumazet 	 * To support dual stack listeners, we need to handle
4494e62d2e11SEric Dumazet 	 * IPv4-mapped case.
4495e62d2e11SEric Dumazet 	 */
4496e62d2e11SEric Dumazet 	if (family == AF_INET)
4497e62d2e11SEric Dumazet 		genhash = tcp_v4_md5_hash_skb(newhash,
4498e62d2e11SEric Dumazet 					      hash_expected,
4499e62d2e11SEric Dumazet 					      NULL, skb);
4500e62d2e11SEric Dumazet 	else
4501e62d2e11SEric Dumazet 		genhash = tp->af_specific->calc_md5_hash(newhash,
4502e62d2e11SEric Dumazet 							 hash_expected,
45037bbb765bSDmitry Safonov 							 NULL, skb);
45047bbb765bSDmitry Safonov 
45057bbb765bSDmitry Safonov 	if (genhash || memcmp(hash_location, newhash, 16) != 0) {
45067bbb765bSDmitry Safonov 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMD5FAILURE);
45077bbb765bSDmitry Safonov 		if (family == AF_INET) {
45087bbb765bSDmitry Safonov 			net_info_ratelimited("MD5 Hash failed for (%pI4, %d)->(%pI4, %d)%s L3 index %d\n",
45097bbb765bSDmitry Safonov 					saddr, ntohs(th->source),
45107bbb765bSDmitry Safonov 					daddr, ntohs(th->dest),
45117bbb765bSDmitry Safonov 					genhash ? " tcp_v4_calc_md5_hash failed"
45127bbb765bSDmitry Safonov 					: "", l3index);
45137bbb765bSDmitry Safonov 		} else {
45147bbb765bSDmitry Safonov 			net_info_ratelimited("MD5 Hash %s for [%pI6c]:%u->[%pI6c]:%u L3 index %d\n",
45157bbb765bSDmitry Safonov 					genhash ? "failed" : "mismatch",
45167bbb765bSDmitry Safonov 					saddr, ntohs(th->source),
45177bbb765bSDmitry Safonov 					daddr, ntohs(th->dest), l3index);
45187bbb765bSDmitry Safonov 		}
45191330b6efSJakub Kicinski 		return SKB_DROP_REASON_TCP_MD5FAILURE;
45207bbb765bSDmitry Safonov 	}
45211330b6efSJakub Kicinski 	return SKB_NOT_DROPPED_YET;
45227bbb765bSDmitry Safonov }
45237bbb765bSDmitry Safonov EXPORT_SYMBOL(tcp_inbound_md5_hash);
45247bbb765bSDmitry Safonov 
4525cfb6eeb4SYOSHIFUJI Hideaki #endif
4526cfb6eeb4SYOSHIFUJI Hideaki 
45274ac02babSAndi Kleen void tcp_done(struct sock *sk)
45284ac02babSAndi Kleen {
4529d983ea6fSEric Dumazet 	struct request_sock *req;
45308336886fSJerry Chu 
4531cab209e5SEric Dumazet 	/* We might be called with a new socket, after
4532cab209e5SEric Dumazet 	 * inet_csk_prepare_forced_close() has been called
4533cab209e5SEric Dumazet 	 * so we can not use lockdep_sock_is_held(sk)
4534cab209e5SEric Dumazet 	 */
4535cab209e5SEric Dumazet 	req = rcu_dereference_protected(tcp_sk(sk)->fastopen_rsk, 1);
45364ac02babSAndi Kleen 
45374ac02babSAndi Kleen 	if (sk->sk_state == TCP_SYN_SENT || sk->sk_state == TCP_SYN_RECV)
4538c10d9310SEric Dumazet 		TCP_INC_STATS(sock_net(sk), TCP_MIB_ATTEMPTFAILS);
45394ac02babSAndi Kleen 
45404ac02babSAndi Kleen 	tcp_set_state(sk, TCP_CLOSE);
45414ac02babSAndi Kleen 	tcp_clear_xmit_timers(sk);
454200db4124SIan Morris 	if (req)
45438336886fSJerry Chu 		reqsk_fastopen_remove(sk, req, false);
45444ac02babSAndi Kleen 
4545e14cadfdSEric Dumazet 	WRITE_ONCE(sk->sk_shutdown, SHUTDOWN_MASK);
45464ac02babSAndi Kleen 
45474ac02babSAndi Kleen 	if (!sock_flag(sk, SOCK_DEAD))
45484ac02babSAndi Kleen 		sk->sk_state_change(sk);
45494ac02babSAndi Kleen 	else
45504ac02babSAndi Kleen 		inet_csk_destroy_sock(sk);
45514ac02babSAndi Kleen }
45524ac02babSAndi Kleen EXPORT_SYMBOL_GPL(tcp_done);
45534ac02babSAndi Kleen 
4554c1e64e29SLorenzo Colitti int tcp_abort(struct sock *sk, int err)
4555c1e64e29SLorenzo Colitti {
4556af9784d0SEric Dumazet 	int state = inet_sk_state_load(sk);
4557af9784d0SEric Dumazet 
4558af9784d0SEric Dumazet 	if (state == TCP_NEW_SYN_RECV) {
455907f6f4a3SEric Dumazet 		struct request_sock *req = inet_reqsk(sk);
456007f6f4a3SEric Dumazet 
456107f6f4a3SEric Dumazet 		local_bh_disable();
4562acc2cf4eSLorenzo Colitti 		inet_csk_reqsk_queue_drop(req->rsk_listener, req);
456307f6f4a3SEric Dumazet 		local_bh_enable();
456407f6f4a3SEric Dumazet 		return 0;
456507f6f4a3SEric Dumazet 	}
4566af9784d0SEric Dumazet 	if (state == TCP_TIME_WAIT) {
4567af9784d0SEric Dumazet 		struct inet_timewait_sock *tw = inet_twsk(sk);
4568af9784d0SEric Dumazet 
4569af9784d0SEric Dumazet 		refcount_inc(&tw->tw_refcnt);
4570af9784d0SEric Dumazet 		local_bh_disable();
4571af9784d0SEric Dumazet 		inet_twsk_deschedule_put(tw);
4572af9784d0SEric Dumazet 		local_bh_enable();
4573af9784d0SEric Dumazet 		return 0;
4574c1e64e29SLorenzo Colitti 	}
4575c1e64e29SLorenzo Colitti 
45764ddbcb88SAditi Ghag 	/* BPF context ensures sock locking. */
45774ddbcb88SAditi Ghag 	if (!has_current_bpf_ctx())
4578c1e64e29SLorenzo Colitti 		/* Don't race with userspace socket closes such as tcp_close. */
4579c1e64e29SLorenzo Colitti 		lock_sock(sk);
4580c1e64e29SLorenzo Colitti 
45812010b93eSLorenzo Colitti 	if (sk->sk_state == TCP_LISTEN) {
45822010b93eSLorenzo Colitti 		tcp_set_state(sk, TCP_CLOSE);
45832010b93eSLorenzo Colitti 		inet_csk_listen_stop(sk);
45842010b93eSLorenzo Colitti 	}
45852010b93eSLorenzo Colitti 
4586c1e64e29SLorenzo Colitti 	/* Don't race with BH socket closes such as inet_csk_listen_stop. */
4587c1e64e29SLorenzo Colitti 	local_bh_disable();
4588c1e64e29SLorenzo Colitti 	bh_lock_sock(sk);
4589c1e64e29SLorenzo Colitti 
4590c1e64e29SLorenzo Colitti 	if (!sock_flag(sk, SOCK_DEAD)) {
4591e13ec3daSEric Dumazet 		WRITE_ONCE(sk->sk_err, err);
4592c1e64e29SLorenzo Colitti 		/* This barrier is coupled with smp_rmb() in tcp_poll() */
4593c1e64e29SLorenzo Colitti 		smp_wmb();
4594e3ae2365SAlexander Aring 		sk_error_report(sk);
4595c1e64e29SLorenzo Colitti 		if (tcp_need_reset(sk->sk_state))
4596c1e64e29SLorenzo Colitti 			tcp_send_active_reset(sk, GFP_ATOMIC);
4597c1e64e29SLorenzo Colitti 		tcp_done(sk);
4598c1e64e29SLorenzo Colitti 	}
4599c1e64e29SLorenzo Colitti 
4600c1e64e29SLorenzo Colitti 	bh_unlock_sock(sk);
4601c1e64e29SLorenzo Colitti 	local_bh_enable();
4602e05836acSSoheil Hassas Yeganeh 	tcp_write_queue_purge(sk);
46034ddbcb88SAditi Ghag 	if (!has_current_bpf_ctx())
4604c1e64e29SLorenzo Colitti 		release_sock(sk);
4605c1e64e29SLorenzo Colitti 	return 0;
4606c1e64e29SLorenzo Colitti }
4607c1e64e29SLorenzo Colitti EXPORT_SYMBOL_GPL(tcp_abort);
4608c1e64e29SLorenzo Colitti 
46095f8ef48dSStephen Hemminger extern struct tcp_congestion_ops tcp_reno;
46101da177e4SLinus Torvalds 
46111da177e4SLinus Torvalds static __initdata unsigned long thash_entries;
46121da177e4SLinus Torvalds static int __init set_thash_entries(char *str)
46131da177e4SLinus Torvalds {
4614413c27d8SEldad Zack 	ssize_t ret;
4615413c27d8SEldad Zack 
46161da177e4SLinus Torvalds 	if (!str)
46171da177e4SLinus Torvalds 		return 0;
4618413c27d8SEldad Zack 
4619413c27d8SEldad Zack 	ret = kstrtoul(str, 0, &thash_entries);
4620413c27d8SEldad Zack 	if (ret)
4621413c27d8SEldad Zack 		return 0;
4622413c27d8SEldad Zack 
46231da177e4SLinus Torvalds 	return 1;
46241da177e4SLinus Torvalds }
46251da177e4SLinus Torvalds __setup("thash_entries=", set_thash_entries);
46261da177e4SLinus Torvalds 
462747d7a88cSFabian Frederick static void __init tcp_init_mem(void)
46284acb4190SGlauber Costa {
4629b66e91ccSEric Dumazet 	unsigned long limit = nr_free_buffer_pages() / 16;
4630b66e91ccSEric Dumazet 
46314acb4190SGlauber Costa 	limit = max(limit, 128UL);
4632b66e91ccSEric Dumazet 	sysctl_tcp_mem[0] = limit / 4 * 3;		/* 4.68 % */
4633b66e91ccSEric Dumazet 	sysctl_tcp_mem[1] = limit;			/* 6.25 % */
4634b66e91ccSEric Dumazet 	sysctl_tcp_mem[2] = sysctl_tcp_mem[0] * 2;	/* 9.37 % */
46354acb4190SGlauber Costa }
46364acb4190SGlauber Costa 
46371da177e4SLinus Torvalds void __init tcp_init(void)
46381da177e4SLinus Torvalds {
4639b49960a0SEric Dumazet 	int max_rshare, max_wshare, cnt;
4640b2d3ea4aSEric Dumazet 	unsigned long limit;
4641074b8517SDimitri Sivanich 	unsigned int i;
46421da177e4SLinus Torvalds 
46433b4929f6SEric Dumazet 	BUILD_BUG_ON(TCP_MIN_SND_MSS <= MAX_TCP_OPTION_SPACE);
4644b2d3ea4aSEric Dumazet 	BUILD_BUG_ON(sizeof(struct tcp_skb_cb) >
4645c593642cSPankaj Bharadiya 		     sizeof_field(struct sk_buff, cb));
46461da177e4SLinus Torvalds 
4647908c7f19STejun Heo 	percpu_counter_init(&tcp_sockets_allocated, 0, GFP_KERNEL);
464819757cebSEric Dumazet 
464919757cebSEric Dumazet 	timer_setup(&tcp_orphan_timer, tcp_orphan_update, TIMER_DEFERRABLE);
465019757cebSEric Dumazet 	mod_timer(&tcp_orphan_timer, jiffies + TCP_ORPHAN_TIMER_PERIOD);
465119757cebSEric Dumazet 
465227da6d37SMartin KaFai Lau 	inet_hashinfo2_init(&tcp_hashinfo, "tcp_listen_portaddr_hash",
465327da6d37SMartin KaFai Lau 			    thash_entries, 21,  /* one slot per 2 MB*/
465427da6d37SMartin KaFai Lau 			    0, 64 * 1024);
46556e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.bind_bucket_cachep =
46566e04e021SArnaldo Carvalho de Melo 		kmem_cache_create("tcp_bind_bucket",
46576e04e021SArnaldo Carvalho de Melo 				  sizeof(struct inet_bind_bucket), 0,
4658990c74e3SVasily Averin 				  SLAB_HWCACHE_ALIGN | SLAB_PANIC |
4659990c74e3SVasily Averin 				  SLAB_ACCOUNT,
4660990c74e3SVasily Averin 				  NULL);
466128044fc1SJoanne Koong 	tcp_hashinfo.bind2_bucket_cachep =
466228044fc1SJoanne Koong 		kmem_cache_create("tcp_bind2_bucket",
466328044fc1SJoanne Koong 				  sizeof(struct inet_bind2_bucket), 0,
466428044fc1SJoanne Koong 				  SLAB_HWCACHE_ALIGN | SLAB_PANIC |
466528044fc1SJoanne Koong 				  SLAB_ACCOUNT,
466628044fc1SJoanne Koong 				  NULL);
46671da177e4SLinus Torvalds 
46681da177e4SLinus Torvalds 	/* Size and allocate the main established and bind bucket
46691da177e4SLinus Torvalds 	 * hash tables.
46701da177e4SLinus Torvalds 	 *
46711da177e4SLinus Torvalds 	 * The methodology is similar to that of the buffer cache.
46721da177e4SLinus Torvalds 	 */
46736e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.ehash =
46741da177e4SLinus Torvalds 		alloc_large_system_hash("TCP established",
46750f7ff927SArnaldo Carvalho de Melo 					sizeof(struct inet_ehash_bucket),
46761da177e4SLinus Torvalds 					thash_entries,
4677fd90b29dSEric Dumazet 					17, /* one slot per 128 KB of memory */
46789e950efaSJohn Heffner 					0,
46791da177e4SLinus Torvalds 					NULL,
4680f373b53bSEric Dumazet 					&tcp_hashinfo.ehash_mask,
468131fe62b9STim Bird 					0,
46820ccfe618SJean Delvare 					thash_entries ? 0 : 512 * 1024);
468305dbc7b5SEric Dumazet 	for (i = 0; i <= tcp_hashinfo.ehash_mask; i++)
46843ab5aee7SEric Dumazet 		INIT_HLIST_NULLS_HEAD(&tcp_hashinfo.ehash[i].chain, i);
468505dbc7b5SEric Dumazet 
4686230140cfSEric Dumazet 	if (inet_ehash_locks_alloc(&tcp_hashinfo))
4687230140cfSEric Dumazet 		panic("TCP: failed to alloc ehash_locks");
46886e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.bhash =
4689593d1ebeSJoanne Koong 		alloc_large_system_hash("TCP bind",
469028044fc1SJoanne Koong 					2 * sizeof(struct inet_bind_hashbucket),
4691f373b53bSEric Dumazet 					tcp_hashinfo.ehash_mask + 1,
4692fd90b29dSEric Dumazet 					17, /* one slot per 128 KB of memory */
46939e950efaSJohn Heffner 					0,
46946e04e021SArnaldo Carvalho de Melo 					&tcp_hashinfo.bhash_size,
46951da177e4SLinus Torvalds 					NULL,
469631fe62b9STim Bird 					0,
46971da177e4SLinus Torvalds 					64 * 1024);
4698074b8517SDimitri Sivanich 	tcp_hashinfo.bhash_size = 1U << tcp_hashinfo.bhash_size;
469928044fc1SJoanne Koong 	tcp_hashinfo.bhash2 = tcp_hashinfo.bhash + tcp_hashinfo.bhash_size;
47006e04e021SArnaldo Carvalho de Melo 	for (i = 0; i < tcp_hashinfo.bhash_size; i++) {
47016e04e021SArnaldo Carvalho de Melo 		spin_lock_init(&tcp_hashinfo.bhash[i].lock);
47026e04e021SArnaldo Carvalho de Melo 		INIT_HLIST_HEAD(&tcp_hashinfo.bhash[i].chain);
470328044fc1SJoanne Koong 		spin_lock_init(&tcp_hashinfo.bhash2[i].lock);
470428044fc1SJoanne Koong 		INIT_HLIST_HEAD(&tcp_hashinfo.bhash2[i].chain);
47051da177e4SLinus Torvalds 	}
47061da177e4SLinus Torvalds 
4707d1e5e640SKuniyuki Iwashima 	tcp_hashinfo.pernet = false;
4708c5ed63d6SEric Dumazet 
4709c5ed63d6SEric Dumazet 	cnt = tcp_hashinfo.ehash_mask + 1;
4710c5ed63d6SEric Dumazet 	sysctl_tcp_max_orphans = cnt / 2;
47111da177e4SLinus Torvalds 
4712a4fe34bfSEric W. Biederman 	tcp_init_mem();
4713c43b874dSJason Wang 	/* Set per-socket limits to no more than 1/128 the pressure threshold */
47145fb84b14SEric Dumazet 	limit = nr_free_buffer_pages() << (PAGE_SHIFT - 7);
4715b49960a0SEric Dumazet 	max_wshare = min(4UL*1024*1024, limit);
4716b49960a0SEric Dumazet 	max_rshare = min(6UL*1024*1024, limit);
47177b4f4b5eSJohn Heffner 
4718100fdd1fSEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[0] = PAGE_SIZE;
4719356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[1] = 16*1024;
4720356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[2] = max(64*1024, max_wshare);
47217b4f4b5eSJohn Heffner 
4722100fdd1fSEric Dumazet 	init_net.ipv4.sysctl_tcp_rmem[0] = PAGE_SIZE;
4723a337531bSYuchung Cheng 	init_net.ipv4.sysctl_tcp_rmem[1] = 131072;
4724a337531bSYuchung Cheng 	init_net.ipv4.sysctl_tcp_rmem[2] = max(131072, max_rshare);
47251da177e4SLinus Torvalds 
4726afd46503SJoe Perches 	pr_info("Hash tables configured (established %u bind %u)\n",
4727f373b53bSEric Dumazet 		tcp_hashinfo.ehash_mask + 1, tcp_hashinfo.bhash_size);
4728317a76f9SStephen Hemminger 
47291946e672SHaishuang Yan 	tcp_v4_init();
473051c5d0c4SDavid S. Miller 	tcp_metrics_init();
473155d8694fSFlorian Westphal 	BUG_ON(tcp_register_congestion_control(&tcp_reno) != 0);
473246d3ceabSEric Dumazet 	tcp_tasklet_init();
4733f870fa0bSMat Martineau 	mptcp_init();
47341da177e4SLinus Torvalds }
4735