xref: /linux/net/ipv4/tcp.c (revision afb83012cc7236c8f5cefbd0fd4ba628ec34ce02)
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>
26357413ebcSMiquel van Smoorenburg #include <linux/swap.h>
264b8059eadSDavid S. Miller #include <linux/cache.h>
265f4c50d99SHerbert Xu #include <linux/err.h>
266da5c78c8SWilliam Allen Simpson #include <linux/time.h>
2675a0e3ad6STejun Heo #include <linux/slab.h>
26898aaa913SMike Maloney #include <linux/errqueue.h>
26960e2a778SUrsula Braun #include <linux/static_key.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 
283dd24c001SEric Dumazet struct percpu_counter tcp_orphan_count;
2840a5578cfSArnaldo Carvalho de Melo EXPORT_SYMBOL_GPL(tcp_orphan_count);
2850a5578cfSArnaldo Carvalho de Melo 
286a4fe34bfSEric W. Biederman long sysctl_tcp_mem[3] __read_mostly;
287a4fe34bfSEric W. Biederman EXPORT_SYMBOL(sysctl_tcp_mem);
2881da177e4SLinus Torvalds 
2898d987e5cSEric Dumazet atomic_long_t tcp_memory_allocated;	/* Current allocated memory. */
2901da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_memory_allocated);
2911748376bSEric Dumazet 
29260e2a778SUrsula Braun #if IS_ENABLED(CONFIG_SMC)
29360e2a778SUrsula Braun DEFINE_STATIC_KEY_FALSE(tcp_have_smc);
29460e2a778SUrsula Braun EXPORT_SYMBOL(tcp_have_smc);
29560e2a778SUrsula Braun #endif
29660e2a778SUrsula Braun 
2971748376bSEric Dumazet /*
2981748376bSEric Dumazet  * Current number of TCP sockets.
2991748376bSEric Dumazet  */
3001748376bSEric Dumazet struct percpu_counter tcp_sockets_allocated;
3011da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_sockets_allocated);
3021da177e4SLinus Torvalds 
3031da177e4SLinus Torvalds /*
3049c55e01cSJens Axboe  * TCP splice context
3059c55e01cSJens Axboe  */
3069c55e01cSJens Axboe struct tcp_splice_state {
3079c55e01cSJens Axboe 	struct pipe_inode_info *pipe;
3089c55e01cSJens Axboe 	size_t len;
3099c55e01cSJens Axboe 	unsigned int flags;
3109c55e01cSJens Axboe };
3119c55e01cSJens Axboe 
3129c55e01cSJens Axboe /*
3131da177e4SLinus Torvalds  * Pressure flag: try to collapse.
3141da177e4SLinus Torvalds  * Technical note: it is used by multiple contexts non atomically.
3153ab224beSHideo Aoki  * All the __sk_mem_schedule() is of this nature: accounting
3161da177e4SLinus Torvalds  * is strict, actions are advisory and have some latency.
3171da177e4SLinus Torvalds  */
31806044751SEric Dumazet unsigned long tcp_memory_pressure __read_mostly;
31906044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_memory_pressure);
3201da177e4SLinus Torvalds 
3212e05fcaeSEric Dumazet DEFINE_STATIC_KEY_FALSE(tcp_rx_skb_cache_key);
3222e05fcaeSEric Dumazet EXPORT_SYMBOL(tcp_rx_skb_cache_key);
3232e05fcaeSEric Dumazet 
3242e05fcaeSEric Dumazet DEFINE_STATIC_KEY_FALSE(tcp_tx_skb_cache_key);
3252e05fcaeSEric Dumazet 
3265c52ba17SPavel Emelyanov void tcp_enter_memory_pressure(struct sock *sk)
3271da177e4SLinus Torvalds {
32806044751SEric Dumazet 	unsigned long val;
32906044751SEric Dumazet 
3301f142c17SEric Dumazet 	if (READ_ONCE(tcp_memory_pressure))
33106044751SEric Dumazet 		return;
33206044751SEric Dumazet 	val = jiffies;
33306044751SEric Dumazet 
33406044751SEric Dumazet 	if (!val)
33506044751SEric Dumazet 		val--;
33606044751SEric Dumazet 	if (!cmpxchg(&tcp_memory_pressure, 0, val))
3374e673444SPavel Emelyanov 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURES);
3381da177e4SLinus Torvalds }
33906044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_enter_memory_pressure);
34006044751SEric Dumazet 
34106044751SEric Dumazet void tcp_leave_memory_pressure(struct sock *sk)
34206044751SEric Dumazet {
34306044751SEric Dumazet 	unsigned long val;
34406044751SEric Dumazet 
3451f142c17SEric Dumazet 	if (!READ_ONCE(tcp_memory_pressure))
34606044751SEric Dumazet 		return;
34706044751SEric Dumazet 	val = xchg(&tcp_memory_pressure, 0);
34806044751SEric Dumazet 	if (val)
34906044751SEric Dumazet 		NET_ADD_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURESCHRONO,
35006044751SEric Dumazet 			      jiffies_to_msecs(jiffies - val));
3511da177e4SLinus Torvalds }
35206044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_leave_memory_pressure);
3531da177e4SLinus Torvalds 
354b103cf34SJulian Anastasov /* Convert seconds to retransmits based on initial and max timeout */
355b103cf34SJulian Anastasov static u8 secs_to_retrans(int seconds, int timeout, int rto_max)
356b103cf34SJulian Anastasov {
357b103cf34SJulian Anastasov 	u8 res = 0;
358b103cf34SJulian Anastasov 
359b103cf34SJulian Anastasov 	if (seconds > 0) {
360b103cf34SJulian Anastasov 		int period = timeout;
361b103cf34SJulian Anastasov 
362b103cf34SJulian Anastasov 		res = 1;
363b103cf34SJulian Anastasov 		while (seconds > period && res < 255) {
364b103cf34SJulian Anastasov 			res++;
365b103cf34SJulian Anastasov 			timeout <<= 1;
366b103cf34SJulian Anastasov 			if (timeout > rto_max)
367b103cf34SJulian Anastasov 				timeout = rto_max;
368b103cf34SJulian Anastasov 			period += timeout;
369b103cf34SJulian Anastasov 		}
370b103cf34SJulian Anastasov 	}
371b103cf34SJulian Anastasov 	return res;
372b103cf34SJulian Anastasov }
373b103cf34SJulian Anastasov 
374b103cf34SJulian Anastasov /* Convert retransmits to seconds based on initial and max timeout */
375b103cf34SJulian Anastasov static int retrans_to_secs(u8 retrans, int timeout, int rto_max)
376b103cf34SJulian Anastasov {
377b103cf34SJulian Anastasov 	int period = 0;
378b103cf34SJulian Anastasov 
379b103cf34SJulian Anastasov 	if (retrans > 0) {
380b103cf34SJulian Anastasov 		period = timeout;
381b103cf34SJulian Anastasov 		while (--retrans) {
382b103cf34SJulian Anastasov 			timeout <<= 1;
383b103cf34SJulian Anastasov 			if (timeout > rto_max)
384b103cf34SJulian Anastasov 				timeout = rto_max;
385b103cf34SJulian Anastasov 			period += timeout;
386b103cf34SJulian Anastasov 		}
387b103cf34SJulian Anastasov 	}
388b103cf34SJulian Anastasov 	return period;
389b103cf34SJulian Anastasov }
390b103cf34SJulian Anastasov 
3910263598cSWei Wang static u64 tcp_compute_delivery_rate(const struct tcp_sock *tp)
3920263598cSWei Wang {
3930263598cSWei Wang 	u32 rate = READ_ONCE(tp->rate_delivered);
3940263598cSWei Wang 	u32 intv = READ_ONCE(tp->rate_interval_us);
3950263598cSWei Wang 	u64 rate64 = 0;
3960263598cSWei Wang 
3970263598cSWei Wang 	if (rate && intv) {
3980263598cSWei Wang 		rate64 = (u64)rate * tp->mss_cache * USEC_PER_SEC;
3990263598cSWei Wang 		do_div(rate64, intv);
4000263598cSWei Wang 	}
4010263598cSWei Wang 	return rate64;
4020263598cSWei Wang }
4030263598cSWei Wang 
404900f65d3SNeal Cardwell /* Address-family independent initialization for a tcp_sock.
405900f65d3SNeal Cardwell  *
406900f65d3SNeal Cardwell  * NOTE: A lot of things set to zero explicitly by call to
407900f65d3SNeal Cardwell  *       sk_alloc() so need not be done here.
408900f65d3SNeal Cardwell  */
409900f65d3SNeal Cardwell void tcp_init_sock(struct sock *sk)
410900f65d3SNeal Cardwell {
411900f65d3SNeal Cardwell 	struct inet_connection_sock *icsk = inet_csk(sk);
412900f65d3SNeal Cardwell 	struct tcp_sock *tp = tcp_sk(sk);
413900f65d3SNeal Cardwell 
4149f5afeaeSYaogong Wang 	tp->out_of_order_queue = RB_ROOT;
41575c119afSEric Dumazet 	sk->tcp_rtx_queue = RB_ROOT;
416900f65d3SNeal Cardwell 	tcp_init_xmit_timers(sk);
41746d3ceabSEric Dumazet 	INIT_LIST_HEAD(&tp->tsq_node);
418e2080072SEric Dumazet 	INIT_LIST_HEAD(&tp->tsorted_sent_queue);
419900f65d3SNeal Cardwell 
420900f65d3SNeal Cardwell 	icsk->icsk_rto = TCP_TIMEOUT_INIT;
421ca584ba0SMartin KaFai Lau 	icsk->icsk_rto_min = TCP_RTO_MIN;
4222b8ee4f0SMartin KaFai Lau 	icsk->icsk_delack_max = TCP_DELACK_MAX;
423740b0f18SEric Dumazet 	tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT);
424ac9517fcSEric Dumazet 	minmax_reset(&tp->rtt_min, tcp_jiffies32, ~0U);
425900f65d3SNeal Cardwell 
426900f65d3SNeal Cardwell 	/* So many TCP implementations out there (incorrectly) count the
427900f65d3SNeal Cardwell 	 * initial SYN frame in their delayed-ACK and congestion control
428900f65d3SNeal Cardwell 	 * algorithms that we must have the following bandaid to talk
429900f65d3SNeal Cardwell 	 * efficiently to them.  -DaveM
430900f65d3SNeal Cardwell 	 */
431900f65d3SNeal Cardwell 	tp->snd_cwnd = TCP_INIT_CWND;
432900f65d3SNeal Cardwell 
433d7722e85SSoheil Hassas Yeganeh 	/* There's a bubble in the pipe until at least the first ACK. */
434d7722e85SSoheil Hassas Yeganeh 	tp->app_limited = ~0U;
435d7722e85SSoheil Hassas Yeganeh 
436900f65d3SNeal Cardwell 	/* See draft-stevens-tcpca-spec-01 for discussion of the
437900f65d3SNeal Cardwell 	 * initialization of these values.
438900f65d3SNeal Cardwell 	 */
439900f65d3SNeal Cardwell 	tp->snd_ssthresh = TCP_INFINITE_SSTHRESH;
440900f65d3SNeal Cardwell 	tp->snd_cwnd_clamp = ~0;
441900f65d3SNeal Cardwell 	tp->mss_cache = TCP_MSS_DEFAULT;
442900f65d3SNeal Cardwell 
4431043e25fSNikolay Borisov 	tp->reordering = sock_net(sk)->ipv4.sysctl_tcp_reordering;
44455d8694fSFlorian Westphal 	tcp_assign_congestion_control(sk);
445900f65d3SNeal Cardwell 
446ceaa1fefSAndrey Vagin 	tp->tsoffset = 0;
4471f255691SPriyaranjan Jha 	tp->rack.reo_wnd_steps = 1;
448ceaa1fefSAndrey Vagin 
449900f65d3SNeal Cardwell 	sk->sk_write_space = sk_stream_write_space;
450900f65d3SNeal Cardwell 	sock_set_flag(sk, SOCK_USE_WRITE_QUEUE);
451900f65d3SNeal Cardwell 
452900f65d3SNeal Cardwell 	icsk->icsk_sync_mss = tcp_sync_mss;
453900f65d3SNeal Cardwell 
454e292f05eSEric Dumazet 	WRITE_ONCE(sk->sk_sndbuf, sock_net(sk)->ipv4.sysctl_tcp_wmem[1]);
455ebb3b78dSEric Dumazet 	WRITE_ONCE(sk->sk_rcvbuf, sock_net(sk)->ipv4.sysctl_tcp_rmem[1]);
456900f65d3SNeal Cardwell 
457900f65d3SNeal Cardwell 	sk_sockets_allocated_inc(sk);
4580a6b2a1dSEric Dumazet 	sk->sk_route_forced_caps = NETIF_F_GSO;
459900f65d3SNeal Cardwell }
460900f65d3SNeal Cardwell EXPORT_SYMBOL(tcp_init_sock);
461900f65d3SNeal Cardwell 
4624e8cc228SEric Dumazet static void tcp_tx_timestamp(struct sock *sk, u16 tsflags)
4634ed2d765SWillem de Bruijn {
4644e8cc228SEric Dumazet 	struct sk_buff *skb = tcp_write_queue_tail(sk);
4654e8cc228SEric Dumazet 
466ad02c4f5SSoheil Hassas Yeganeh 	if (tsflags && skb) {
4674ed2d765SWillem de Bruijn 		struct skb_shared_info *shinfo = skb_shinfo(skb);
4686b084928SSoheil Hassas Yeganeh 		struct tcp_skb_cb *tcb = TCP_SKB_CB(skb);
4694ed2d765SWillem de Bruijn 
470c14ac945SSoheil Hassas Yeganeh 		sock_tx_timestamp(sk, tsflags, &shinfo->tx_flags);
4710a2cf20cSSoheil Hassas Yeganeh 		if (tsflags & SOF_TIMESTAMPING_TX_ACK)
4720a2cf20cSSoheil Hassas Yeganeh 			tcb->txstamp_ack = 1;
4730a2cf20cSSoheil Hassas Yeganeh 		if (tsflags & SOF_TIMESTAMPING_TX_RECORD_MASK)
4744ed2d765SWillem de Bruijn 			shinfo->tskey = TCP_SKB_CB(skb)->seq + skb->len - 1;
4754ed2d765SWillem de Bruijn 	}
476f066e2b0SWillem de Bruijn }
4774ed2d765SWillem de Bruijn 
4788934ce2fSJohn Fastabend static inline bool tcp_stream_is_readable(const struct tcp_sock *tp,
4798934ce2fSJohn Fastabend 					  int target, struct sock *sk)
4808934ce2fSJohn Fastabend {
48124adbc16SEric Dumazet 	int avail = READ_ONCE(tp->rcv_nxt) - READ_ONCE(tp->copied_seq);
48224adbc16SEric Dumazet 
48324adbc16SEric Dumazet 	if (avail > 0) {
48424adbc16SEric Dumazet 		if (avail >= target)
48524adbc16SEric Dumazet 			return true;
48624adbc16SEric Dumazet 		if (tcp_rmem_pressure(sk))
48724adbc16SEric Dumazet 			return true;
48824adbc16SEric Dumazet 	}
48924adbc16SEric Dumazet 	if (sk->sk_prot->stream_memory_read)
49024adbc16SEric Dumazet 		return sk->sk_prot->stream_memory_read(sk);
49124adbc16SEric Dumazet 	return false;
4928934ce2fSJohn Fastabend }
4938934ce2fSJohn Fastabend 
4941da177e4SLinus Torvalds /*
495a11e1d43SLinus Torvalds  *	Wait for a TCP event.
496a11e1d43SLinus Torvalds  *
497a11e1d43SLinus Torvalds  *	Note that we don't need to lock the socket, as the upper poll layers
498a11e1d43SLinus Torvalds  *	take care of normal races (between the test and the event) and we don't
499a11e1d43SLinus Torvalds  *	go look at any of the socket buffers directly.
5001da177e4SLinus Torvalds  */
501a11e1d43SLinus Torvalds __poll_t tcp_poll(struct file *file, struct socket *sock, poll_table *wait)
5021da177e4SLinus Torvalds {
503a11e1d43SLinus Torvalds 	__poll_t mask;
5041da177e4SLinus Torvalds 	struct sock *sk = sock->sk;
505cf533ea5SEric Dumazet 	const struct tcp_sock *tp = tcp_sk(sk);
50600fd38d9SEric Dumazet 	int state;
5071da177e4SLinus Torvalds 
50889ab066dSKarsten Graul 	sock_poll_wait(file, sock, wait);
509a11e1d43SLinus Torvalds 
510986ffdfdSYafang Shao 	state = inet_sk_state_load(sk);
51100fd38d9SEric Dumazet 	if (state == TCP_LISTEN)
512dc40c7bcSArnaldo Carvalho de Melo 		return inet_csk_listen_poll(sk);
5131da177e4SLinus Torvalds 
514a11e1d43SLinus Torvalds 	/* Socket is not locked. We are protected from async events
515a11e1d43SLinus Torvalds 	 * by poll logic and correct handling of state changes
516a11e1d43SLinus Torvalds 	 * made by other threads is impossible in any case.
517a11e1d43SLinus Torvalds 	 */
518a11e1d43SLinus Torvalds 
519a11e1d43SLinus Torvalds 	mask = 0;
520a11e1d43SLinus Torvalds 
5211da177e4SLinus Torvalds 	/*
522a9a08845SLinus Torvalds 	 * EPOLLHUP is certainly not done right. But poll() doesn't
5231da177e4SLinus Torvalds 	 * have a notion of HUP in just one direction, and for a
5241da177e4SLinus Torvalds 	 * socket the read side is more interesting.
5251da177e4SLinus Torvalds 	 *
526a9a08845SLinus Torvalds 	 * Some poll() documentation says that EPOLLHUP is incompatible
527a9a08845SLinus Torvalds 	 * with the EPOLLOUT/POLLWR flags, so somebody should check this
5281da177e4SLinus Torvalds 	 * all. But careful, it tends to be safer to return too many
5291da177e4SLinus Torvalds 	 * bits than too few, and you can easily break real applications
5301da177e4SLinus Torvalds 	 * if you don't tell them that something has hung up!
5311da177e4SLinus Torvalds 	 *
5321da177e4SLinus Torvalds 	 * Check-me.
5331da177e4SLinus Torvalds 	 *
534a9a08845SLinus Torvalds 	 * Check number 1. EPOLLHUP is _UNMASKABLE_ event (see UNIX98 and
5351da177e4SLinus Torvalds 	 * our fs/select.c). It means that after we received EOF,
5361da177e4SLinus Torvalds 	 * poll always returns immediately, making impossible poll() on write()
537a9a08845SLinus Torvalds 	 * in state CLOSE_WAIT. One solution is evident --- to set EPOLLHUP
5381da177e4SLinus Torvalds 	 * if and only if shutdown has been made in both directions.
5391da177e4SLinus Torvalds 	 * Actually, it is interesting to look how Solaris and DUX
540a9a08845SLinus Torvalds 	 * solve this dilemma. I would prefer, if EPOLLHUP were maskable,
5411da177e4SLinus Torvalds 	 * then we could set it on SND_SHUTDOWN. BTW examples given
5421da177e4SLinus Torvalds 	 * in Stevens' books assume exactly this behaviour, it explains
543a9a08845SLinus Torvalds 	 * why EPOLLHUP is incompatible with EPOLLOUT.	--ANK
5441da177e4SLinus Torvalds 	 *
5451da177e4SLinus Torvalds 	 * NOTE. Check for TCP_CLOSE is added. The goal is to prevent
5461da177e4SLinus Torvalds 	 * blocking on fresh not-connected or disconnected socket. --ANK
5471da177e4SLinus Torvalds 	 */
54800fd38d9SEric Dumazet 	if (sk->sk_shutdown == SHUTDOWN_MASK || state == TCP_CLOSE)
549a9a08845SLinus Torvalds 		mask |= EPOLLHUP;
5501da177e4SLinus Torvalds 	if (sk->sk_shutdown & RCV_SHUTDOWN)
551a9a08845SLinus Torvalds 		mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
5521da177e4SLinus Torvalds 
5538336886fSJerry Chu 	/* Connected or passive Fast Open socket? */
55400fd38d9SEric Dumazet 	if (state != TCP_SYN_SENT &&
555d983ea6fSEric Dumazet 	    (state != TCP_SYN_RECV || rcu_access_pointer(tp->fastopen_rsk))) {
556c7004482SDavid S. Miller 		int target = sock_rcvlowat(sk, 0, INT_MAX);
557c7004482SDavid S. Miller 
558d9b55bf7SEric Dumazet 		if (READ_ONCE(tp->urg_seq) == READ_ONCE(tp->copied_seq) &&
559c7004482SDavid S. Miller 		    !sock_flag(sk, SOCK_URGINLINE) &&
560c7004482SDavid S. Miller 		    tp->urg_data)
561b634f875SAlexandra Kossovsky 			target++;
562c7004482SDavid S. Miller 
5638934ce2fSJohn Fastabend 		if (tcp_stream_is_readable(tp, target, sk))
564a9a08845SLinus Torvalds 			mask |= EPOLLIN | EPOLLRDNORM;
5651da177e4SLinus Torvalds 
5661da177e4SLinus Torvalds 		if (!(sk->sk_shutdown & SEND_SHUTDOWN)) {
5678ba3c9d1SSoheil Hassas Yeganeh 			if (__sk_stream_is_writeable(sk, 1)) {
568a9a08845SLinus Torvalds 				mask |= EPOLLOUT | EPOLLWRNORM;
5691da177e4SLinus Torvalds 			} else {  /* send SIGIO later */
5709cd3e072SEric Dumazet 				sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
5711da177e4SLinus Torvalds 				set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
5721da177e4SLinus Torvalds 
5731da177e4SLinus Torvalds 				/* Race breaker. If space is freed after
5741da177e4SLinus Torvalds 				 * wspace test but before the flags are set,
5753c715127Sjbaron@akamai.com 				 * IO signal will be lost. Memory barrier
5763c715127Sjbaron@akamai.com 				 * pairs with the input side.
5771da177e4SLinus Torvalds 				 */
5783c715127Sjbaron@akamai.com 				smp_mb__after_atomic();
5798ba3c9d1SSoheil Hassas Yeganeh 				if (__sk_stream_is_writeable(sk, 1))
580a9a08845SLinus Torvalds 					mask |= EPOLLOUT | EPOLLWRNORM;
5811da177e4SLinus Torvalds 			}
582d84ba638SKOSAKI Motohiro 		} else
583a9a08845SLinus Torvalds 			mask |= EPOLLOUT | EPOLLWRNORM;
5841da177e4SLinus Torvalds 
5851da177e4SLinus Torvalds 		if (tp->urg_data & TCP_URG_VALID)
586a9a08845SLinus Torvalds 			mask |= EPOLLPRI;
587d68be71eSDavide Caratti 	} else if (state == TCP_SYN_SENT && inet_sk(sk)->defer_connect) {
58819f6d3f3SWei Wang 		/* Active TCP fastopen socket with defer_connect
589a9a08845SLinus Torvalds 		 * Return EPOLLOUT so application can call write()
59019f6d3f3SWei Wang 		 * in order for kernel to generate SYN+data
59119f6d3f3SWei Wang 		 */
592a9a08845SLinus Torvalds 		mask |= EPOLLOUT | EPOLLWRNORM;
5931da177e4SLinus Torvalds 	}
594a4d25803STom Marshall 	/* This barrier is coupled with smp_wmb() in tcp_reset() */
595a4d25803STom Marshall 	smp_rmb();
5963ef7cf57SEric Dumazet 	if (sk->sk_err || !skb_queue_empty_lockless(&sk->sk_error_queue))
597a9a08845SLinus Torvalds 		mask |= EPOLLERR;
598a4d25803STom Marshall 
5991da177e4SLinus Torvalds 	return mask;
6001da177e4SLinus Torvalds }
601a11e1d43SLinus Torvalds EXPORT_SYMBOL(tcp_poll);
6021da177e4SLinus Torvalds 
6031da177e4SLinus Torvalds int tcp_ioctl(struct sock *sk, int cmd, unsigned long arg)
6041da177e4SLinus Torvalds {
6051da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
6061da177e4SLinus Torvalds 	int answ;
6070e71c55cSEric Dumazet 	bool slow;
6081da177e4SLinus Torvalds 
6091da177e4SLinus Torvalds 	switch (cmd) {
6101da177e4SLinus Torvalds 	case SIOCINQ:
6111da177e4SLinus Torvalds 		if (sk->sk_state == TCP_LISTEN)
6121da177e4SLinus Torvalds 			return -EINVAL;
6131da177e4SLinus Torvalds 
6140e71c55cSEric Dumazet 		slow = lock_sock_fast(sk);
615473bd239STom Herbert 		answ = tcp_inq(sk);
6160e71c55cSEric Dumazet 		unlock_sock_fast(sk, slow);
6171da177e4SLinus Torvalds 		break;
6181da177e4SLinus Torvalds 	case SIOCATMARK:
619d9b55bf7SEric Dumazet 		answ = tp->urg_data &&
620d9b55bf7SEric Dumazet 		       READ_ONCE(tp->urg_seq) == READ_ONCE(tp->copied_seq);
6211da177e4SLinus Torvalds 		break;
6221da177e4SLinus Torvalds 	case SIOCOUTQ:
6231da177e4SLinus Torvalds 		if (sk->sk_state == TCP_LISTEN)
6241da177e4SLinus Torvalds 			return -EINVAL;
6251da177e4SLinus Torvalds 
6261da177e4SLinus Torvalds 		if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
6271da177e4SLinus Torvalds 			answ = 0;
6281da177e4SLinus Torvalds 		else
6290f317464SEric Dumazet 			answ = READ_ONCE(tp->write_seq) - tp->snd_una;
6301da177e4SLinus Torvalds 		break;
6312f4e1b39SMario Schuknecht 	case SIOCOUTQNSD:
6322f4e1b39SMario Schuknecht 		if (sk->sk_state == TCP_LISTEN)
6332f4e1b39SMario Schuknecht 			return -EINVAL;
6342f4e1b39SMario Schuknecht 
6352f4e1b39SMario Schuknecht 		if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
6362f4e1b39SMario Schuknecht 			answ = 0;
6372f4e1b39SMario Schuknecht 		else
638e0d694d6SEric Dumazet 			answ = READ_ONCE(tp->write_seq) -
639e0d694d6SEric Dumazet 			       READ_ONCE(tp->snd_nxt);
6402f4e1b39SMario Schuknecht 		break;
6411da177e4SLinus Torvalds 	default:
6421da177e4SLinus Torvalds 		return -ENOIOCTLCMD;
6433ff50b79SStephen Hemminger 	}
6441da177e4SLinus Torvalds 
6451da177e4SLinus Torvalds 	return put_user(answ, (int __user *)arg);
6461da177e4SLinus Torvalds }
6474bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_ioctl);
6481da177e4SLinus Torvalds 
6491da177e4SLinus Torvalds static inline 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 
660f4a775d1SEric Dumazet static void 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 
6651da177e4SLinus Torvalds 	skb->csum    = 0;
666352d4800SArnaldo Carvalho de Melo 	tcb->seq     = tcb->end_seq = tp->write_seq;
6674de075e0SEric Dumazet 	tcb->tcp_flags = TCPHDR_ACK;
668352d4800SArnaldo Carvalho de Melo 	tcb->sacked  = 0;
669f4a775d1SEric Dumazet 	__skb_header_release(skb);
670fe067e8aSDavid S. Miller 	tcp_add_write_queue_tail(sk, skb);
671ab4e846aSEric Dumazet 	sk_wmem_queued_add(sk, skb->truesize);
6723ab224beSHideo Aoki 	sk_mem_charge(sk, skb->truesize);
67389ebd197SDavid S. Miller 	if (tp->nonagle & TCP_NAGLE_PUSH)
6741da177e4SLinus Torvalds 		tp->nonagle &= ~TCP_NAGLE_PUSH;
6756f021c62SEric Dumazet 
6766f021c62SEric Dumazet 	tcp_slow_start_after_idle_check(sk);
6771da177e4SLinus Torvalds }
6781da177e4SLinus Torvalds 
679afeca340SKrishna Kumar static inline void tcp_mark_urg(struct tcp_sock *tp, int flags)
6801da177e4SLinus Torvalds {
68133f5f57eSIlpo Järvinen 	if (flags & MSG_OOB)
6821da177e4SLinus Torvalds 		tp->snd_up = tp->write_seq;
6831da177e4SLinus Torvalds }
6841da177e4SLinus Torvalds 
685f54b3111SEric Dumazet /* If a not yet filled skb is pushed, do not send it if
686a181ceb5SEric Dumazet  * we have data packets in Qdisc or NIC queues :
687f54b3111SEric Dumazet  * Because TX completion will happen shortly, it gives a chance
688f54b3111SEric Dumazet  * to coalesce future sendmsg() payload into this skb, without
689f54b3111SEric Dumazet  * need for a timer, and with no latency trade off.
690f54b3111SEric Dumazet  * As packets containing data payload have a bigger truesize
691a181ceb5SEric Dumazet  * than pure acks (dataless) packets, the last checks prevent
692a181ceb5SEric Dumazet  * autocorking if we only have an ACK in Qdisc/NIC queues,
693a181ceb5SEric Dumazet  * or if TX completion was delayed after we processed ACK packet.
694f54b3111SEric Dumazet  */
695f54b3111SEric Dumazet static bool tcp_should_autocork(struct sock *sk, struct sk_buff *skb,
696f54b3111SEric Dumazet 				int size_goal)
6971da177e4SLinus Torvalds {
698f54b3111SEric Dumazet 	return skb->len < size_goal &&
699790f00e1SEric Dumazet 	       sock_net(sk)->ipv4.sysctl_tcp_autocorking &&
700114f39feSEric Dumazet 	       !tcp_rtx_queue_empty(sk) &&
70114afee4bSReshetova, Elena 	       refcount_read(&sk->sk_wmem_alloc) > skb->truesize;
702f54b3111SEric Dumazet }
7039e412ba7SIlpo Järvinen 
70435b2c321SMat Martineau void tcp_push(struct sock *sk, int flags, int mss_now,
705f54b3111SEric Dumazet 	      int nonagle, int size_goal)
706f54b3111SEric Dumazet {
707f54b3111SEric Dumazet 	struct tcp_sock *tp = tcp_sk(sk);
708f54b3111SEric Dumazet 	struct sk_buff *skb;
709f54b3111SEric Dumazet 
710f54b3111SEric Dumazet 	skb = tcp_write_queue_tail(sk);
71175c119afSEric Dumazet 	if (!skb)
71275c119afSEric Dumazet 		return;
7131da177e4SLinus Torvalds 	if (!(flags & MSG_MORE) || forced_push(tp))
714f54b3111SEric Dumazet 		tcp_mark_push(tp, skb);
715afeca340SKrishna Kumar 
716afeca340SKrishna Kumar 	tcp_mark_urg(tp, flags);
717f54b3111SEric Dumazet 
718f54b3111SEric Dumazet 	if (tcp_should_autocork(sk, skb, size_goal)) {
719f54b3111SEric Dumazet 
720f54b3111SEric Dumazet 		/* avoid atomic op if TSQ_THROTTLED bit is already set */
7217aa5470cSEric Dumazet 		if (!test_bit(TSQ_THROTTLED, &sk->sk_tsq_flags)) {
722f54b3111SEric Dumazet 			NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPAUTOCORKING);
7237aa5470cSEric Dumazet 			set_bit(TSQ_THROTTLED, &sk->sk_tsq_flags);
7241da177e4SLinus Torvalds 		}
725a181ceb5SEric Dumazet 		/* It is possible TX completion already happened
726a181ceb5SEric Dumazet 		 * before we set TSQ_THROTTLED.
727a181ceb5SEric Dumazet 		 */
72814afee4bSReshetova, Elena 		if (refcount_read(&sk->sk_wmem_alloc) > skb->truesize)
729f54b3111SEric Dumazet 			return;
730f54b3111SEric Dumazet 	}
731f54b3111SEric Dumazet 
732f54b3111SEric Dumazet 	if (flags & MSG_MORE)
733f54b3111SEric Dumazet 		nonagle = TCP_NAGLE_CORK;
734f54b3111SEric Dumazet 
735f54b3111SEric Dumazet 	__tcp_push_pending_frames(sk, mss_now, nonagle);
7361da177e4SLinus Torvalds }
7371da177e4SLinus Torvalds 
7386ff7751dSAdrian Bunk static int tcp_splice_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb,
7399c55e01cSJens Axboe 				unsigned int offset, size_t len)
7409c55e01cSJens Axboe {
7419c55e01cSJens Axboe 	struct tcp_splice_state *tss = rd_desc->arg.data;
74233966dd0SWilly Tarreau 	int ret;
7439c55e01cSJens Axboe 
744a60e3cc7SHannes Frederic Sowa 	ret = skb_splice_bits(skb, skb->sk, offset, tss->pipe,
74525869262SAl Viro 			      min(rd_desc->count, len), tss->flags);
74633966dd0SWilly Tarreau 	if (ret > 0)
74733966dd0SWilly Tarreau 		rd_desc->count -= ret;
74833966dd0SWilly Tarreau 	return ret;
7499c55e01cSJens Axboe }
7509c55e01cSJens Axboe 
7519c55e01cSJens Axboe static int __tcp_splice_read(struct sock *sk, struct tcp_splice_state *tss)
7529c55e01cSJens Axboe {
7539c55e01cSJens Axboe 	/* Store TCP splice context information in read_descriptor_t. */
7549c55e01cSJens Axboe 	read_descriptor_t rd_desc = {
7559c55e01cSJens Axboe 		.arg.data = tss,
75633966dd0SWilly Tarreau 		.count	  = tss->len,
7579c55e01cSJens Axboe 	};
7589c55e01cSJens Axboe 
7599c55e01cSJens Axboe 	return tcp_read_sock(sk, &rd_desc, tcp_splice_data_recv);
7609c55e01cSJens Axboe }
7619c55e01cSJens Axboe 
7629c55e01cSJens Axboe /**
7639c55e01cSJens Axboe  *  tcp_splice_read - splice data from TCP socket to a pipe
7649c55e01cSJens Axboe  * @sock:	socket to splice from
7659c55e01cSJens Axboe  * @ppos:	position (not valid)
7669c55e01cSJens Axboe  * @pipe:	pipe to splice to
7679c55e01cSJens Axboe  * @len:	number of bytes to splice
7689c55e01cSJens Axboe  * @flags:	splice modifier flags
7699c55e01cSJens Axboe  *
7709c55e01cSJens Axboe  * Description:
7719c55e01cSJens Axboe  *    Will read pages from given socket and fill them into a pipe.
7729c55e01cSJens Axboe  *
7739c55e01cSJens Axboe  **/
7749c55e01cSJens Axboe ssize_t tcp_splice_read(struct socket *sock, loff_t *ppos,
7759c55e01cSJens Axboe 			struct pipe_inode_info *pipe, size_t len,
7769c55e01cSJens Axboe 			unsigned int flags)
7779c55e01cSJens Axboe {
7789c55e01cSJens Axboe 	struct sock *sk = sock->sk;
7799c55e01cSJens Axboe 	struct tcp_splice_state tss = {
7809c55e01cSJens Axboe 		.pipe = pipe,
7819c55e01cSJens Axboe 		.len = len,
7829c55e01cSJens Axboe 		.flags = flags,
7839c55e01cSJens Axboe 	};
7849c55e01cSJens Axboe 	long timeo;
7859c55e01cSJens Axboe 	ssize_t spliced;
7869c55e01cSJens Axboe 	int ret;
7879c55e01cSJens Axboe 
7883a047bf8SChangli Gao 	sock_rps_record_flow(sk);
7899c55e01cSJens Axboe 	/*
7909c55e01cSJens Axboe 	 * We can't seek on a socket input
7919c55e01cSJens Axboe 	 */
7929c55e01cSJens Axboe 	if (unlikely(*ppos))
7939c55e01cSJens Axboe 		return -ESPIPE;
7949c55e01cSJens Axboe 
7959c55e01cSJens Axboe 	ret = spliced = 0;
7969c55e01cSJens Axboe 
7979c55e01cSJens Axboe 	lock_sock(sk);
7989c55e01cSJens Axboe 
79942324c62SEric Dumazet 	timeo = sock_rcvtimeo(sk, sock->file->f_flags & O_NONBLOCK);
8009c55e01cSJens Axboe 	while (tss.len) {
8019c55e01cSJens Axboe 		ret = __tcp_splice_read(sk, &tss);
8029c55e01cSJens Axboe 		if (ret < 0)
8039c55e01cSJens Axboe 			break;
8049c55e01cSJens Axboe 		else if (!ret) {
8059c55e01cSJens Axboe 			if (spliced)
8069c55e01cSJens Axboe 				break;
8079c55e01cSJens Axboe 			if (sock_flag(sk, SOCK_DONE))
8089c55e01cSJens Axboe 				break;
8099c55e01cSJens Axboe 			if (sk->sk_err) {
8109c55e01cSJens Axboe 				ret = sock_error(sk);
8119c55e01cSJens Axboe 				break;
8129c55e01cSJens Axboe 			}
8139c55e01cSJens Axboe 			if (sk->sk_shutdown & RCV_SHUTDOWN)
8149c55e01cSJens Axboe 				break;
8159c55e01cSJens Axboe 			if (sk->sk_state == TCP_CLOSE) {
8169c55e01cSJens Axboe 				/*
8179c55e01cSJens Axboe 				 * This occurs when user tries to read
8189c55e01cSJens Axboe 				 * from never connected socket.
8199c55e01cSJens Axboe 				 */
8209c55e01cSJens Axboe 				ret = -ENOTCONN;
8219c55e01cSJens Axboe 				break;
8229c55e01cSJens Axboe 			}
8239c55e01cSJens Axboe 			if (!timeo) {
8249c55e01cSJens Axboe 				ret = -EAGAIN;
8259c55e01cSJens Axboe 				break;
8269c55e01cSJens Axboe 			}
827ccf7abb9SEric Dumazet 			/* if __tcp_splice_read() got nothing while we have
828ccf7abb9SEric Dumazet 			 * an skb in receive queue, we do not want to loop.
829ccf7abb9SEric Dumazet 			 * This might happen with URG data.
830ccf7abb9SEric Dumazet 			 */
831ccf7abb9SEric Dumazet 			if (!skb_queue_empty(&sk->sk_receive_queue))
832ccf7abb9SEric Dumazet 				break;
833dfbafc99SSabrina Dubroca 			sk_wait_data(sk, &timeo, NULL);
8349c55e01cSJens Axboe 			if (signal_pending(current)) {
8359c55e01cSJens Axboe 				ret = sock_intr_errno(timeo);
8369c55e01cSJens Axboe 				break;
8379c55e01cSJens Axboe 			}
8389c55e01cSJens Axboe 			continue;
8399c55e01cSJens Axboe 		}
8409c55e01cSJens Axboe 		tss.len -= ret;
8419c55e01cSJens Axboe 		spliced += ret;
8429c55e01cSJens Axboe 
84333966dd0SWilly Tarreau 		if (!timeo)
84433966dd0SWilly Tarreau 			break;
8459c55e01cSJens Axboe 		release_sock(sk);
8469c55e01cSJens Axboe 		lock_sock(sk);
8479c55e01cSJens Axboe 
8489c55e01cSJens Axboe 		if (sk->sk_err || sk->sk_state == TCP_CLOSE ||
84933966dd0SWilly Tarreau 		    (sk->sk_shutdown & RCV_SHUTDOWN) ||
8509c55e01cSJens Axboe 		    signal_pending(current))
8519c55e01cSJens Axboe 			break;
8529c55e01cSJens Axboe 	}
8539c55e01cSJens Axboe 
8549c55e01cSJens Axboe 	release_sock(sk);
8559c55e01cSJens Axboe 
8569c55e01cSJens Axboe 	if (spliced)
8579c55e01cSJens Axboe 		return spliced;
8589c55e01cSJens Axboe 
8599c55e01cSJens Axboe 	return ret;
8609c55e01cSJens Axboe }
8614bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_splice_read);
8629c55e01cSJens Axboe 
863eb934478SEric Dumazet struct sk_buff *sk_stream_alloc_skb(struct sock *sk, int size, gfp_t gfp,
864eb934478SEric Dumazet 				    bool force_schedule)
865f561d0f2SPavel Emelyanov {
866f561d0f2SPavel Emelyanov 	struct sk_buff *skb;
867f561d0f2SPavel Emelyanov 
8684f661542SEric Dumazet 	if (likely(!size)) {
869472c2e07SEric Dumazet 		skb = sk->sk_tx_skb_cache;
870858f5017SEric Dumazet 		if (skb) {
871d7cc399eSEric Dumazet 			skb->truesize = SKB_TRUESIZE(skb_end_offset(skb));
872472c2e07SEric Dumazet 			sk->sk_tx_skb_cache = NULL;
873472c2e07SEric Dumazet 			pskb_trim(skb, 0);
874472c2e07SEric Dumazet 			INIT_LIST_HEAD(&skb->tcp_tsorted_anchor);
875eb70a1aeSEric Dumazet 			skb_shinfo(skb)->tx_flags = 0;
876eb70a1aeSEric Dumazet 			memset(TCP_SKB_CB(skb), 0, sizeof(struct tcp_skb_cb));
877472c2e07SEric Dumazet 			return skb;
878472c2e07SEric Dumazet 		}
879472c2e07SEric Dumazet 	}
880f561d0f2SPavel Emelyanov 	/* The TCP header must be at least 32-bit aligned.  */
881f561d0f2SPavel Emelyanov 	size = ALIGN(size, 4);
882f561d0f2SPavel Emelyanov 
8838e4d980aSEric Dumazet 	if (unlikely(tcp_under_memory_pressure(sk)))
8848e4d980aSEric Dumazet 		sk_mem_reclaim_partial(sk);
8858e4d980aSEric Dumazet 
886f561d0f2SPavel Emelyanov 	skb = alloc_skb_fclone(size + sk->sk_prot->max_header, gfp);
8878e4d980aSEric Dumazet 	if (likely(skb)) {
888eb934478SEric Dumazet 		bool mem_scheduled;
8898e4d980aSEric Dumazet 
890eb934478SEric Dumazet 		if (force_schedule) {
891eb934478SEric Dumazet 			mem_scheduled = true;
8928e4d980aSEric Dumazet 			sk_forced_mem_schedule(sk, skb->truesize);
8938e4d980aSEric Dumazet 		} else {
894eb934478SEric Dumazet 			mem_scheduled = sk_wmem_schedule(sk, skb->truesize);
8958e4d980aSEric Dumazet 		}
896eb934478SEric Dumazet 		if (likely(mem_scheduled)) {
897a21d4572SEric Dumazet 			skb_reserve(skb, sk->sk_prot->max_header);
898f561d0f2SPavel Emelyanov 			/*
899f561d0f2SPavel Emelyanov 			 * Make sure that we have exactly size bytes
900f561d0f2SPavel Emelyanov 			 * available to the caller, no more, no less.
901f561d0f2SPavel Emelyanov 			 */
90216fad69cSEric Dumazet 			skb->reserved_tailroom = skb->end - skb->tail - size;
903e2080072SEric Dumazet 			INIT_LIST_HEAD(&skb->tcp_tsorted_anchor);
904f561d0f2SPavel Emelyanov 			return skb;
905f561d0f2SPavel Emelyanov 		}
906f561d0f2SPavel Emelyanov 		__kfree_skb(skb);
907f561d0f2SPavel Emelyanov 	} else {
9085c52ba17SPavel Emelyanov 		sk->sk_prot->enter_memory_pressure(sk);
909f561d0f2SPavel Emelyanov 		sk_stream_moderate_sndbuf(sk);
910f561d0f2SPavel Emelyanov 	}
911f561d0f2SPavel Emelyanov 	return NULL;
912f561d0f2SPavel Emelyanov }
913f561d0f2SPavel Emelyanov 
9140c54b85fSIlpo Järvinen static unsigned int tcp_xmit_size_goal(struct sock *sk, u32 mss_now,
9150c54b85fSIlpo Järvinen 				       int large_allowed)
9160c54b85fSIlpo Järvinen {
9170c54b85fSIlpo Järvinen 	struct tcp_sock *tp = tcp_sk(sk);
9186c09fa09SEric Dumazet 	u32 new_size_goal, size_goal;
9190c54b85fSIlpo Järvinen 
92074d4a8f8SEric Dumazet 	if (!large_allowed)
921605ad7f1SEric Dumazet 		return mss_now;
9220c54b85fSIlpo Järvinen 
9236c09fa09SEric Dumazet 	/* Note : tcp_tso_autosize() will eventually split this later */
9246c09fa09SEric Dumazet 	new_size_goal = sk->sk_gso_max_size - 1 - MAX_TCP_HEADER;
925605ad7f1SEric Dumazet 	new_size_goal = tcp_bound_to_half_wnd(tp, new_size_goal);
9262a3a041cSIlpo Järvinen 
9272a3a041cSIlpo Järvinen 	/* We try hard to avoid divides here */
928605ad7f1SEric Dumazet 	size_goal = tp->gso_segs * mss_now;
929605ad7f1SEric Dumazet 	if (unlikely(new_size_goal < size_goal ||
930605ad7f1SEric Dumazet 		     new_size_goal >= size_goal + mss_now)) {
931605ad7f1SEric Dumazet 		tp->gso_segs = min_t(u16, new_size_goal / mss_now,
9321485348dSBen Hutchings 				     sk->sk_gso_max_segs);
933605ad7f1SEric Dumazet 		size_goal = tp->gso_segs * mss_now;
9340c54b85fSIlpo Järvinen 	}
9350c54b85fSIlpo Järvinen 
936605ad7f1SEric Dumazet 	return max(size_goal, mss_now);
9370c54b85fSIlpo Järvinen }
9380c54b85fSIlpo Järvinen 
93935b2c321SMat Martineau int tcp_send_mss(struct sock *sk, int *size_goal, int flags)
9400c54b85fSIlpo Järvinen {
9410c54b85fSIlpo Järvinen 	int mss_now;
9420c54b85fSIlpo Järvinen 
9430c54b85fSIlpo Järvinen 	mss_now = tcp_current_mss(sk);
9440c54b85fSIlpo Järvinen 	*size_goal = tcp_xmit_size_goal(sk, mss_now, !(flags & MSG_OOB));
9450c54b85fSIlpo Järvinen 
9460c54b85fSIlpo Järvinen 	return mss_now;
9470c54b85fSIlpo Järvinen }
9480c54b85fSIlpo Järvinen 
949fdfc5c85SEric Dumazet /* In some cases, both sendpage() and sendmsg() could have added
950fdfc5c85SEric Dumazet  * an skb to the write queue, but failed adding payload on it.
951fdfc5c85SEric Dumazet  * We need to remove it to consume less memory, but more
952fdfc5c85SEric Dumazet  * importantly be able to generate EPOLLOUT for Edge Trigger epoll()
953fdfc5c85SEric Dumazet  * users.
954fdfc5c85SEric Dumazet  */
955fdfc5c85SEric Dumazet static void tcp_remove_empty_skb(struct sock *sk, struct sk_buff *skb)
956fdfc5c85SEric Dumazet {
957fdfc5c85SEric Dumazet 	if (skb && !skb->len) {
958fdfc5c85SEric Dumazet 		tcp_unlink_write_queue(skb, sk);
959fdfc5c85SEric Dumazet 		if (tcp_write_queue_empty(sk))
960fdfc5c85SEric Dumazet 			tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
961fdfc5c85SEric Dumazet 		sk_wmem_free_skb(sk, skb);
962fdfc5c85SEric Dumazet 	}
963fdfc5c85SEric Dumazet }
964fdfc5c85SEric Dumazet 
965e3b5616aSDave Watson ssize_t do_tcp_sendpages(struct sock *sk, struct page *page, int offset,
96664022d0bSEric Dumazet 			 size_t size, int flags)
9671da177e4SLinus Torvalds {
9681da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
969c1b4a7e6SDavid S. Miller 	int mss_now, size_goal;
9701da177e4SLinus Torvalds 	int err;
9711da177e4SLinus Torvalds 	ssize_t copied;
9721da177e4SLinus Torvalds 	long timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
9731da177e4SLinus Torvalds 
974a10674bfSVasily Averin 	if (IS_ENABLED(CONFIG_DEBUG_VM) &&
975a10674bfSVasily Averin 	    WARN_ONCE(PageSlab(page), "page must not be a Slab one"))
976a10674bfSVasily Averin 		return -EINVAL;
977a10674bfSVasily Averin 
9788336886fSJerry Chu 	/* Wait for a connection to finish. One exception is TCP Fast Open
9798336886fSJerry Chu 	 * (passive side) where data is allowed to be sent before a connection
9808336886fSJerry Chu 	 * is fully established.
9818336886fSJerry Chu 	 */
9828336886fSJerry Chu 	if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) &&
9838336886fSJerry Chu 	    !tcp_passive_fastopen(sk)) {
984686a5624SYuvaraja Mariappan 		err = sk_stream_wait_connect(sk, &timeo);
985686a5624SYuvaraja Mariappan 		if (err != 0)
9861da177e4SLinus Torvalds 			goto out_err;
9878336886fSJerry Chu 	}
9881da177e4SLinus Torvalds 
9899cd3e072SEric Dumazet 	sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
9901da177e4SLinus Torvalds 
9910c54b85fSIlpo Järvinen 	mss_now = tcp_send_mss(sk, &size_goal, flags);
9921da177e4SLinus Torvalds 	copied = 0;
9931da177e4SLinus Torvalds 
9941da177e4SLinus Torvalds 	err = -EPIPE;
9951da177e4SLinus Torvalds 	if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
9960d6a775eSIlpo Järvinen 		goto out_err;
9971da177e4SLinus Torvalds 
99864022d0bSEric Dumazet 	while (size > 0) {
999fe067e8aSDavid S. Miller 		struct sk_buff *skb = tcp_write_queue_tail(sk);
100038ba0a65SEric Dumazet 		int copy, i;
100138ba0a65SEric Dumazet 		bool can_coalesce;
10021da177e4SLinus Torvalds 
100375c119afSEric Dumazet 		if (!skb || (copy = size_goal - skb->len) <= 0 ||
1004c134ecb8SMartin KaFai Lau 		    !tcp_skb_can_collapse_to(skb)) {
10051da177e4SLinus Torvalds new_segment:
10061da177e4SLinus Torvalds 			if (!sk_stream_memory_free(sk))
1007*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
10081da177e4SLinus Torvalds 
1009eb934478SEric Dumazet 			skb = sk_stream_alloc_skb(sk, 0, sk->sk_allocation,
101075c119afSEric Dumazet 					tcp_rtx_and_write_queues_empty(sk));
10111da177e4SLinus Torvalds 			if (!skb)
1012*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
10131da177e4SLinus Torvalds 
101441477662SJakub Kicinski #ifdef CONFIG_TLS_DEVICE
101541477662SJakub Kicinski 			skb->decrypted = !!(flags & MSG_SENDPAGE_DECRYPTED);
101641477662SJakub Kicinski #endif
10179e412ba7SIlpo Järvinen 			skb_entail(sk, skb);
1018c1b4a7e6SDavid S. Miller 			copy = size_goal;
10191da177e4SLinus Torvalds 		}
10201da177e4SLinus Torvalds 
10211da177e4SLinus Torvalds 		if (copy > size)
10221da177e4SLinus Torvalds 			copy = size;
10231da177e4SLinus Torvalds 
10241da177e4SLinus Torvalds 		i = skb_shinfo(skb)->nr_frags;
10251da177e4SLinus Torvalds 		can_coalesce = skb_can_coalesce(skb, i, page, offset);
10265f74f82eSHans Westgaard Ry 		if (!can_coalesce && i >= sysctl_max_skb_frags) {
10271da177e4SLinus Torvalds 			tcp_mark_push(tp, skb);
10281da177e4SLinus Torvalds 			goto new_segment;
10291da177e4SLinus Torvalds 		}
10303ab224beSHideo Aoki 		if (!sk_wmem_schedule(sk, copy))
1031*afb83012SSoheil Hassas Yeganeh 			goto wait_for_space;
10321da177e4SLinus Torvalds 
10331da177e4SLinus Torvalds 		if (can_coalesce) {
10349e903e08SEric Dumazet 			skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy);
10351da177e4SLinus Torvalds 		} else {
10361da177e4SLinus Torvalds 			get_page(page);
10371da177e4SLinus Torvalds 			skb_fill_page_desc(skb, i, page, offset, copy);
10381da177e4SLinus Torvalds 		}
1039312fc2b4SJohn Fastabend 
1040312fc2b4SJohn Fastabend 		if (!(flags & MSG_NO_SHARED_FRAGS))
1041c9af6db4SPravin B Shelar 			skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1042cef401deSEric Dumazet 
10431da177e4SLinus Torvalds 		skb->len += copy;
10441da177e4SLinus Torvalds 		skb->data_len += copy;
10451da177e4SLinus Torvalds 		skb->truesize += copy;
1046ab4e846aSEric Dumazet 		sk_wmem_queued_add(sk, copy);
10473ab224beSHideo Aoki 		sk_mem_charge(sk, copy);
104884fa7933SPatrick McHardy 		skb->ip_summed = CHECKSUM_PARTIAL;
10490f317464SEric Dumazet 		WRITE_ONCE(tp->write_seq, tp->write_seq + copy);
10501da177e4SLinus Torvalds 		TCP_SKB_CB(skb)->end_seq += copy;
1051cd7d8498SEric Dumazet 		tcp_skb_pcount_set(skb, 0);
10521da177e4SLinus Torvalds 
10531da177e4SLinus Torvalds 		if (!copied)
10544de075e0SEric Dumazet 			TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH;
10551da177e4SLinus Torvalds 
10561da177e4SLinus Torvalds 		copied += copy;
105764022d0bSEric Dumazet 		offset += copy;
1058686a5624SYuvaraja Mariappan 		size -= copy;
1059ad02c4f5SSoheil Hassas Yeganeh 		if (!size)
10601da177e4SLinus Torvalds 			goto out;
10611da177e4SLinus Torvalds 
106269d15067SHerbert Xu 		if (skb->len < size_goal || (flags & MSG_OOB))
10631da177e4SLinus Torvalds 			continue;
10641da177e4SLinus Torvalds 
10651da177e4SLinus Torvalds 		if (forced_push(tp)) {
10661da177e4SLinus Torvalds 			tcp_mark_push(tp, skb);
10679e412ba7SIlpo Järvinen 			__tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);
1068fe067e8aSDavid S. Miller 		} else if (skb == tcp_send_head(sk))
10691da177e4SLinus Torvalds 			tcp_push_one(sk, mss_now);
10701da177e4SLinus Torvalds 		continue;
10711da177e4SLinus Torvalds 
1072*afb83012SSoheil Hassas Yeganeh wait_for_space:
10731da177e4SLinus Torvalds 		set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1074f54b3111SEric Dumazet 		tcp_push(sk, flags & ~MSG_MORE, mss_now,
1075f54b3111SEric Dumazet 			 TCP_NAGLE_PUSH, size_goal);
10761da177e4SLinus Torvalds 
1077686a5624SYuvaraja Mariappan 		err = sk_stream_wait_memory(sk, &timeo);
1078686a5624SYuvaraja Mariappan 		if (err != 0)
10791da177e4SLinus Torvalds 			goto do_error;
10801da177e4SLinus Torvalds 
10810c54b85fSIlpo Järvinen 		mss_now = tcp_send_mss(sk, &size_goal, flags);
10821da177e4SLinus Torvalds 	}
10831da177e4SLinus Torvalds 
10841da177e4SLinus Torvalds out:
1085ad02c4f5SSoheil Hassas Yeganeh 	if (copied) {
10864e8cc228SEric Dumazet 		tcp_tx_timestamp(sk, sk->sk_tsflags);
1087ad02c4f5SSoheil Hassas Yeganeh 		if (!(flags & MSG_SENDPAGE_NOTLAST))
1088f54b3111SEric Dumazet 			tcp_push(sk, flags, mss_now, tp->nonagle, size_goal);
1089ad02c4f5SSoheil Hassas Yeganeh 	}
10901da177e4SLinus Torvalds 	return copied;
10911da177e4SLinus Torvalds 
10921da177e4SLinus Torvalds do_error:
1093fdfc5c85SEric Dumazet 	tcp_remove_empty_skb(sk, tcp_write_queue_tail(sk));
10941da177e4SLinus Torvalds 	if (copied)
10951da177e4SLinus Torvalds 		goto out;
10961da177e4SLinus Torvalds out_err:
1097ce5ec440SJason Baron 	/* make sure we wake any epoll edge trigger waiter */
1098216808c6SEric Dumazet 	if (unlikely(tcp_rtx_and_write_queues_empty(sk) && err == -EAGAIN)) {
1099ce5ec440SJason Baron 		sk->sk_write_space(sk);
1100b0f71bd3SFrancis Yan 		tcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED);
1101b0f71bd3SFrancis Yan 	}
11021da177e4SLinus Torvalds 	return sk_stream_error(sk, flags, err);
11031da177e4SLinus Torvalds }
1104e3b5616aSDave Watson EXPORT_SYMBOL_GPL(do_tcp_sendpages);
11051da177e4SLinus Torvalds 
1106306b13ebSTom Herbert int tcp_sendpage_locked(struct sock *sk, struct page *page, int offset,
11071da177e4SLinus Torvalds 			size_t size, int flags)
11081da177e4SLinus Torvalds {
1109dead7cdbSEric Dumazet 	if (!(sk->sk_route_caps & NETIF_F_SG))
1110bd9dfc54SEric Dumazet 		return sock_no_sendpage_locked(sk, page, offset, size, flags);
11111da177e4SLinus Torvalds 
1112d7722e85SSoheil Hassas Yeganeh 	tcp_rate_check_app_limited(sk);  /* is sending application-limited? */
1113d7722e85SSoheil Hassas Yeganeh 
1114306b13ebSTom Herbert 	return do_tcp_sendpages(sk, page, offset, size, flags);
1115306b13ebSTom Herbert }
1116774c4673SDavid S. Miller EXPORT_SYMBOL_GPL(tcp_sendpage_locked);
1117306b13ebSTom Herbert 
1118306b13ebSTom Herbert int tcp_sendpage(struct sock *sk, struct page *page, int offset,
1119306b13ebSTom Herbert 		 size_t size, int flags)
1120306b13ebSTom Herbert {
1121306b13ebSTom Herbert 	int ret;
1122306b13ebSTom Herbert 
1123306b13ebSTom Herbert 	lock_sock(sk);
1124306b13ebSTom Herbert 	ret = tcp_sendpage_locked(sk, page, offset, size, flags);
11251da177e4SLinus Torvalds 	release_sock(sk);
1126306b13ebSTom Herbert 
1127306b13ebSTom Herbert 	return ret;
11281da177e4SLinus Torvalds }
11294bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_sendpage);
11301da177e4SLinus Torvalds 
1131cf60af03SYuchung Cheng void tcp_free_fastopen_req(struct tcp_sock *tp)
1132cf60af03SYuchung Cheng {
113300db4124SIan Morris 	if (tp->fastopen_req) {
1134cf60af03SYuchung Cheng 		kfree(tp->fastopen_req);
1135cf60af03SYuchung Cheng 		tp->fastopen_req = NULL;
1136cf60af03SYuchung Cheng 	}
1137cf60af03SYuchung Cheng }
1138cf60af03SYuchung Cheng 
1139f5ddcbbbSEric Dumazet static int tcp_sendmsg_fastopen(struct sock *sk, struct msghdr *msg,
1140f859a448SWillem de Bruijn 				int *copied, size_t size,
1141f859a448SWillem de Bruijn 				struct ubuf_info *uarg)
1142cf60af03SYuchung Cheng {
1143cf60af03SYuchung Cheng 	struct tcp_sock *tp = tcp_sk(sk);
114419f6d3f3SWei Wang 	struct inet_sock *inet = inet_sk(sk);
1145ba615f67SWei Wang 	struct sockaddr *uaddr = msg->msg_name;
1146cf60af03SYuchung Cheng 	int err, flags;
1147cf60af03SYuchung Cheng 
1148e1cfcbe8SHaishuang Yan 	if (!(sock_net(sk)->ipv4.sysctl_tcp_fastopen & TFO_CLIENT_ENABLE) ||
1149ba615f67SWei Wang 	    (uaddr && msg->msg_namelen >= sizeof(uaddr->sa_family) &&
1150ba615f67SWei Wang 	     uaddr->sa_family == AF_UNSPEC))
1151cf60af03SYuchung Cheng 		return -EOPNOTSUPP;
115200db4124SIan Morris 	if (tp->fastopen_req)
1153cf60af03SYuchung Cheng 		return -EALREADY; /* Another Fast Open is in progress */
1154cf60af03SYuchung Cheng 
1155cf60af03SYuchung Cheng 	tp->fastopen_req = kzalloc(sizeof(struct tcp_fastopen_request),
1156cf60af03SYuchung Cheng 				   sk->sk_allocation);
115751456b29SIan Morris 	if (unlikely(!tp->fastopen_req))
1158cf60af03SYuchung Cheng 		return -ENOBUFS;
1159cf60af03SYuchung Cheng 	tp->fastopen_req->data = msg;
1160f5ddcbbbSEric Dumazet 	tp->fastopen_req->size = size;
1161f859a448SWillem de Bruijn 	tp->fastopen_req->uarg = uarg;
1162cf60af03SYuchung Cheng 
116319f6d3f3SWei Wang 	if (inet->defer_connect) {
116419f6d3f3SWei Wang 		err = tcp_connect(sk);
116519f6d3f3SWei Wang 		/* Same failure procedure as in tcp_v4/6_connect */
116619f6d3f3SWei Wang 		if (err) {
116719f6d3f3SWei Wang 			tcp_set_state(sk, TCP_CLOSE);
116819f6d3f3SWei Wang 			inet->inet_dport = 0;
116919f6d3f3SWei Wang 			sk->sk_route_caps = 0;
117019f6d3f3SWei Wang 		}
117119f6d3f3SWei Wang 	}
1172cf60af03SYuchung Cheng 	flags = (msg->msg_flags & MSG_DONTWAIT) ? O_NONBLOCK : 0;
1173ba615f67SWei Wang 	err = __inet_stream_connect(sk->sk_socket, uaddr,
11743979ad7eSWilly Tarreau 				    msg->msg_namelen, flags, 1);
11757db92362SWei Wang 	/* fastopen_req could already be freed in __inet_stream_connect
11767db92362SWei Wang 	 * if the connection times out or gets rst
11777db92362SWei Wang 	 */
11787db92362SWei Wang 	if (tp->fastopen_req) {
1179f5ddcbbbSEric Dumazet 		*copied = tp->fastopen_req->copied;
1180cf60af03SYuchung Cheng 		tcp_free_fastopen_req(tp);
11817db92362SWei Wang 		inet->defer_connect = 0;
11827db92362SWei Wang 	}
1183cf60af03SYuchung Cheng 	return err;
1184cf60af03SYuchung Cheng }
1185cf60af03SYuchung Cheng 
1186306b13ebSTom Herbert int tcp_sendmsg_locked(struct sock *sk, struct msghdr *msg, size_t size)
11871da177e4SLinus Torvalds {
11881da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
1189f214f915SWillem de Bruijn 	struct ubuf_info *uarg = NULL;
11901da177e4SLinus Torvalds 	struct sk_buff *skb;
1191c14ac945SSoheil Hassas Yeganeh 	struct sockcm_cookie sockc;
119257be5bdaSAl Viro 	int flags, err, copied = 0;
119357be5bdaSAl Viro 	int mss_now = 0, size_goal, copied_syn = 0;
11941a991488SEric Dumazet 	int process_backlog = 0;
119574d4a8f8SEric Dumazet 	bool zc = false;
11961da177e4SLinus Torvalds 	long timeo;
11971da177e4SLinus Torvalds 
11981da177e4SLinus Torvalds 	flags = msg->msg_flags;
1199f214f915SWillem de Bruijn 
12005cf4a853SVincent Whitchurch 	if (flags & MSG_ZEROCOPY && size && sock_flag(sk, SOCK_ZEROCOPY)) {
120175c119afSEric Dumazet 		skb = tcp_write_queue_tail(sk);
1202f214f915SWillem de Bruijn 		uarg = sock_zerocopy_realloc(sk, size, skb_zcopy(skb));
1203f214f915SWillem de Bruijn 		if (!uarg) {
1204f214f915SWillem de Bruijn 			err = -ENOBUFS;
1205f214f915SWillem de Bruijn 			goto out_err;
1206f214f915SWillem de Bruijn 		}
1207f214f915SWillem de Bruijn 
1208dead7cdbSEric Dumazet 		zc = sk->sk_route_caps & NETIF_F_SG;
120902583adeSWillem de Bruijn 		if (!zc)
1210f214f915SWillem de Bruijn 			uarg->zerocopy = 0;
1211f214f915SWillem de Bruijn 	}
1212f214f915SWillem de Bruijn 
121316ae6aa1SYuchung Cheng 	if (unlikely(flags & MSG_FASTOPEN || inet_sk(sk)->defer_connect) &&
121416ae6aa1SYuchung Cheng 	    !tp->repair) {
1215f859a448SWillem de Bruijn 		err = tcp_sendmsg_fastopen(sk, msg, &copied_syn, size, uarg);
1216cf60af03SYuchung Cheng 		if (err == -EINPROGRESS && copied_syn > 0)
1217cf60af03SYuchung Cheng 			goto out;
1218cf60af03SYuchung Cheng 		else if (err)
1219cf60af03SYuchung Cheng 			goto out_err;
1220cf60af03SYuchung Cheng 	}
1221cf60af03SYuchung Cheng 
12221da177e4SLinus Torvalds 	timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
12231da177e4SLinus Torvalds 
1224d7722e85SSoheil Hassas Yeganeh 	tcp_rate_check_app_limited(sk);  /* is sending application-limited? */
1225d7722e85SSoheil Hassas Yeganeh 
12268336886fSJerry Chu 	/* Wait for a connection to finish. One exception is TCP Fast Open
12278336886fSJerry Chu 	 * (passive side) where data is allowed to be sent before a connection
12288336886fSJerry Chu 	 * is fully established.
12298336886fSJerry Chu 	 */
12308336886fSJerry Chu 	if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) &&
12318336886fSJerry Chu 	    !tcp_passive_fastopen(sk)) {
1232686a5624SYuvaraja Mariappan 		err = sk_stream_wait_connect(sk, &timeo);
1233686a5624SYuvaraja Mariappan 		if (err != 0)
1234cf60af03SYuchung Cheng 			goto do_error;
12358336886fSJerry Chu 	}
12361da177e4SLinus Torvalds 
1237c0e88ff0SPavel Emelyanov 	if (unlikely(tp->repair)) {
1238c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_RECV_QUEUE) {
1239c0e88ff0SPavel Emelyanov 			copied = tcp_send_rcvq(sk, msg, size);
12405924f17aSChristoph Paasch 			goto out_nopush;
1241c0e88ff0SPavel Emelyanov 		}
1242c0e88ff0SPavel Emelyanov 
1243c0e88ff0SPavel Emelyanov 		err = -EINVAL;
1244c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_NO_QUEUE)
1245c0e88ff0SPavel Emelyanov 			goto out_err;
1246c0e88ff0SPavel Emelyanov 
1247c0e88ff0SPavel Emelyanov 		/* 'common' sending to sendq */
1248c0e88ff0SPavel Emelyanov 	}
1249c0e88ff0SPavel Emelyanov 
1250657a0667SWillem de Bruijn 	sockcm_init(&sockc, sk);
1251c14ac945SSoheil Hassas Yeganeh 	if (msg->msg_controllen) {
1252c14ac945SSoheil Hassas Yeganeh 		err = sock_cmsg_send(sk, msg, &sockc);
1253c14ac945SSoheil Hassas Yeganeh 		if (unlikely(err)) {
1254c14ac945SSoheil Hassas Yeganeh 			err = -EINVAL;
1255c14ac945SSoheil Hassas Yeganeh 			goto out_err;
1256c14ac945SSoheil Hassas Yeganeh 		}
1257c14ac945SSoheil Hassas Yeganeh 	}
1258c14ac945SSoheil Hassas Yeganeh 
12591da177e4SLinus Torvalds 	/* This should be in poll */
12609cd3e072SEric Dumazet 	sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
12611da177e4SLinus Torvalds 
12621da177e4SLinus Torvalds 	/* Ok commence sending. */
12631da177e4SLinus Torvalds 	copied = 0;
12641da177e4SLinus Torvalds 
1265d41a69f1SEric Dumazet restart:
1266d41a69f1SEric Dumazet 	mss_now = tcp_send_mss(sk, &size_goal, flags);
1267d41a69f1SEric Dumazet 
12681da177e4SLinus Torvalds 	err = -EPIPE;
12691da177e4SLinus Torvalds 	if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
127079d8665bSEric Dumazet 		goto do_error;
12711da177e4SLinus Torvalds 
127201e97e65SAl Viro 	while (msg_data_left(msg)) {
12736828b92bSHerbert Xu 		int copy = 0;
12741da177e4SLinus Torvalds 
1275fe067e8aSDavid S. Miller 		skb = tcp_write_queue_tail(sk);
127665ec6097SEric Dumazet 		if (skb)
127765ec6097SEric Dumazet 			copy = size_goal - skb->len;
12781da177e4SLinus Torvalds 
1279c134ecb8SMartin KaFai Lau 		if (copy <= 0 || !tcp_skb_can_collapse_to(skb)) {
12803613b3dbSEric Dumazet 			bool first_skb;
12813613b3dbSEric Dumazet 
12821da177e4SLinus Torvalds new_segment:
12831da177e4SLinus Torvalds 			if (!sk_stream_memory_free(sk))
1284*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
12851da177e4SLinus Torvalds 
12861a991488SEric Dumazet 			if (unlikely(process_backlog >= 16)) {
12871a991488SEric Dumazet 				process_backlog = 0;
12881a991488SEric Dumazet 				if (sk_flush_backlog(sk))
1289d41a69f1SEric Dumazet 					goto restart;
1290d4011239SEric Dumazet 			}
129175c119afSEric Dumazet 			first_skb = tcp_rtx_and_write_queues_empty(sk);
1292472c2e07SEric Dumazet 			skb = sk_stream_alloc_skb(sk, 0, sk->sk_allocation,
12933613b3dbSEric Dumazet 						  first_skb);
12941da177e4SLinus Torvalds 			if (!skb)
1295*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
12961da177e4SLinus Torvalds 
12971a991488SEric Dumazet 			process_backlog++;
129884fa7933SPatrick McHardy 			skb->ip_summed = CHECKSUM_PARTIAL;
12991da177e4SLinus Torvalds 
13009e412ba7SIlpo Järvinen 			skb_entail(sk, skb);
1301c1b4a7e6SDavid S. Miller 			copy = size_goal;
13029d186cacSAndrey Vagin 
13039d186cacSAndrey Vagin 			/* All packets are restored as if they have
1304d3edd06eSEric Dumazet 			 * already been sent. skb_mstamp_ns isn't set to
13059d186cacSAndrey Vagin 			 * avoid wrong rtt estimation.
13069d186cacSAndrey Vagin 			 */
13079d186cacSAndrey Vagin 			if (tp->repair)
13089d186cacSAndrey Vagin 				TCP_SKB_CB(skb)->sacked |= TCPCB_REPAIRED;
13091da177e4SLinus Torvalds 		}
13101da177e4SLinus Torvalds 
13111da177e4SLinus Torvalds 		/* Try to append data to the end of skb. */
131201e97e65SAl Viro 		if (copy > msg_data_left(msg))
131301e97e65SAl Viro 			copy = msg_data_left(msg);
13141da177e4SLinus Torvalds 
13151da177e4SLinus Torvalds 		/* Where to copy to? */
131602583adeSWillem de Bruijn 		if (skb_availroom(skb) > 0 && !zc) {
13171da177e4SLinus Torvalds 			/* We have some space in skb head. Superb! */
1318a21d4572SEric Dumazet 			copy = min_t(int, copy, skb_availroom(skb));
131957be5bdaSAl Viro 			err = skb_add_data_nocache(sk, skb, &msg->msg_iter, copy);
1320c6e1a0d1STom Herbert 			if (err)
13211da177e4SLinus Torvalds 				goto do_fault;
132202583adeSWillem de Bruijn 		} else if (!zc) {
13235640f768SEric Dumazet 			bool merge = true;
13241da177e4SLinus Torvalds 			int i = skb_shinfo(skb)->nr_frags;
13255640f768SEric Dumazet 			struct page_frag *pfrag = sk_page_frag(sk);
1326761965eaSEric Dumazet 
13275640f768SEric Dumazet 			if (!sk_page_frag_refill(sk, pfrag))
1328*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
1329761965eaSEric Dumazet 
13305640f768SEric Dumazet 			if (!skb_can_coalesce(skb, i, pfrag->page,
13315640f768SEric Dumazet 					      pfrag->offset)) {
133274d4a8f8SEric Dumazet 				if (i >= sysctl_max_skb_frags) {
13331da177e4SLinus Torvalds 					tcp_mark_push(tp, skb);
13341da177e4SLinus Torvalds 					goto new_segment;
13351da177e4SLinus Torvalds 				}
13365640f768SEric Dumazet 				merge = false;
13375640f768SEric Dumazet 			}
1338ef015786SHerbert Xu 
13395640f768SEric Dumazet 			copy = min_t(int, copy, pfrag->size - pfrag->offset);
1340ef015786SHerbert Xu 
13413ab224beSHideo Aoki 			if (!sk_wmem_schedule(sk, copy))
1342*afb83012SSoheil Hassas Yeganeh 				goto wait_for_space;
13431da177e4SLinus Torvalds 
134457be5bdaSAl Viro 			err = skb_copy_to_page_nocache(sk, &msg->msg_iter, skb,
13455640f768SEric Dumazet 						       pfrag->page,
13465640f768SEric Dumazet 						       pfrag->offset,
13475640f768SEric Dumazet 						       copy);
13485640f768SEric Dumazet 			if (err)
13491da177e4SLinus Torvalds 				goto do_error;
13501da177e4SLinus Torvalds 
13511da177e4SLinus Torvalds 			/* Update the skb. */
13521da177e4SLinus Torvalds 			if (merge) {
13539e903e08SEric Dumazet 				skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy);
13541da177e4SLinus Torvalds 			} else {
13555640f768SEric Dumazet 				skb_fill_page_desc(skb, i, pfrag->page,
13565640f768SEric Dumazet 						   pfrag->offset, copy);
13574e33e346SEric Dumazet 				page_ref_inc(pfrag->page);
13581da177e4SLinus Torvalds 			}
13595640f768SEric Dumazet 			pfrag->offset += copy;
1360f214f915SWillem de Bruijn 		} else {
1361f214f915SWillem de Bruijn 			err = skb_zerocopy_iter_stream(sk, skb, msg, copy, uarg);
1362111856c7SWillem de Bruijn 			if (err == -EMSGSIZE || err == -EEXIST) {
1363111856c7SWillem de Bruijn 				tcp_mark_push(tp, skb);
1364f214f915SWillem de Bruijn 				goto new_segment;
1365111856c7SWillem de Bruijn 			}
1366f214f915SWillem de Bruijn 			if (err < 0)
1367f214f915SWillem de Bruijn 				goto do_error;
1368f214f915SWillem de Bruijn 			copy = err;
13691da177e4SLinus Torvalds 		}
13701da177e4SLinus Torvalds 
13711da177e4SLinus Torvalds 		if (!copied)
13724de075e0SEric Dumazet 			TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH;
13731da177e4SLinus Torvalds 
13740f317464SEric Dumazet 		WRITE_ONCE(tp->write_seq, tp->write_seq + copy);
13751da177e4SLinus Torvalds 		TCP_SKB_CB(skb)->end_seq += copy;
1376cd7d8498SEric Dumazet 		tcp_skb_pcount_set(skb, 0);
13771da177e4SLinus Torvalds 
13781da177e4SLinus Torvalds 		copied += copy;
137901e97e65SAl Viro 		if (!msg_data_left(msg)) {
1380c134ecb8SMartin KaFai Lau 			if (unlikely(flags & MSG_EOR))
1381c134ecb8SMartin KaFai Lau 				TCP_SKB_CB(skb)->eor = 1;
13821da177e4SLinus Torvalds 			goto out;
13834ed2d765SWillem de Bruijn 		}
13841da177e4SLinus Torvalds 
138565ec6097SEric Dumazet 		if (skb->len < size_goal || (flags & MSG_OOB) || unlikely(tp->repair))
13861da177e4SLinus Torvalds 			continue;
13871da177e4SLinus Torvalds 
13881da177e4SLinus Torvalds 		if (forced_push(tp)) {
13891da177e4SLinus Torvalds 			tcp_mark_push(tp, skb);
13909e412ba7SIlpo Järvinen 			__tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);
1391fe067e8aSDavid S. Miller 		} else if (skb == tcp_send_head(sk))
13921da177e4SLinus Torvalds 			tcp_push_one(sk, mss_now);
13931da177e4SLinus Torvalds 		continue;
13941da177e4SLinus Torvalds 
1395*afb83012SSoheil Hassas Yeganeh wait_for_space:
13961da177e4SLinus Torvalds 		set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1397ec342325SAndrew Vagin 		if (copied)
1398f54b3111SEric Dumazet 			tcp_push(sk, flags & ~MSG_MORE, mss_now,
1399f54b3111SEric Dumazet 				 TCP_NAGLE_PUSH, size_goal);
14001da177e4SLinus Torvalds 
1401686a5624SYuvaraja Mariappan 		err = sk_stream_wait_memory(sk, &timeo);
1402686a5624SYuvaraja Mariappan 		if (err != 0)
14031da177e4SLinus Torvalds 			goto do_error;
14041da177e4SLinus Torvalds 
14050c54b85fSIlpo Järvinen 		mss_now = tcp_send_mss(sk, &size_goal, flags);
14061da177e4SLinus Torvalds 	}
14071da177e4SLinus Torvalds 
14081da177e4SLinus Torvalds out:
1409ad02c4f5SSoheil Hassas Yeganeh 	if (copied) {
14104e8cc228SEric Dumazet 		tcp_tx_timestamp(sk, sockc.tsflags);
1411f54b3111SEric Dumazet 		tcp_push(sk, flags, mss_now, tp->nonagle, size_goal);
1412ad02c4f5SSoheil Hassas Yeganeh 	}
14135924f17aSChristoph Paasch out_nopush:
1414f214f915SWillem de Bruijn 	sock_zerocopy_put(uarg);
1415cf60af03SYuchung Cheng 	return copied + copied_syn;
14161da177e4SLinus Torvalds 
14171da177e4SLinus Torvalds do_error:
1418fdfc5c85SEric Dumazet 	skb = tcp_write_queue_tail(sk);
1419fdfc5c85SEric Dumazet do_fault:
1420fdfc5c85SEric Dumazet 	tcp_remove_empty_skb(sk, skb);
1421fdfc5c85SEric Dumazet 
1422cf60af03SYuchung Cheng 	if (copied + copied_syn)
14231da177e4SLinus Torvalds 		goto out;
14241da177e4SLinus Torvalds out_err:
142552900d22SWillem de Bruijn 	sock_zerocopy_put_abort(uarg, true);
14261da177e4SLinus Torvalds 	err = sk_stream_error(sk, flags, err);
1427ce5ec440SJason Baron 	/* make sure we wake any epoll edge trigger waiter */
1428216808c6SEric Dumazet 	if (unlikely(tcp_rtx_and_write_queues_empty(sk) && err == -EAGAIN)) {
1429ce5ec440SJason Baron 		sk->sk_write_space(sk);
1430b0f71bd3SFrancis Yan 		tcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED);
1431b0f71bd3SFrancis Yan 	}
14321da177e4SLinus Torvalds 	return err;
14331da177e4SLinus Torvalds }
1434774c4673SDavid S. Miller EXPORT_SYMBOL_GPL(tcp_sendmsg_locked);
1435306b13ebSTom Herbert 
1436306b13ebSTom Herbert int tcp_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
1437306b13ebSTom Herbert {
1438306b13ebSTom Herbert 	int ret;
1439306b13ebSTom Herbert 
1440306b13ebSTom Herbert 	lock_sock(sk);
1441306b13ebSTom Herbert 	ret = tcp_sendmsg_locked(sk, msg, size);
1442306b13ebSTom Herbert 	release_sock(sk);
1443306b13ebSTom Herbert 
1444306b13ebSTom Herbert 	return ret;
1445306b13ebSTom Herbert }
14464bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_sendmsg);
14471da177e4SLinus Torvalds 
14481da177e4SLinus Torvalds /*
14491da177e4SLinus Torvalds  *	Handle reading urgent data. BSD has very simple semantics for
14501da177e4SLinus Torvalds  *	this, no blocking and very strange errors 8)
14511da177e4SLinus Torvalds  */
14521da177e4SLinus Torvalds 
1453377f0a08SRami Rosen static int tcp_recv_urg(struct sock *sk, struct msghdr *msg, int len, int flags)
14541da177e4SLinus Torvalds {
14551da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
14561da177e4SLinus Torvalds 
14571da177e4SLinus Torvalds 	/* No URG data to read. */
14581da177e4SLinus Torvalds 	if (sock_flag(sk, SOCK_URGINLINE) || !tp->urg_data ||
14591da177e4SLinus Torvalds 	    tp->urg_data == TCP_URG_READ)
14601da177e4SLinus Torvalds 		return -EINVAL;	/* Yes this is right ! */
14611da177e4SLinus Torvalds 
14621da177e4SLinus Torvalds 	if (sk->sk_state == TCP_CLOSE && !sock_flag(sk, SOCK_DONE))
14631da177e4SLinus Torvalds 		return -ENOTCONN;
14641da177e4SLinus Torvalds 
14651da177e4SLinus Torvalds 	if (tp->urg_data & TCP_URG_VALID) {
14661da177e4SLinus Torvalds 		int err = 0;
14671da177e4SLinus Torvalds 		char c = tp->urg_data;
14681da177e4SLinus Torvalds 
14691da177e4SLinus Torvalds 		if (!(flags & MSG_PEEK))
14701da177e4SLinus Torvalds 			tp->urg_data = TCP_URG_READ;
14711da177e4SLinus Torvalds 
14721da177e4SLinus Torvalds 		/* Read urgent data. */
14731da177e4SLinus Torvalds 		msg->msg_flags |= MSG_OOB;
14741da177e4SLinus Torvalds 
14751da177e4SLinus Torvalds 		if (len > 0) {
14761da177e4SLinus Torvalds 			if (!(flags & MSG_TRUNC))
14777eab8d9eSAl Viro 				err = memcpy_to_msg(msg, &c, 1);
14781da177e4SLinus Torvalds 			len = 1;
14791da177e4SLinus Torvalds 		} else
14801da177e4SLinus Torvalds 			msg->msg_flags |= MSG_TRUNC;
14811da177e4SLinus Torvalds 
14821da177e4SLinus Torvalds 		return err ? -EFAULT : len;
14831da177e4SLinus Torvalds 	}
14841da177e4SLinus Torvalds 
14851da177e4SLinus Torvalds 	if (sk->sk_state == TCP_CLOSE || (sk->sk_shutdown & RCV_SHUTDOWN))
14861da177e4SLinus Torvalds 		return 0;
14871da177e4SLinus Torvalds 
14881da177e4SLinus Torvalds 	/* Fixed the recv(..., MSG_OOB) behaviour.  BSD docs and
14891da177e4SLinus Torvalds 	 * the available implementations agree in this case:
14901da177e4SLinus Torvalds 	 * this call should never block, independent of the
14911da177e4SLinus Torvalds 	 * blocking state of the socket.
14921da177e4SLinus Torvalds 	 * Mike <pall@rz.uni-karlsruhe.de>
14931da177e4SLinus Torvalds 	 */
14941da177e4SLinus Torvalds 	return -EAGAIN;
14951da177e4SLinus Torvalds }
14961da177e4SLinus Torvalds 
1497c0e88ff0SPavel Emelyanov static int tcp_peek_sndq(struct sock *sk, struct msghdr *msg, int len)
1498c0e88ff0SPavel Emelyanov {
1499c0e88ff0SPavel Emelyanov 	struct sk_buff *skb;
1500c0e88ff0SPavel Emelyanov 	int copied = 0, err = 0;
1501c0e88ff0SPavel Emelyanov 
1502c0e88ff0SPavel Emelyanov 	/* XXX -- need to support SO_PEEK_OFF */
1503c0e88ff0SPavel Emelyanov 
150475c119afSEric Dumazet 	skb_rbtree_walk(skb, &sk->tcp_rtx_queue) {
150575c119afSEric Dumazet 		err = skb_copy_datagram_msg(skb, 0, msg, skb->len);
150675c119afSEric Dumazet 		if (err)
150775c119afSEric Dumazet 			return err;
150875c119afSEric Dumazet 		copied += skb->len;
150975c119afSEric Dumazet 	}
151075c119afSEric Dumazet 
1511c0e88ff0SPavel Emelyanov 	skb_queue_walk(&sk->sk_write_queue, skb) {
151251f3d02bSDavid S. Miller 		err = skb_copy_datagram_msg(skb, 0, msg, skb->len);
1513c0e88ff0SPavel Emelyanov 		if (err)
1514c0e88ff0SPavel Emelyanov 			break;
1515c0e88ff0SPavel Emelyanov 
1516c0e88ff0SPavel Emelyanov 		copied += skb->len;
1517c0e88ff0SPavel Emelyanov 	}
1518c0e88ff0SPavel Emelyanov 
1519c0e88ff0SPavel Emelyanov 	return err ?: copied;
1520c0e88ff0SPavel Emelyanov }
1521c0e88ff0SPavel Emelyanov 
15221da177e4SLinus Torvalds /* Clean up the receive buffer for full frames taken by the user,
15231da177e4SLinus Torvalds  * then send an ACK if necessary.  COPIED is the number of bytes
15241da177e4SLinus Torvalds  * tcp_recvmsg has given to the user so far, it speeds up the
15251da177e4SLinus Torvalds  * calculation of whether or not we must ACK for the sake of
15261da177e4SLinus Torvalds  * a window update.
15271da177e4SLinus Torvalds  */
1528c76c6956SPaolo Abeni void tcp_cleanup_rbuf(struct sock *sk, int copied)
15291da177e4SLinus Torvalds {
15301da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
1531a2a385d6SEric Dumazet 	bool time_to_ack = false;
15321da177e4SLinus Torvalds 
15331da177e4SLinus Torvalds 	struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
15341da177e4SLinus Torvalds 
1535d792c100SIlpo Järvinen 	WARN(skb && !before(tp->copied_seq, TCP_SKB_CB(skb)->end_seq),
15362af6fd8bSJoe Perches 	     "cleanup rbuf bug: copied %X seq %X rcvnxt %X\n",
1537d792c100SIlpo Järvinen 	     tp->copied_seq, TCP_SKB_CB(skb)->end_seq, tp->rcv_nxt);
15381da177e4SLinus Torvalds 
1539463c84b9SArnaldo Carvalho de Melo 	if (inet_csk_ack_scheduled(sk)) {
1540463c84b9SArnaldo Carvalho de Melo 		const struct inet_connection_sock *icsk = inet_csk(sk);
15411da177e4SLinus Torvalds 		   /* Delayed ACKs frequently hit locked sockets during bulk
15421da177e4SLinus Torvalds 		    * receive. */
1543463c84b9SArnaldo Carvalho de Melo 		if (icsk->icsk_ack.blocked ||
15441da177e4SLinus Torvalds 		    /* Once-per-two-segments ACK was not sent by tcp_input.c */
1545463c84b9SArnaldo Carvalho de Melo 		    tp->rcv_nxt - tp->rcv_wup > icsk->icsk_ack.rcv_mss ||
15461da177e4SLinus Torvalds 		    /*
15471da177e4SLinus Torvalds 		     * If this read emptied read buffer, we send ACK, if
15481da177e4SLinus Torvalds 		     * connection is not bidirectional, user drained
15491da177e4SLinus Torvalds 		     * receive buffer and there was a small segment
15501da177e4SLinus Torvalds 		     * in queue.
15511da177e4SLinus Torvalds 		     */
15521ef9696cSAlexey Kuznetsov 		    (copied > 0 &&
15531ef9696cSAlexey Kuznetsov 		     ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED2) ||
15541ef9696cSAlexey Kuznetsov 		      ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED) &&
155531954cd8SWei Wang 		       !inet_csk_in_pingpong_mode(sk))) &&
15561ef9696cSAlexey Kuznetsov 		      !atomic_read(&sk->sk_rmem_alloc)))
1557a2a385d6SEric Dumazet 			time_to_ack = true;
15581da177e4SLinus Torvalds 	}
15591da177e4SLinus Torvalds 
15601da177e4SLinus Torvalds 	/* We send an ACK if we can now advertise a non-zero window
15611da177e4SLinus Torvalds 	 * which has been raised "significantly".
15621da177e4SLinus Torvalds 	 *
15631da177e4SLinus Torvalds 	 * Even if window raised up to infinity, do not send window open ACK
15641da177e4SLinus Torvalds 	 * in states, where we will not receive more. It is useless.
15651da177e4SLinus Torvalds 	 */
15661da177e4SLinus Torvalds 	if (copied > 0 && !time_to_ack && !(sk->sk_shutdown & RCV_SHUTDOWN)) {
15671da177e4SLinus Torvalds 		__u32 rcv_window_now = tcp_receive_window(tp);
15681da177e4SLinus Torvalds 
15691da177e4SLinus Torvalds 		/* Optimize, __tcp_select_window() is not cheap. */
15701da177e4SLinus Torvalds 		if (2*rcv_window_now <= tp->window_clamp) {
15711da177e4SLinus Torvalds 			__u32 new_window = __tcp_select_window(sk);
15721da177e4SLinus Torvalds 
15731da177e4SLinus Torvalds 			/* Send ACK now, if this read freed lots of space
15741da177e4SLinus Torvalds 			 * in our buffer. Certainly, new_window is new window.
15751da177e4SLinus Torvalds 			 * We can advertise it now, if it is not less than current one.
15761da177e4SLinus Torvalds 			 * "Lots" means "at least twice" here.
15771da177e4SLinus Torvalds 			 */
15781da177e4SLinus Torvalds 			if (new_window && new_window >= 2 * rcv_window_now)
1579a2a385d6SEric Dumazet 				time_to_ack = true;
15801da177e4SLinus Torvalds 		}
15811da177e4SLinus Torvalds 	}
15821da177e4SLinus Torvalds 	if (time_to_ack)
15831da177e4SLinus Torvalds 		tcp_send_ack(sk);
15841da177e4SLinus Torvalds }
15851da177e4SLinus Torvalds 
1586f26845b4SEric Dumazet static struct sk_buff *tcp_recv_skb(struct sock *sk, u32 seq, u32 *off)
15871da177e4SLinus Torvalds {
15881da177e4SLinus Torvalds 	struct sk_buff *skb;
15891da177e4SLinus Torvalds 	u32 offset;
15901da177e4SLinus Torvalds 
1591f26845b4SEric Dumazet 	while ((skb = skb_peek(&sk->sk_receive_queue)) != NULL) {
15921da177e4SLinus Torvalds 		offset = seq - TCP_SKB_CB(skb)->seq;
15939d691539SEric Dumazet 		if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) {
15949d691539SEric Dumazet 			pr_err_once("%s: found a SYN, please report !\n", __func__);
15951da177e4SLinus Torvalds 			offset--;
15969d691539SEric Dumazet 		}
1597e11ecddfSEric Dumazet 		if (offset < skb->len || (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)) {
15981da177e4SLinus Torvalds 			*off = offset;
15991da177e4SLinus Torvalds 			return skb;
16001da177e4SLinus Torvalds 		}
1601f26845b4SEric Dumazet 		/* This looks weird, but this can happen if TCP collapsing
1602f26845b4SEric Dumazet 		 * splitted a fat GRO packet, while we released socket lock
1603f26845b4SEric Dumazet 		 * in skb_splice_bits()
1604f26845b4SEric Dumazet 		 */
16057bced397SDan Williams 		sk_eat_skb(sk, skb);
16061da177e4SLinus Torvalds 	}
16071da177e4SLinus Torvalds 	return NULL;
16081da177e4SLinus Torvalds }
16091da177e4SLinus Torvalds 
16101da177e4SLinus Torvalds /*
16111da177e4SLinus Torvalds  * This routine provides an alternative to tcp_recvmsg() for routines
16121da177e4SLinus Torvalds  * that would like to handle copying from skbuffs directly in 'sendfile'
16131da177e4SLinus Torvalds  * fashion.
16141da177e4SLinus Torvalds  * Note:
16151da177e4SLinus Torvalds  *	- It is assumed that the socket was locked by the caller.
16161da177e4SLinus Torvalds  *	- The routine does not block.
16171da177e4SLinus Torvalds  *	- At present, there is no support for reading OOB data
16181da177e4SLinus Torvalds  *	  or for 'peeking' the socket using this routine
16191da177e4SLinus Torvalds  *	  (although both would be easy to implement).
16201da177e4SLinus Torvalds  */
16211da177e4SLinus Torvalds int tcp_read_sock(struct sock *sk, read_descriptor_t *desc,
16221da177e4SLinus Torvalds 		  sk_read_actor_t recv_actor)
16231da177e4SLinus Torvalds {
16241da177e4SLinus Torvalds 	struct sk_buff *skb;
16251da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
16261da177e4SLinus Torvalds 	u32 seq = tp->copied_seq;
16271da177e4SLinus Torvalds 	u32 offset;
16281da177e4SLinus Torvalds 	int copied = 0;
16291da177e4SLinus Torvalds 
16301da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN)
16311da177e4SLinus Torvalds 		return -ENOTCONN;
16321da177e4SLinus Torvalds 	while ((skb = tcp_recv_skb(sk, seq, &offset)) != NULL) {
16331da177e4SLinus Torvalds 		if (offset < skb->len) {
1634374e7b59SOctavian Purdila 			int used;
1635374e7b59SOctavian Purdila 			size_t len;
16361da177e4SLinus Torvalds 
16371da177e4SLinus Torvalds 			len = skb->len - offset;
16381da177e4SLinus Torvalds 			/* Stop reading if we hit a patch of urgent data */
16391da177e4SLinus Torvalds 			if (tp->urg_data) {
16401da177e4SLinus Torvalds 				u32 urg_offset = tp->urg_seq - seq;
16411da177e4SLinus Torvalds 				if (urg_offset < len)
16421da177e4SLinus Torvalds 					len = urg_offset;
16431da177e4SLinus Torvalds 				if (!len)
16441da177e4SLinus Torvalds 					break;
16451da177e4SLinus Torvalds 			}
16461da177e4SLinus Torvalds 			used = recv_actor(desc, skb, offset, len);
1647ff905b1eSEric Dumazet 			if (used <= 0) {
1648ddb61a57SJens Axboe 				if (!copied)
1649ddb61a57SJens Axboe 					copied = used;
1650ddb61a57SJens Axboe 				break;
1651ddb61a57SJens Axboe 			} else if (used <= len) {
16521da177e4SLinus Torvalds 				seq += used;
16531da177e4SLinus Torvalds 				copied += used;
16541da177e4SLinus Torvalds 				offset += used;
16551da177e4SLinus Torvalds 			}
165602275a2eSWilly Tarreau 			/* If recv_actor drops the lock (e.g. TCP splice
1657293ad604SOctavian Purdila 			 * receive) the skb pointer might be invalid when
1658293ad604SOctavian Purdila 			 * getting here: tcp_collapse might have deleted it
1659293ad604SOctavian Purdila 			 * while aggregating skbs from the socket queue.
1660293ad604SOctavian Purdila 			 */
1661293ad604SOctavian Purdila 			skb = tcp_recv_skb(sk, seq - 1, &offset);
166202275a2eSWilly Tarreau 			if (!skb)
16631da177e4SLinus Torvalds 				break;
166402275a2eSWilly Tarreau 			/* TCP coalescing might have appended data to the skb.
166502275a2eSWilly Tarreau 			 * Try to splice more frags
166602275a2eSWilly Tarreau 			 */
166702275a2eSWilly Tarreau 			if (offset + 1 != skb->len)
166802275a2eSWilly Tarreau 				continue;
16691da177e4SLinus Torvalds 		}
1670e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) {
16717bced397SDan Williams 			sk_eat_skb(sk, skb);
16721da177e4SLinus Torvalds 			++seq;
16731da177e4SLinus Torvalds 			break;
16741da177e4SLinus Torvalds 		}
16757bced397SDan Williams 		sk_eat_skb(sk, skb);
16761da177e4SLinus Torvalds 		if (!desc->count)
16771da177e4SLinus Torvalds 			break;
16787db48e98SEric Dumazet 		WRITE_ONCE(tp->copied_seq, seq);
16791da177e4SLinus Torvalds 	}
16807db48e98SEric Dumazet 	WRITE_ONCE(tp->copied_seq, seq);
16811da177e4SLinus Torvalds 
16821da177e4SLinus Torvalds 	tcp_rcv_space_adjust(sk);
16831da177e4SLinus Torvalds 
16841da177e4SLinus Torvalds 	/* Clean up data we have read: This will do ACK frames. */
1685f26845b4SEric Dumazet 	if (copied > 0) {
1686f26845b4SEric Dumazet 		tcp_recv_skb(sk, seq, &offset);
16870e4b4992SChris Leech 		tcp_cleanup_rbuf(sk, copied);
1688f26845b4SEric Dumazet 	}
16891da177e4SLinus Torvalds 	return copied;
16901da177e4SLinus Torvalds }
16914bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_read_sock);
16921da177e4SLinus Torvalds 
169332035585STom Herbert int tcp_peek_len(struct socket *sock)
169432035585STom Herbert {
169532035585STom Herbert 	return tcp_inq(sock->sk);
169632035585STom Herbert }
169732035585STom Herbert EXPORT_SYMBOL(tcp_peek_len);
169832035585STom Herbert 
1699d1361840SEric Dumazet /* Make sure sk_rcvbuf is big enough to satisfy SO_RCVLOWAT hint */
1700d1361840SEric Dumazet int tcp_set_rcvlowat(struct sock *sk, int val)
1701d1361840SEric Dumazet {
1702867f816bSSoheil Hassas Yeganeh 	int cap;
1703867f816bSSoheil Hassas Yeganeh 
1704867f816bSSoheil Hassas Yeganeh 	if (sk->sk_userlocks & SOCK_RCVBUF_LOCK)
1705867f816bSSoheil Hassas Yeganeh 		cap = sk->sk_rcvbuf >> 1;
1706867f816bSSoheil Hassas Yeganeh 	else
1707867f816bSSoheil Hassas Yeganeh 		cap = sock_net(sk)->ipv4.sysctl_tcp_rmem[2] >> 1;
1708867f816bSSoheil Hassas Yeganeh 	val = min(val, cap);
1709eac66402SEric Dumazet 	WRITE_ONCE(sk->sk_rcvlowat, val ? : 1);
171003f45c88SEric Dumazet 
171103f45c88SEric Dumazet 	/* Check if we need to signal EPOLLIN right now */
171203f45c88SEric Dumazet 	tcp_data_ready(sk);
171303f45c88SEric Dumazet 
1714d1361840SEric Dumazet 	if (sk->sk_userlocks & SOCK_RCVBUF_LOCK)
1715d1361840SEric Dumazet 		return 0;
1716d1361840SEric Dumazet 
1717d1361840SEric Dumazet 	val <<= 1;
1718d1361840SEric Dumazet 	if (val > sk->sk_rcvbuf) {
1719ebb3b78dSEric Dumazet 		WRITE_ONCE(sk->sk_rcvbuf, val);
1720d1361840SEric Dumazet 		tcp_sk(sk)->window_clamp = tcp_win_from_space(sk, val);
1721d1361840SEric Dumazet 	}
1722d1361840SEric Dumazet 	return 0;
1723d1361840SEric Dumazet }
1724d1361840SEric Dumazet EXPORT_SYMBOL(tcp_set_rcvlowat);
1725d1361840SEric Dumazet 
172605255b82SEric Dumazet #ifdef CONFIG_MMU
172705255b82SEric Dumazet static const struct vm_operations_struct tcp_vm_ops = {
172805255b82SEric Dumazet };
172905255b82SEric Dumazet 
173093ab6cc6SEric Dumazet int tcp_mmap(struct file *file, struct socket *sock,
173193ab6cc6SEric Dumazet 	     struct vm_area_struct *vma)
173293ab6cc6SEric Dumazet {
173305255b82SEric Dumazet 	if (vma->vm_flags & (VM_WRITE | VM_EXEC))
173405255b82SEric Dumazet 		return -EPERM;
173505255b82SEric Dumazet 	vma->vm_flags &= ~(VM_MAYWRITE | VM_MAYEXEC);
173605255b82SEric Dumazet 
17373e4e28c5SMichel Lespinasse 	/* Instruct vm_insert_page() to not mmap_read_lock(mm) */
173805255b82SEric Dumazet 	vma->vm_flags |= VM_MIXEDMAP;
173905255b82SEric Dumazet 
174005255b82SEric Dumazet 	vma->vm_ops = &tcp_vm_ops;
174105255b82SEric Dumazet 	return 0;
174205255b82SEric Dumazet }
174305255b82SEric Dumazet EXPORT_SYMBOL(tcp_mmap);
174405255b82SEric Dumazet 
17453763a24cSArjun Roy static int tcp_zerocopy_vm_insert_batch(struct vm_area_struct *vma,
17463763a24cSArjun Roy 					struct page **pages,
17473763a24cSArjun Roy 					unsigned long pages_to_map,
17483763a24cSArjun Roy 					unsigned long *insert_addr,
17493763a24cSArjun Roy 					u32 *length_with_pending,
17503763a24cSArjun Roy 					u32 *seq,
17513763a24cSArjun Roy 					struct tcp_zerocopy_receive *zc)
17523763a24cSArjun Roy {
17533763a24cSArjun Roy 	unsigned long pages_remaining = pages_to_map;
17543763a24cSArjun Roy 	int bytes_mapped;
17553763a24cSArjun Roy 	int ret;
17563763a24cSArjun Roy 
17573763a24cSArjun Roy 	ret = vm_insert_pages(vma, *insert_addr, pages, &pages_remaining);
17583763a24cSArjun Roy 	bytes_mapped = PAGE_SIZE * (pages_to_map - pages_remaining);
17593763a24cSArjun Roy 	/* Even if vm_insert_pages fails, it may have partially succeeded in
17603763a24cSArjun Roy 	 * mapping (some but not all of the pages).
17613763a24cSArjun Roy 	 */
17623763a24cSArjun Roy 	*seq += bytes_mapped;
17633763a24cSArjun Roy 	*insert_addr += bytes_mapped;
17643763a24cSArjun Roy 	if (ret) {
17653763a24cSArjun Roy 		/* But if vm_insert_pages did fail, we have to unroll some state
17663763a24cSArjun Roy 		 * we speculatively touched before.
17673763a24cSArjun Roy 		 */
17683763a24cSArjun Roy 		const int bytes_not_mapped = PAGE_SIZE * pages_remaining;
17693763a24cSArjun Roy 		*length_with_pending -= bytes_not_mapped;
17703763a24cSArjun Roy 		zc->recv_skip_hint += bytes_not_mapped;
17713763a24cSArjun Roy 	}
17723763a24cSArjun Roy 	return ret;
17733763a24cSArjun Roy }
17743763a24cSArjun Roy 
177505255b82SEric Dumazet static int tcp_zerocopy_receive(struct sock *sk,
177605255b82SEric Dumazet 				struct tcp_zerocopy_receive *zc)
177705255b82SEric Dumazet {
177805255b82SEric Dumazet 	unsigned long address = (unsigned long)zc->address;
1779c208bdb9SSoheil Hassas Yeganeh 	u32 length = 0, seq, offset, zap_len;
17803763a24cSArjun Roy 	#define PAGE_BATCH_SIZE 8
17813763a24cSArjun Roy 	struct page *pages[PAGE_BATCH_SIZE];
178205255b82SEric Dumazet 	const skb_frag_t *frags = NULL;
178305255b82SEric Dumazet 	struct vm_area_struct *vma;
178405255b82SEric Dumazet 	struct sk_buff *skb = NULL;
17853763a24cSArjun Roy 	unsigned long pg_idx = 0;
17863763a24cSArjun Roy 	unsigned long curr_addr;
178793ab6cc6SEric Dumazet 	struct tcp_sock *tp;
17888f2b0293SSoheil Hassas Yeganeh 	int inq;
178993ab6cc6SEric Dumazet 	int ret;
179093ab6cc6SEric Dumazet 
179105255b82SEric Dumazet 	if (address & (PAGE_SIZE - 1) || address != zc->address)
179293ab6cc6SEric Dumazet 		return -EINVAL;
179393ab6cc6SEric Dumazet 
179493ab6cc6SEric Dumazet 	if (sk->sk_state == TCP_LISTEN)
179505255b82SEric Dumazet 		return -ENOTCONN;
179693ab6cc6SEric Dumazet 
179793ab6cc6SEric Dumazet 	sock_rps_record_flow(sk);
179893ab6cc6SEric Dumazet 
17993763a24cSArjun Roy 	tp = tcp_sk(sk);
18003763a24cSArjun Roy 
1801d8ed45c5SMichel Lespinasse 	mmap_read_lock(current->mm);
180293ab6cc6SEric Dumazet 
180305255b82SEric Dumazet 	vma = find_vma(current->mm, address);
1804e776af60SEric Dumazet 	if (!vma || vma->vm_start > address || vma->vm_ops != &tcp_vm_ops) {
1805d8ed45c5SMichel Lespinasse 		mmap_read_unlock(current->mm);
1806e776af60SEric Dumazet 		return -EINVAL;
1807e776af60SEric Dumazet 	}
180805255b82SEric Dumazet 	zc->length = min_t(unsigned long, zc->length, vma->vm_end - address);
180905255b82SEric Dumazet 
181005255b82SEric Dumazet 	seq = tp->copied_seq;
18118f2b0293SSoheil Hassas Yeganeh 	inq = tcp_inq(sk);
18128f2b0293SSoheil Hassas Yeganeh 	zc->length = min_t(u32, zc->length, inq);
1813c208bdb9SSoheil Hassas Yeganeh 	zap_len = zc->length & ~(PAGE_SIZE - 1);
1814c208bdb9SSoheil Hassas Yeganeh 	if (zap_len) {
1815c208bdb9SSoheil Hassas Yeganeh 		zap_page_range(vma, address, zap_len);
181605255b82SEric Dumazet 		zc->recv_skip_hint = 0;
18178f2b0293SSoheil Hassas Yeganeh 	} else {
1818c208bdb9SSoheil Hassas Yeganeh 		zc->recv_skip_hint = zc->length;
18198f2b0293SSoheil Hassas Yeganeh 	}
182005255b82SEric Dumazet 	ret = 0;
18213763a24cSArjun Roy 	curr_addr = address;
182205255b82SEric Dumazet 	while (length + PAGE_SIZE <= zc->length) {
182305255b82SEric Dumazet 		if (zc->recv_skip_hint < PAGE_SIZE) {
18243763a24cSArjun Roy 			/* If we're here, finish the current batch. */
18253763a24cSArjun Roy 			if (pg_idx) {
18263763a24cSArjun Roy 				ret = tcp_zerocopy_vm_insert_batch(vma, pages,
18273763a24cSArjun Roy 								   pg_idx,
18283763a24cSArjun Roy 								   &curr_addr,
18293763a24cSArjun Roy 								   &length,
18303763a24cSArjun Roy 								   &seq, zc);
18313763a24cSArjun Roy 				if (ret)
18323763a24cSArjun Roy 					goto out;
18333763a24cSArjun Roy 				pg_idx = 0;
18343763a24cSArjun Roy 			}
183505255b82SEric Dumazet 			if (skb) {
18360e627190SArjun Roy 				if (zc->recv_skip_hint > 0)
18370e627190SArjun Roy 					break;
183805255b82SEric Dumazet 				skb = skb->next;
183905255b82SEric Dumazet 				offset = seq - TCP_SKB_CB(skb)->seq;
184005255b82SEric Dumazet 			} else {
184193ab6cc6SEric Dumazet 				skb = tcp_recv_skb(sk, seq, &offset);
184205255b82SEric Dumazet 			}
184305255b82SEric Dumazet 			zc->recv_skip_hint = skb->len - offset;
184493ab6cc6SEric Dumazet 			offset -= skb_headlen(skb);
184505255b82SEric Dumazet 			if ((int)offset < 0 || skb_has_frag_list(skb))
184605255b82SEric Dumazet 				break;
184793ab6cc6SEric Dumazet 			frags = skb_shinfo(skb)->frags;
184893ab6cc6SEric Dumazet 			while (offset) {
1849d8e18a51SMatthew Wilcox (Oracle) 				if (skb_frag_size(frags) > offset)
185093ab6cc6SEric Dumazet 					goto out;
1851d8e18a51SMatthew Wilcox (Oracle) 				offset -= skb_frag_size(frags);
185293ab6cc6SEric Dumazet 				frags++;
185393ab6cc6SEric Dumazet 			}
185405255b82SEric Dumazet 		}
1855b54c9d5bSJonathan Lemon 		if (skb_frag_size(frags) != PAGE_SIZE || skb_frag_off(frags)) {
1856789762ceSSoheil Hassas Yeganeh 			int remaining = zc->recv_skip_hint;
1857789762ceSSoheil Hassas Yeganeh 
18583afb0961SEric Dumazet 			while (remaining && (skb_frag_size(frags) != PAGE_SIZE ||
1859b54c9d5bSJonathan Lemon 					     skb_frag_off(frags))) {
18603afb0961SEric Dumazet 				remaining -= skb_frag_size(frags);
1861789762ceSSoheil Hassas Yeganeh 				frags++;
1862789762ceSSoheil Hassas Yeganeh 			}
1863789762ceSSoheil Hassas Yeganeh 			zc->recv_skip_hint -= remaining;
186405255b82SEric Dumazet 			break;
1865789762ceSSoheil Hassas Yeganeh 		}
18663763a24cSArjun Roy 		pages[pg_idx] = skb_frag_page(frags);
18673763a24cSArjun Roy 		pg_idx++;
186805255b82SEric Dumazet 		length += PAGE_SIZE;
186905255b82SEric Dumazet 		zc->recv_skip_hint -= PAGE_SIZE;
187005255b82SEric Dumazet 		frags++;
18713763a24cSArjun Roy 		if (pg_idx == PAGE_BATCH_SIZE) {
18723763a24cSArjun Roy 			ret = tcp_zerocopy_vm_insert_batch(vma, pages, pg_idx,
18733763a24cSArjun Roy 							   &curr_addr, &length,
18743763a24cSArjun Roy 							   &seq, zc);
18753763a24cSArjun Roy 			if (ret)
18763763a24cSArjun Roy 				goto out;
18773763a24cSArjun Roy 			pg_idx = 0;
18783763a24cSArjun Roy 		}
18793763a24cSArjun Roy 	}
18803763a24cSArjun Roy 	if (pg_idx) {
18813763a24cSArjun Roy 		ret = tcp_zerocopy_vm_insert_batch(vma, pages, pg_idx,
18823763a24cSArjun Roy 						   &curr_addr, &length, &seq,
18833763a24cSArjun Roy 						   zc);
188493ab6cc6SEric Dumazet 	}
188505255b82SEric Dumazet out:
1886d8ed45c5SMichel Lespinasse 	mmap_read_unlock(current->mm);
188705255b82SEric Dumazet 	if (length) {
18887db48e98SEric Dumazet 		WRITE_ONCE(tp->copied_seq, seq);
188993ab6cc6SEric Dumazet 		tcp_rcv_space_adjust(sk);
189093ab6cc6SEric Dumazet 
189193ab6cc6SEric Dumazet 		/* Clean up data we have read: This will do ACK frames. */
189293ab6cc6SEric Dumazet 		tcp_recv_skb(sk, seq, &offset);
189305255b82SEric Dumazet 		tcp_cleanup_rbuf(sk, length);
189493ab6cc6SEric Dumazet 		ret = 0;
189505255b82SEric Dumazet 		if (length == zc->length)
189605255b82SEric Dumazet 			zc->recv_skip_hint = 0;
189705255b82SEric Dumazet 	} else {
189805255b82SEric Dumazet 		if (!zc->recv_skip_hint && sock_flag(sk, SOCK_DONE))
189905255b82SEric Dumazet 			ret = -EIO;
190005255b82SEric Dumazet 	}
190105255b82SEric Dumazet 	zc->length = length;
190293ab6cc6SEric Dumazet 	return ret;
190393ab6cc6SEric Dumazet }
190405255b82SEric Dumazet #endif
190593ab6cc6SEric Dumazet 
190698aaa913SMike Maloney static void tcp_update_recv_tstamps(struct sk_buff *skb,
19079718475eSDeepa Dinamani 				    struct scm_timestamping_internal *tss)
190898aaa913SMike Maloney {
190998aaa913SMike Maloney 	if (skb->tstamp)
19109718475eSDeepa Dinamani 		tss->ts[0] = ktime_to_timespec64(skb->tstamp);
191198aaa913SMike Maloney 	else
19129718475eSDeepa Dinamani 		tss->ts[0] = (struct timespec64) {0};
191398aaa913SMike Maloney 
191498aaa913SMike Maloney 	if (skb_hwtstamps(skb)->hwtstamp)
19159718475eSDeepa Dinamani 		tss->ts[2] = ktime_to_timespec64(skb_hwtstamps(skb)->hwtstamp);
191698aaa913SMike Maloney 	else
19179718475eSDeepa Dinamani 		tss->ts[2] = (struct timespec64) {0};
191898aaa913SMike Maloney }
191998aaa913SMike Maloney 
192098aaa913SMike Maloney /* Similar to __sock_recv_timestamp, but does not require an skb */
1921809a79e9SWei Yongjun static void tcp_recv_timestamp(struct msghdr *msg, const struct sock *sk,
19229718475eSDeepa Dinamani 			       struct scm_timestamping_internal *tss)
192398aaa913SMike Maloney {
1924887feae3SDeepa Dinamani 	int new_tstamp = sock_flag(sk, SOCK_TSTAMP_NEW);
192598aaa913SMike Maloney 	bool has_timestamping = false;
192698aaa913SMike Maloney 
192798aaa913SMike Maloney 	if (tss->ts[0].tv_sec || tss->ts[0].tv_nsec) {
192898aaa913SMike Maloney 		if (sock_flag(sk, SOCK_RCVTSTAMP)) {
192998aaa913SMike Maloney 			if (sock_flag(sk, SOCK_RCVTSTAMPNS)) {
1930887feae3SDeepa Dinamani 				if (new_tstamp) {
1931df1b4ba9SArnd Bergmann 					struct __kernel_timespec kts = {
1932df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
1933df1b4ba9SArnd Bergmann 						.tv_nsec = tss->ts[0].tv_nsec,
1934df1b4ba9SArnd Bergmann 					};
1935887feae3SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_NEW,
1936887feae3SDeepa Dinamani 						 sizeof(kts), &kts);
1937887feae3SDeepa Dinamani 				} else {
1938df1b4ba9SArnd Bergmann 					struct __kernel_old_timespec ts_old = {
1939df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
1940df1b4ba9SArnd Bergmann 						.tv_nsec = tss->ts[0].tv_nsec,
1941df1b4ba9SArnd Bergmann 					};
19427f1bc6e9SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_OLD,
19439718475eSDeepa Dinamani 						 sizeof(ts_old), &ts_old);
1944887feae3SDeepa Dinamani 				}
194598aaa913SMike Maloney 			} else {
1946887feae3SDeepa Dinamani 				if (new_tstamp) {
1947df1b4ba9SArnd Bergmann 					struct __kernel_sock_timeval stv = {
1948df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
1949df1b4ba9SArnd Bergmann 						.tv_usec = tss->ts[0].tv_nsec / 1000,
1950df1b4ba9SArnd Bergmann 					};
1951887feae3SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_NEW,
1952887feae3SDeepa Dinamani 						 sizeof(stv), &stv);
1953887feae3SDeepa Dinamani 				} else {
1954df1b4ba9SArnd Bergmann 					struct __kernel_old_timeval tv = {
1955df1b4ba9SArnd Bergmann 						.tv_sec = tss->ts[0].tv_sec,
1956df1b4ba9SArnd Bergmann 						.tv_usec = tss->ts[0].tv_nsec / 1000,
1957df1b4ba9SArnd Bergmann 					};
19587f1bc6e9SDeepa Dinamani 					put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_OLD,
195998aaa913SMike Maloney 						 sizeof(tv), &tv);
196098aaa913SMike Maloney 				}
196198aaa913SMike Maloney 			}
1962887feae3SDeepa Dinamani 		}
196398aaa913SMike Maloney 
196498aaa913SMike Maloney 		if (sk->sk_tsflags & SOF_TIMESTAMPING_SOFTWARE)
196598aaa913SMike Maloney 			has_timestamping = true;
196698aaa913SMike Maloney 		else
19679718475eSDeepa Dinamani 			tss->ts[0] = (struct timespec64) {0};
196898aaa913SMike Maloney 	}
196998aaa913SMike Maloney 
197098aaa913SMike Maloney 	if (tss->ts[2].tv_sec || tss->ts[2].tv_nsec) {
197198aaa913SMike Maloney 		if (sk->sk_tsflags & SOF_TIMESTAMPING_RAW_HARDWARE)
197298aaa913SMike Maloney 			has_timestamping = true;
197398aaa913SMike Maloney 		else
19749718475eSDeepa Dinamani 			tss->ts[2] = (struct timespec64) {0};
197598aaa913SMike Maloney 	}
197698aaa913SMike Maloney 
197798aaa913SMike Maloney 	if (has_timestamping) {
19789718475eSDeepa Dinamani 		tss->ts[1] = (struct timespec64) {0};
19799718475eSDeepa Dinamani 		if (sock_flag(sk, SOCK_TSTAMP_NEW))
19809718475eSDeepa Dinamani 			put_cmsg_scm_timestamping64(msg, tss);
19819718475eSDeepa Dinamani 		else
19829718475eSDeepa Dinamani 			put_cmsg_scm_timestamping(msg, tss);
198398aaa913SMike Maloney 	}
198498aaa913SMike Maloney }
198598aaa913SMike Maloney 
1986b75eba76SSoheil Hassas Yeganeh static int tcp_inq_hint(struct sock *sk)
1987b75eba76SSoheil Hassas Yeganeh {
1988b75eba76SSoheil Hassas Yeganeh 	const struct tcp_sock *tp = tcp_sk(sk);
1989b75eba76SSoheil Hassas Yeganeh 	u32 copied_seq = READ_ONCE(tp->copied_seq);
1990b75eba76SSoheil Hassas Yeganeh 	u32 rcv_nxt = READ_ONCE(tp->rcv_nxt);
1991b75eba76SSoheil Hassas Yeganeh 	int inq;
1992b75eba76SSoheil Hassas Yeganeh 
1993b75eba76SSoheil Hassas Yeganeh 	inq = rcv_nxt - copied_seq;
1994b75eba76SSoheil Hassas Yeganeh 	if (unlikely(inq < 0 || copied_seq != READ_ONCE(tp->copied_seq))) {
1995b75eba76SSoheil Hassas Yeganeh 		lock_sock(sk);
1996b75eba76SSoheil Hassas Yeganeh 		inq = tp->rcv_nxt - tp->copied_seq;
1997b75eba76SSoheil Hassas Yeganeh 		release_sock(sk);
1998b75eba76SSoheil Hassas Yeganeh 	}
19996466e715SSoheil Hassas Yeganeh 	/* After receiving a FIN, tell the user-space to continue reading
20006466e715SSoheil Hassas Yeganeh 	 * by returning a non-zero inq.
20016466e715SSoheil Hassas Yeganeh 	 */
20026466e715SSoheil Hassas Yeganeh 	if (inq == 0 && sock_flag(sk, SOCK_DONE))
20036466e715SSoheil Hassas Yeganeh 		inq = 1;
2004b75eba76SSoheil Hassas Yeganeh 	return inq;
2005b75eba76SSoheil Hassas Yeganeh }
2006b75eba76SSoheil Hassas Yeganeh 
20071da177e4SLinus Torvalds /*
20081da177e4SLinus Torvalds  *	This routine copies from a sock struct into the user buffer.
20091da177e4SLinus Torvalds  *
20101da177e4SLinus Torvalds  *	Technical note: in 2.3 we work on _locked_ socket, so that
20111da177e4SLinus Torvalds  *	tricks with *seq access order and skb->users are not required.
20121da177e4SLinus Torvalds  *	Probably, code can be easily improved even more.
20131da177e4SLinus Torvalds  */
20141da177e4SLinus Torvalds 
20151b784140SYing Xue int tcp_recvmsg(struct sock *sk, struct msghdr *msg, size_t len, int nonblock,
20161b784140SYing Xue 		int flags, int *addr_len)
20171da177e4SLinus Torvalds {
20181da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
20191da177e4SLinus Torvalds 	int copied = 0;
20201da177e4SLinus Torvalds 	u32 peek_seq;
20211da177e4SLinus Torvalds 	u32 *seq;
20221da177e4SLinus Torvalds 	unsigned long used;
2023b75eba76SSoheil Hassas Yeganeh 	int err, inq;
20241da177e4SLinus Torvalds 	int target;		/* Read at least this many bytes */
20251da177e4SLinus Torvalds 	long timeo;
2026dfbafc99SSabrina Dubroca 	struct sk_buff *skb, *last;
202777527313SIlpo Järvinen 	u32 urg_hole = 0;
20289718475eSDeepa Dinamani 	struct scm_timestamping_internal tss;
2029a5a7daa5SEric Dumazet 	int cmsg_flags;
20301da177e4SLinus Torvalds 
20314ed2d765SWillem de Bruijn 	if (unlikely(flags & MSG_ERRQUEUE))
2032f4713a3dSWillem de Bruijn 		return inet_recv_error(sk, msg, len, addr_len);
20334ed2d765SWillem de Bruijn 
20343f926af3SEric Dumazet 	if (sk_can_busy_loop(sk) && skb_queue_empty_lockless(&sk->sk_receive_queue) &&
2035cbf55001SEliezer Tamir 	    (sk->sk_state == TCP_ESTABLISHED))
2036cbf55001SEliezer Tamir 		sk_busy_loop(sk, nonblock);
2037d30e383bSEliezer Tamir 
20381da177e4SLinus Torvalds 	lock_sock(sk);
20391da177e4SLinus Torvalds 
20401da177e4SLinus Torvalds 	err = -ENOTCONN;
20411da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN)
20421da177e4SLinus Torvalds 		goto out;
20431da177e4SLinus Torvalds 
2044a5a7daa5SEric Dumazet 	cmsg_flags = tp->recvmsg_inq ? 1 : 0;
20451da177e4SLinus Torvalds 	timeo = sock_rcvtimeo(sk, nonblock);
20461da177e4SLinus Torvalds 
20471da177e4SLinus Torvalds 	/* Urgent data needs to be handled specially. */
20481da177e4SLinus Torvalds 	if (flags & MSG_OOB)
20491da177e4SLinus Torvalds 		goto recv_urg;
20501da177e4SLinus Torvalds 
2051c0e88ff0SPavel Emelyanov 	if (unlikely(tp->repair)) {
2052c0e88ff0SPavel Emelyanov 		err = -EPERM;
2053c0e88ff0SPavel Emelyanov 		if (!(flags & MSG_PEEK))
2054c0e88ff0SPavel Emelyanov 			goto out;
2055c0e88ff0SPavel Emelyanov 
2056c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_SEND_QUEUE)
2057c0e88ff0SPavel Emelyanov 			goto recv_sndq;
2058c0e88ff0SPavel Emelyanov 
2059c0e88ff0SPavel Emelyanov 		err = -EINVAL;
2060c0e88ff0SPavel Emelyanov 		if (tp->repair_queue == TCP_NO_QUEUE)
2061c0e88ff0SPavel Emelyanov 			goto out;
2062c0e88ff0SPavel Emelyanov 
2063c0e88ff0SPavel Emelyanov 		/* 'common' recv queue MSG_PEEK-ing */
2064c0e88ff0SPavel Emelyanov 	}
2065c0e88ff0SPavel Emelyanov 
20661da177e4SLinus Torvalds 	seq = &tp->copied_seq;
20671da177e4SLinus Torvalds 	if (flags & MSG_PEEK) {
20681da177e4SLinus Torvalds 		peek_seq = tp->copied_seq;
20691da177e4SLinus Torvalds 		seq = &peek_seq;
20701da177e4SLinus Torvalds 	}
20711da177e4SLinus Torvalds 
20721da177e4SLinus Torvalds 	target = sock_rcvlowat(sk, flags & MSG_WAITALL, len);
20731da177e4SLinus Torvalds 
20741da177e4SLinus Torvalds 	do {
20751da177e4SLinus Torvalds 		u32 offset;
20761da177e4SLinus Torvalds 
20771da177e4SLinus Torvalds 		/* Are we at urgent data? Stop if we have read anything or have SIGURG pending. */
20781da177e4SLinus Torvalds 		if (tp->urg_data && tp->urg_seq == *seq) {
20791da177e4SLinus Torvalds 			if (copied)
20801da177e4SLinus Torvalds 				break;
20811da177e4SLinus Torvalds 			if (signal_pending(current)) {
20821da177e4SLinus Torvalds 				copied = timeo ? sock_intr_errno(timeo) : -EAGAIN;
20831da177e4SLinus Torvalds 				break;
20841da177e4SLinus Torvalds 			}
20851da177e4SLinus Torvalds 		}
20861da177e4SLinus Torvalds 
20871da177e4SLinus Torvalds 		/* Next get a buffer. */
20881da177e4SLinus Torvalds 
2089dfbafc99SSabrina Dubroca 		last = skb_peek_tail(&sk->sk_receive_queue);
209091521944SDavid S. Miller 		skb_queue_walk(&sk->sk_receive_queue, skb) {
2091dfbafc99SSabrina Dubroca 			last = skb;
20921da177e4SLinus Torvalds 			/* Now that we have two receive queues this
20931da177e4SLinus Torvalds 			 * shouldn't happen.
20941da177e4SLinus Torvalds 			 */
2095d792c100SIlpo Järvinen 			if (WARN(before(*seq, TCP_SKB_CB(skb)->seq),
2096e56b8ce3SRandy Dunlap 				 "TCP recvmsg seq # bug: copied %X, seq %X, rcvnxt %X, fl %X\n",
20972af6fd8bSJoe Perches 				 *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt,
2098d792c100SIlpo Järvinen 				 flags))
20991da177e4SLinus Torvalds 				break;
2100d792c100SIlpo Järvinen 
21011da177e4SLinus Torvalds 			offset = *seq - TCP_SKB_CB(skb)->seq;
21029d691539SEric Dumazet 			if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) {
21039d691539SEric Dumazet 				pr_err_once("%s: found a SYN, please report !\n", __func__);
21041da177e4SLinus Torvalds 				offset--;
21059d691539SEric Dumazet 			}
21061da177e4SLinus Torvalds 			if (offset < skb->len)
21071da177e4SLinus Torvalds 				goto found_ok_skb;
2108e11ecddfSEric Dumazet 			if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
21091da177e4SLinus Torvalds 				goto found_fin_ok;
21102af6fd8bSJoe Perches 			WARN(!(flags & MSG_PEEK),
2111e56b8ce3SRandy Dunlap 			     "TCP recvmsg seq # bug 2: copied %X, seq %X, rcvnxt %X, fl %X\n",
21122af6fd8bSJoe Perches 			     *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt, flags);
211391521944SDavid S. Miller 		}
21141da177e4SLinus Torvalds 
21151da177e4SLinus Torvalds 		/* Well, if we have backlog, try to process it now yet. */
21161da177e4SLinus Torvalds 
21179ed498c6SEric Dumazet 		if (copied >= target && !READ_ONCE(sk->sk_backlog.tail))
21181da177e4SLinus Torvalds 			break;
21191da177e4SLinus Torvalds 
21201da177e4SLinus Torvalds 		if (copied) {
21211da177e4SLinus Torvalds 			if (sk->sk_err ||
21221da177e4SLinus Torvalds 			    sk->sk_state == TCP_CLOSE ||
21231da177e4SLinus Torvalds 			    (sk->sk_shutdown & RCV_SHUTDOWN) ||
21241da177e4SLinus Torvalds 			    !timeo ||
2125518a09efSDavid S. Miller 			    signal_pending(current))
21261da177e4SLinus Torvalds 				break;
21271da177e4SLinus Torvalds 		} else {
21281da177e4SLinus Torvalds 			if (sock_flag(sk, SOCK_DONE))
21291da177e4SLinus Torvalds 				break;
21301da177e4SLinus Torvalds 
21311da177e4SLinus Torvalds 			if (sk->sk_err) {
21321da177e4SLinus Torvalds 				copied = sock_error(sk);
21331da177e4SLinus Torvalds 				break;
21341da177e4SLinus Torvalds 			}
21351da177e4SLinus Torvalds 
21361da177e4SLinus Torvalds 			if (sk->sk_shutdown & RCV_SHUTDOWN)
21371da177e4SLinus Torvalds 				break;
21381da177e4SLinus Torvalds 
21391da177e4SLinus Torvalds 			if (sk->sk_state == TCP_CLOSE) {
21401da177e4SLinus Torvalds 				/* This occurs when user tries to read
21411da177e4SLinus Torvalds 				 * from never connected socket.
21421da177e4SLinus Torvalds 				 */
21431da177e4SLinus Torvalds 				copied = -ENOTCONN;
21441da177e4SLinus Torvalds 				break;
21451da177e4SLinus Torvalds 			}
21461da177e4SLinus Torvalds 
21471da177e4SLinus Torvalds 			if (!timeo) {
21481da177e4SLinus Torvalds 				copied = -EAGAIN;
21491da177e4SLinus Torvalds 				break;
21501da177e4SLinus Torvalds 			}
21511da177e4SLinus Torvalds 
21521da177e4SLinus Torvalds 			if (signal_pending(current)) {
21531da177e4SLinus Torvalds 				copied = sock_intr_errno(timeo);
21541da177e4SLinus Torvalds 				break;
21551da177e4SLinus Torvalds 			}
21561da177e4SLinus Torvalds 		}
21571da177e4SLinus Torvalds 
21580e4b4992SChris Leech 		tcp_cleanup_rbuf(sk, copied);
21591da177e4SLinus Torvalds 
21601da177e4SLinus Torvalds 		if (copied >= target) {
21611da177e4SLinus Torvalds 			/* Do not sleep, just process backlog. */
21621da177e4SLinus Torvalds 			release_sock(sk);
21631da177e4SLinus Torvalds 			lock_sock(sk);
2164dfbafc99SSabrina Dubroca 		} else {
2165dfbafc99SSabrina Dubroca 			sk_wait_data(sk, &timeo, last);
2166dfbafc99SSabrina Dubroca 		}
21671da177e4SLinus Torvalds 
216877527313SIlpo Järvinen 		if ((flags & MSG_PEEK) &&
216977527313SIlpo Järvinen 		    (peek_seq - copied - urg_hole != tp->copied_seq)) {
2170e87cc472SJoe Perches 			net_dbg_ratelimited("TCP(%s:%d): Application bug, race in MSG_PEEK\n",
2171e87cc472SJoe Perches 					    current->comm,
2172e87cc472SJoe Perches 					    task_pid_nr(current));
21731da177e4SLinus Torvalds 			peek_seq = tp->copied_seq;
21741da177e4SLinus Torvalds 		}
21751da177e4SLinus Torvalds 		continue;
21761da177e4SLinus Torvalds 
21771da177e4SLinus Torvalds found_ok_skb:
21781da177e4SLinus Torvalds 		/* Ok so how much can we use? */
21791da177e4SLinus Torvalds 		used = skb->len - offset;
21801da177e4SLinus Torvalds 		if (len < used)
21811da177e4SLinus Torvalds 			used = len;
21821da177e4SLinus Torvalds 
21831da177e4SLinus Torvalds 		/* Do we have urgent data here? */
21841da177e4SLinus Torvalds 		if (tp->urg_data) {
21851da177e4SLinus Torvalds 			u32 urg_offset = tp->urg_seq - *seq;
21861da177e4SLinus Torvalds 			if (urg_offset < used) {
21871da177e4SLinus Torvalds 				if (!urg_offset) {
21881da177e4SLinus Torvalds 					if (!sock_flag(sk, SOCK_URGINLINE)) {
21897db48e98SEric Dumazet 						WRITE_ONCE(*seq, *seq + 1);
219077527313SIlpo Järvinen 						urg_hole++;
21911da177e4SLinus Torvalds 						offset++;
21921da177e4SLinus Torvalds 						used--;
21931da177e4SLinus Torvalds 						if (!used)
21941da177e4SLinus Torvalds 							goto skip_copy;
21951da177e4SLinus Torvalds 					}
21961da177e4SLinus Torvalds 				} else
21971da177e4SLinus Torvalds 					used = urg_offset;
21981da177e4SLinus Torvalds 			}
21991da177e4SLinus Torvalds 		}
22001da177e4SLinus Torvalds 
22011da177e4SLinus Torvalds 		if (!(flags & MSG_TRUNC)) {
220251f3d02bSDavid S. Miller 			err = skb_copy_datagram_msg(skb, offset, msg, used);
22031da177e4SLinus Torvalds 			if (err) {
22041da177e4SLinus Torvalds 				/* Exception. Bailout! */
22051da177e4SLinus Torvalds 				if (!copied)
22061da177e4SLinus Torvalds 					copied = -EFAULT;
22071da177e4SLinus Torvalds 				break;
22081da177e4SLinus Torvalds 			}
22091da177e4SLinus Torvalds 		}
22101da177e4SLinus Torvalds 
22117db48e98SEric Dumazet 		WRITE_ONCE(*seq, *seq + used);
22121da177e4SLinus Torvalds 		copied += used;
22131da177e4SLinus Torvalds 		len -= used;
22141da177e4SLinus Torvalds 
22151da177e4SLinus Torvalds 		tcp_rcv_space_adjust(sk);
22161da177e4SLinus Torvalds 
22171da177e4SLinus Torvalds skip_copy:
221831770e34SFlorian Westphal 		if (tp->urg_data && after(tp->copied_seq, tp->urg_seq)) {
22191da177e4SLinus Torvalds 			tp->urg_data = 0;
222031770e34SFlorian Westphal 			tcp_fast_path_check(sk);
222131770e34SFlorian Westphal 		}
22221da177e4SLinus Torvalds 
222398aaa913SMike Maloney 		if (TCP_SKB_CB(skb)->has_rxtstamp) {
222498aaa913SMike Maloney 			tcp_update_recv_tstamps(skb, &tss);
2225a5a7daa5SEric Dumazet 			cmsg_flags |= 2;
222698aaa913SMike Maloney 		}
2227cc4de047SKelly Littlepage 
2228cc4de047SKelly Littlepage 		if (used + offset < skb->len)
2229cc4de047SKelly Littlepage 			continue;
2230cc4de047SKelly Littlepage 
2231e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
22321da177e4SLinus Torvalds 			goto found_fin_ok;
22337bced397SDan Williams 		if (!(flags & MSG_PEEK))
22347bced397SDan Williams 			sk_eat_skb(sk, skb);
22351da177e4SLinus Torvalds 		continue;
22361da177e4SLinus Torvalds 
22371da177e4SLinus Torvalds found_fin_ok:
22381da177e4SLinus Torvalds 		/* Process the FIN. */
22397db48e98SEric Dumazet 		WRITE_ONCE(*seq, *seq + 1);
22407bced397SDan Williams 		if (!(flags & MSG_PEEK))
22417bced397SDan Williams 			sk_eat_skb(sk, skb);
22421da177e4SLinus Torvalds 		break;
22431da177e4SLinus Torvalds 	} while (len > 0);
22441da177e4SLinus Torvalds 
22451da177e4SLinus Torvalds 	/* According to UNIX98, msg_name/msg_namelen are ignored
22461da177e4SLinus Torvalds 	 * on connected socket. I was just happy when found this 8) --ANK
22471da177e4SLinus Torvalds 	 */
22481da177e4SLinus Torvalds 
22491da177e4SLinus Torvalds 	/* Clean up data we have read: This will do ACK frames. */
22500e4b4992SChris Leech 	tcp_cleanup_rbuf(sk, copied);
22511da177e4SLinus Torvalds 
22521da177e4SLinus Torvalds 	release_sock(sk);
2253b75eba76SSoheil Hassas Yeganeh 
2254a5a7daa5SEric Dumazet 	if (cmsg_flags) {
2255a5a7daa5SEric Dumazet 		if (cmsg_flags & 2)
2256b75eba76SSoheil Hassas Yeganeh 			tcp_recv_timestamp(msg, sk, &tss);
2257a5a7daa5SEric Dumazet 		if (cmsg_flags & 1) {
2258b75eba76SSoheil Hassas Yeganeh 			inq = tcp_inq_hint(sk);
2259b75eba76SSoheil Hassas Yeganeh 			put_cmsg(msg, SOL_TCP, TCP_CM_INQ, sizeof(inq), &inq);
2260b75eba76SSoheil Hassas Yeganeh 		}
2261b75eba76SSoheil Hassas Yeganeh 	}
2262b75eba76SSoheil Hassas Yeganeh 
22631da177e4SLinus Torvalds 	return copied;
22641da177e4SLinus Torvalds 
22651da177e4SLinus Torvalds out:
22661da177e4SLinus Torvalds 	release_sock(sk);
22671da177e4SLinus Torvalds 	return err;
22681da177e4SLinus Torvalds 
22691da177e4SLinus Torvalds recv_urg:
2270377f0a08SRami Rosen 	err = tcp_recv_urg(sk, msg, len, flags);
22711da177e4SLinus Torvalds 	goto out;
2272c0e88ff0SPavel Emelyanov 
2273c0e88ff0SPavel Emelyanov recv_sndq:
2274c0e88ff0SPavel Emelyanov 	err = tcp_peek_sndq(sk, msg, len);
2275c0e88ff0SPavel Emelyanov 	goto out;
22761da177e4SLinus Torvalds }
22774bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_recvmsg);
22781da177e4SLinus Torvalds 
2279490d5046SIlpo Järvinen void tcp_set_state(struct sock *sk, int state)
2280490d5046SIlpo Järvinen {
2281490d5046SIlpo Järvinen 	int oldstate = sk->sk_state;
2282490d5046SIlpo Järvinen 
2283d4487491SLawrence Brakmo 	/* We defined a new enum for TCP states that are exported in BPF
2284d4487491SLawrence Brakmo 	 * so as not force the internal TCP states to be frozen. The
2285d4487491SLawrence Brakmo 	 * following checks will detect if an internal state value ever
2286d4487491SLawrence Brakmo 	 * differs from the BPF value. If this ever happens, then we will
2287d4487491SLawrence Brakmo 	 * need to remap the internal value to the BPF value before calling
2288d4487491SLawrence Brakmo 	 * tcp_call_bpf_2arg.
2289d4487491SLawrence Brakmo 	 */
2290d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_ESTABLISHED != (int)TCP_ESTABLISHED);
2291d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_SYN_SENT != (int)TCP_SYN_SENT);
2292d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_SYN_RECV != (int)TCP_SYN_RECV);
2293d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT1 != (int)TCP_FIN_WAIT1);
2294d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT2 != (int)TCP_FIN_WAIT2);
2295d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_TIME_WAIT != (int)TCP_TIME_WAIT);
2296d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSE != (int)TCP_CLOSE);
2297d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSE_WAIT != (int)TCP_CLOSE_WAIT);
2298d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_LAST_ACK != (int)TCP_LAST_ACK);
2299d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_LISTEN != (int)TCP_LISTEN);
2300d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_CLOSING != (int)TCP_CLOSING);
2301d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_NEW_SYN_RECV != (int)TCP_NEW_SYN_RECV);
2302d4487491SLawrence Brakmo 	BUILD_BUG_ON((int)BPF_TCP_MAX_STATES != (int)TCP_MAX_STATES);
2303d4487491SLawrence Brakmo 
2304d4487491SLawrence Brakmo 	if (BPF_SOCK_OPS_TEST_FLAG(tcp_sk(sk), BPF_SOCK_OPS_STATE_CB_FLAG))
2305d4487491SLawrence Brakmo 		tcp_call_bpf_2arg(sk, BPF_SOCK_OPS_STATE_CB, oldstate, state);
2306e8fce239SSong Liu 
2307490d5046SIlpo Järvinen 	switch (state) {
2308490d5046SIlpo Järvinen 	case TCP_ESTABLISHED:
2309490d5046SIlpo Järvinen 		if (oldstate != TCP_ESTABLISHED)
231081cc8a75SPavel Emelyanov 			TCP_INC_STATS(sock_net(sk), TCP_MIB_CURRESTAB);
2311490d5046SIlpo Järvinen 		break;
2312490d5046SIlpo Järvinen 
2313490d5046SIlpo Järvinen 	case TCP_CLOSE:
2314490d5046SIlpo Järvinen 		if (oldstate == TCP_CLOSE_WAIT || oldstate == TCP_ESTABLISHED)
231581cc8a75SPavel Emelyanov 			TCP_INC_STATS(sock_net(sk), TCP_MIB_ESTABRESETS);
2316490d5046SIlpo Järvinen 
2317490d5046SIlpo Järvinen 		sk->sk_prot->unhash(sk);
2318490d5046SIlpo Järvinen 		if (inet_csk(sk)->icsk_bind_hash &&
2319490d5046SIlpo Järvinen 		    !(sk->sk_userlocks & SOCK_BINDPORT_LOCK))
2320ab1e0a13SArnaldo Carvalho de Melo 			inet_put_port(sk);
2321a8eceea8SJoe Perches 		fallthrough;
2322490d5046SIlpo Järvinen 	default:
2323490d5046SIlpo Järvinen 		if (oldstate == TCP_ESTABLISHED)
232474688e48SPavel Emelyanov 			TCP_DEC_STATS(sock_net(sk), TCP_MIB_CURRESTAB);
2325490d5046SIlpo Järvinen 	}
2326490d5046SIlpo Järvinen 
2327490d5046SIlpo Järvinen 	/* Change state AFTER socket is unhashed to avoid closed
2328490d5046SIlpo Järvinen 	 * socket sitting in hash tables.
2329490d5046SIlpo Järvinen 	 */
2330563e0bb0SYafang Shao 	inet_sk_state_store(sk, state);
2331490d5046SIlpo Järvinen }
2332490d5046SIlpo Järvinen EXPORT_SYMBOL_GPL(tcp_set_state);
2333490d5046SIlpo Järvinen 
23341da177e4SLinus Torvalds /*
23351da177e4SLinus Torvalds  *	State processing on a close. This implements the state shift for
23361da177e4SLinus Torvalds  *	sending our FIN frame. Note that we only send a FIN for some
23371da177e4SLinus Torvalds  *	states. A shutdown() may have already sent the FIN, or we may be
23381da177e4SLinus Torvalds  *	closed.
23391da177e4SLinus Torvalds  */
23401da177e4SLinus Torvalds 
23419b5b5cffSArjan van de Ven static const unsigned char new_state[16] = {
23421da177e4SLinus Torvalds   /* current state:        new state:      action:	*/
23430980c1e3SEric Dumazet   [0 /* (Invalid) */]	= TCP_CLOSE,
23440980c1e3SEric Dumazet   [TCP_ESTABLISHED]	= TCP_FIN_WAIT1 | TCP_ACTION_FIN,
23450980c1e3SEric Dumazet   [TCP_SYN_SENT]	= TCP_CLOSE,
23460980c1e3SEric Dumazet   [TCP_SYN_RECV]	= TCP_FIN_WAIT1 | TCP_ACTION_FIN,
23470980c1e3SEric Dumazet   [TCP_FIN_WAIT1]	= TCP_FIN_WAIT1,
23480980c1e3SEric Dumazet   [TCP_FIN_WAIT2]	= TCP_FIN_WAIT2,
23490980c1e3SEric Dumazet   [TCP_TIME_WAIT]	= TCP_CLOSE,
23500980c1e3SEric Dumazet   [TCP_CLOSE]		= TCP_CLOSE,
23510980c1e3SEric Dumazet   [TCP_CLOSE_WAIT]	= TCP_LAST_ACK  | TCP_ACTION_FIN,
23520980c1e3SEric Dumazet   [TCP_LAST_ACK]	= TCP_LAST_ACK,
23530980c1e3SEric Dumazet   [TCP_LISTEN]		= TCP_CLOSE,
23540980c1e3SEric Dumazet   [TCP_CLOSING]		= TCP_CLOSING,
23550980c1e3SEric Dumazet   [TCP_NEW_SYN_RECV]	= TCP_CLOSE,	/* should not happen ! */
23561da177e4SLinus Torvalds };
23571da177e4SLinus Torvalds 
23581da177e4SLinus Torvalds static int tcp_close_state(struct sock *sk)
23591da177e4SLinus Torvalds {
23601da177e4SLinus Torvalds 	int next = (int)new_state[sk->sk_state];
23611da177e4SLinus Torvalds 	int ns = next & TCP_STATE_MASK;
23621da177e4SLinus Torvalds 
23631da177e4SLinus Torvalds 	tcp_set_state(sk, ns);
23641da177e4SLinus Torvalds 
23651da177e4SLinus Torvalds 	return next & TCP_ACTION_FIN;
23661da177e4SLinus Torvalds }
23671da177e4SLinus Torvalds 
23681da177e4SLinus Torvalds /*
23691da177e4SLinus Torvalds  *	Shutdown the sending side of a connection. Much like close except
23701f29b058SSatoru SATOH  *	that we don't receive shut down or sock_set_flag(sk, SOCK_DEAD).
23711da177e4SLinus Torvalds  */
23721da177e4SLinus Torvalds 
23731da177e4SLinus Torvalds void tcp_shutdown(struct sock *sk, int how)
23741da177e4SLinus Torvalds {
23751da177e4SLinus Torvalds 	/*	We need to grab some memory, and put together a FIN,
23761da177e4SLinus Torvalds 	 *	and then put it into the queue to be sent.
23771da177e4SLinus Torvalds 	 *		Tim MacKenzie(tym@dibbler.cs.monash.edu.au) 4 Dec '92.
23781da177e4SLinus Torvalds 	 */
23791da177e4SLinus Torvalds 	if (!(how & SEND_SHUTDOWN))
23801da177e4SLinus Torvalds 		return;
23811da177e4SLinus Torvalds 
23821da177e4SLinus Torvalds 	/* If we've already sent a FIN, or it's a closed state, skip this. */
23831da177e4SLinus Torvalds 	if ((1 << sk->sk_state) &
23841da177e4SLinus Torvalds 	    (TCPF_ESTABLISHED | TCPF_SYN_SENT |
23851da177e4SLinus Torvalds 	     TCPF_SYN_RECV | TCPF_CLOSE_WAIT)) {
23861da177e4SLinus Torvalds 		/* Clear out any half completed packets.  FIN if needed. */
23871da177e4SLinus Torvalds 		if (tcp_close_state(sk))
23881da177e4SLinus Torvalds 			tcp_send_fin(sk);
23891da177e4SLinus Torvalds 	}
23901da177e4SLinus Torvalds }
23914bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_shutdown);
23921da177e4SLinus Torvalds 
2393efcdbf24SArun Sharma bool tcp_check_oom(struct sock *sk, int shift)
2394efcdbf24SArun Sharma {
2395efcdbf24SArun Sharma 	bool too_many_orphans, out_of_socket_memory;
2396efcdbf24SArun Sharma 
2397efcdbf24SArun Sharma 	too_many_orphans = tcp_too_many_orphans(sk, shift);
2398efcdbf24SArun Sharma 	out_of_socket_memory = tcp_out_of_memory(sk);
2399efcdbf24SArun Sharma 
2400e87cc472SJoe Perches 	if (too_many_orphans)
2401e87cc472SJoe Perches 		net_info_ratelimited("too many orphaned sockets\n");
2402e87cc472SJoe Perches 	if (out_of_socket_memory)
2403e87cc472SJoe Perches 		net_info_ratelimited("out of memory -- consider tuning tcp_mem\n");
2404efcdbf24SArun Sharma 	return too_many_orphans || out_of_socket_memory;
2405efcdbf24SArun Sharma }
2406efcdbf24SArun Sharma 
24071da177e4SLinus Torvalds void tcp_close(struct sock *sk, long timeout)
24081da177e4SLinus Torvalds {
24091da177e4SLinus Torvalds 	struct sk_buff *skb;
24101da177e4SLinus Torvalds 	int data_was_unread = 0;
241175c2d907SHerbert Xu 	int state;
24121da177e4SLinus Torvalds 
24131da177e4SLinus Torvalds 	lock_sock(sk);
24141da177e4SLinus Torvalds 	sk->sk_shutdown = SHUTDOWN_MASK;
24151da177e4SLinus Torvalds 
24161da177e4SLinus Torvalds 	if (sk->sk_state == TCP_LISTEN) {
24171da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
24181da177e4SLinus Torvalds 
24191da177e4SLinus Torvalds 		/* Special case. */
24200a5578cfSArnaldo Carvalho de Melo 		inet_csk_listen_stop(sk);
24211da177e4SLinus Torvalds 
24221da177e4SLinus Torvalds 		goto adjudge_to_death;
24231da177e4SLinus Torvalds 	}
24241da177e4SLinus Torvalds 
24251da177e4SLinus Torvalds 	/*  We need to flush the recv. buffs.  We do this only on the
24261da177e4SLinus Torvalds 	 *  descriptor close, not protocol-sourced closes, because the
24271da177e4SLinus Torvalds 	 *  reader process may not have drained the data yet!
24281da177e4SLinus Torvalds 	 */
24291da177e4SLinus Torvalds 	while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
2430e11ecddfSEric Dumazet 		u32 len = TCP_SKB_CB(skb)->end_seq - TCP_SKB_CB(skb)->seq;
2431e11ecddfSEric Dumazet 
2432e11ecddfSEric Dumazet 		if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)
2433e11ecddfSEric Dumazet 			len--;
24341da177e4SLinus Torvalds 		data_was_unread += len;
24351da177e4SLinus Torvalds 		__kfree_skb(skb);
24361da177e4SLinus Torvalds 	}
24371da177e4SLinus Torvalds 
24383ab224beSHideo Aoki 	sk_mem_reclaim(sk);
24391da177e4SLinus Torvalds 
2440565b7b2dSKonstantin Khorenko 	/* If socket has been already reset (e.g. in tcp_reset()) - kill it. */
2441565b7b2dSKonstantin Khorenko 	if (sk->sk_state == TCP_CLOSE)
2442565b7b2dSKonstantin Khorenko 		goto adjudge_to_death;
2443565b7b2dSKonstantin Khorenko 
244465bb723cSGerrit Renker 	/* As outlined in RFC 2525, section 2.17, we send a RST here because
244565bb723cSGerrit Renker 	 * data was lost. To witness the awful effects of the old behavior of
244665bb723cSGerrit Renker 	 * always doing a FIN, run an older 2.1.x kernel or 2.0.x, start a bulk
244765bb723cSGerrit Renker 	 * GET in an FTP client, suspend the process, wait for the client to
244865bb723cSGerrit Renker 	 * advertise a zero window, then kill -9 the FTP client, wheee...
244965bb723cSGerrit Renker 	 * Note: timeout is always zero in such a case.
24501da177e4SLinus Torvalds 	 */
2451ee995283SPavel Emelyanov 	if (unlikely(tcp_sk(sk)->repair)) {
2452ee995283SPavel Emelyanov 		sk->sk_prot->disconnect(sk, 0);
2453ee995283SPavel Emelyanov 	} else if (data_was_unread) {
24541da177e4SLinus Torvalds 		/* Unread data was tossed, zap the connection. */
24556aef70a8SEric Dumazet 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONCLOSE);
24561da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
2457aa133076SWu Fengguang 		tcp_send_active_reset(sk, sk->sk_allocation);
24581da177e4SLinus Torvalds 	} else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
24591da177e4SLinus Torvalds 		/* Check zero linger _after_ checking for unread data. */
24601da177e4SLinus Torvalds 		sk->sk_prot->disconnect(sk, 0);
24616aef70a8SEric Dumazet 		NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONDATA);
24621da177e4SLinus Torvalds 	} else if (tcp_close_state(sk)) {
24631da177e4SLinus Torvalds 		/* We FIN if the application ate all the data before
24641da177e4SLinus Torvalds 		 * zapping the connection.
24651da177e4SLinus Torvalds 		 */
24661da177e4SLinus Torvalds 
24671da177e4SLinus Torvalds 		/* RED-PEN. Formally speaking, we have broken TCP state
24681da177e4SLinus Torvalds 		 * machine. State transitions:
24691da177e4SLinus Torvalds 		 *
24701da177e4SLinus Torvalds 		 * TCP_ESTABLISHED -> TCP_FIN_WAIT1
24711da177e4SLinus Torvalds 		 * TCP_SYN_RECV	-> TCP_FIN_WAIT1 (forget it, it's impossible)
24721da177e4SLinus Torvalds 		 * TCP_CLOSE_WAIT -> TCP_LAST_ACK
24731da177e4SLinus Torvalds 		 *
24741da177e4SLinus Torvalds 		 * are legal only when FIN has been sent (i.e. in window),
24751da177e4SLinus Torvalds 		 * rather than queued out of window. Purists blame.
24761da177e4SLinus Torvalds 		 *
24771da177e4SLinus Torvalds 		 * F.e. "RFC state" is ESTABLISHED,
24781da177e4SLinus Torvalds 		 * if Linux state is FIN-WAIT-1, but FIN is still not sent.
24791da177e4SLinus Torvalds 		 *
24801da177e4SLinus Torvalds 		 * The visible declinations are that sometimes
24811da177e4SLinus Torvalds 		 * we enter time-wait state, when it is not required really
24821da177e4SLinus Torvalds 		 * (harmless), do not send active resets, when they are
24831da177e4SLinus Torvalds 		 * required by specs (TCP_ESTABLISHED, TCP_CLOSE_WAIT, when
24841da177e4SLinus Torvalds 		 * they look as CLOSING or LAST_ACK for Linux)
24851da177e4SLinus Torvalds 		 * Probably, I missed some more holelets.
24861da177e4SLinus Torvalds 		 * 						--ANK
24878336886fSJerry Chu 		 * XXX (TFO) - To start off we don't support SYN+ACK+FIN
24888336886fSJerry Chu 		 * in a single packet! (May consider it later but will
24898336886fSJerry Chu 		 * probably need API support or TCP_CORK SYN-ACK until
24908336886fSJerry Chu 		 * data is written and socket is closed.)
24911da177e4SLinus Torvalds 		 */
24921da177e4SLinus Torvalds 		tcp_send_fin(sk);
24931da177e4SLinus Torvalds 	}
24941da177e4SLinus Torvalds 
24951da177e4SLinus Torvalds 	sk_stream_wait_close(sk, timeout);
24961da177e4SLinus Torvalds 
24971da177e4SLinus Torvalds adjudge_to_death:
249875c2d907SHerbert Xu 	state = sk->sk_state;
249975c2d907SHerbert Xu 	sock_hold(sk);
250075c2d907SHerbert Xu 	sock_orphan(sk);
250175c2d907SHerbert Xu 
25021da177e4SLinus Torvalds 	local_bh_disable();
25031da177e4SLinus Torvalds 	bh_lock_sock(sk);
25048873c064SEric Dumazet 	/* remove backlog if any, without releasing ownership. */
25058873c064SEric Dumazet 	__release_sock(sk);
25061da177e4SLinus Torvalds 
2507eb4dea58SHerbert Xu 	percpu_counter_inc(sk->sk_prot->orphan_count);
2508eb4dea58SHerbert Xu 
250975c2d907SHerbert Xu 	/* Have we already been destroyed by a softirq or backlog? */
251075c2d907SHerbert Xu 	if (state != TCP_CLOSE && sk->sk_state == TCP_CLOSE)
251175c2d907SHerbert Xu 		goto out;
25121da177e4SLinus Torvalds 
25131da177e4SLinus Torvalds 	/*	This is a (useful) BSD violating of the RFC. There is a
25141da177e4SLinus Torvalds 	 *	problem with TCP as specified in that the other end could
25151da177e4SLinus Torvalds 	 *	keep a socket open forever with no application left this end.
2516b10bd54cSJesper Juhl 	 *	We use a 1 minute timeout (about the same as BSD) then kill
25171da177e4SLinus Torvalds 	 *	our end. If they send after that then tough - BUT: long enough
25181da177e4SLinus Torvalds 	 *	that we won't make the old 4*rto = almost no time - whoops
25191da177e4SLinus Torvalds 	 *	reset mistake.
25201da177e4SLinus Torvalds 	 *
25211da177e4SLinus Torvalds 	 *	Nope, it was not mistake. It is really desired behaviour
25221da177e4SLinus Torvalds 	 *	f.e. on http servers, when such sockets are useless, but
25231da177e4SLinus Torvalds 	 *	consume significant resources. Let's do it with special
25241da177e4SLinus Torvalds 	 *	linger2	option.					--ANK
25251da177e4SLinus Torvalds 	 */
25261da177e4SLinus Torvalds 
25271da177e4SLinus Torvalds 	if (sk->sk_state == TCP_FIN_WAIT2) {
25281da177e4SLinus Torvalds 		struct tcp_sock *tp = tcp_sk(sk);
25291da177e4SLinus Torvalds 		if (tp->linger2 < 0) {
25301da177e4SLinus Torvalds 			tcp_set_state(sk, TCP_CLOSE);
25311da177e4SLinus Torvalds 			tcp_send_active_reset(sk, GFP_ATOMIC);
253202a1d6e7SEric Dumazet 			__NET_INC_STATS(sock_net(sk),
2533de0744afSPavel Emelyanov 					LINUX_MIB_TCPABORTONLINGER);
25341da177e4SLinus Torvalds 		} else {
2535463c84b9SArnaldo Carvalho de Melo 			const int tmo = tcp_fin_time(sk);
25361da177e4SLinus Torvalds 
25371da177e4SLinus Torvalds 			if (tmo > TCP_TIMEWAIT_LEN) {
253852499afeSDavid S. Miller 				inet_csk_reset_keepalive_timer(sk,
253952499afeSDavid S. Miller 						tmo - TCP_TIMEWAIT_LEN);
25401da177e4SLinus Torvalds 			} else {
25411da177e4SLinus Torvalds 				tcp_time_wait(sk, TCP_FIN_WAIT2, tmo);
25421da177e4SLinus Torvalds 				goto out;
25431da177e4SLinus Torvalds 			}
25441da177e4SLinus Torvalds 		}
25451da177e4SLinus Torvalds 	}
25461da177e4SLinus Torvalds 	if (sk->sk_state != TCP_CLOSE) {
25473ab224beSHideo Aoki 		sk_mem_reclaim(sk);
2548efcdbf24SArun Sharma 		if (tcp_check_oom(sk, 0)) {
25491da177e4SLinus Torvalds 			tcp_set_state(sk, TCP_CLOSE);
25501da177e4SLinus Torvalds 			tcp_send_active_reset(sk, GFP_ATOMIC);
255102a1d6e7SEric Dumazet 			__NET_INC_STATS(sock_net(sk),
2552de0744afSPavel Emelyanov 					LINUX_MIB_TCPABORTONMEMORY);
25534ee806d5SDan Streetman 		} else if (!check_net(sock_net(sk))) {
25544ee806d5SDan Streetman 			/* Not possible to send reset; just close */
25554ee806d5SDan Streetman 			tcp_set_state(sk, TCP_CLOSE);
25561da177e4SLinus Torvalds 		}
25571da177e4SLinus Torvalds 	}
25581da177e4SLinus Torvalds 
25598336886fSJerry Chu 	if (sk->sk_state == TCP_CLOSE) {
2560d983ea6fSEric Dumazet 		struct request_sock *req;
2561d983ea6fSEric Dumazet 
2562d983ea6fSEric Dumazet 		req = rcu_dereference_protected(tcp_sk(sk)->fastopen_rsk,
2563d983ea6fSEric Dumazet 						lockdep_sock_is_held(sk));
25648336886fSJerry Chu 		/* We could get here with a non-NULL req if the socket is
25658336886fSJerry Chu 		 * aborted (e.g., closed with unread data) before 3WHS
25668336886fSJerry Chu 		 * finishes.
25678336886fSJerry Chu 		 */
256800db4124SIan Morris 		if (req)
25698336886fSJerry Chu 			reqsk_fastopen_remove(sk, req, false);
25700a5578cfSArnaldo Carvalho de Melo 		inet_csk_destroy_sock(sk);
25718336886fSJerry Chu 	}
25721da177e4SLinus Torvalds 	/* Otherwise, socket is reprieved until protocol close. */
25731da177e4SLinus Torvalds 
25741da177e4SLinus Torvalds out:
25751da177e4SLinus Torvalds 	bh_unlock_sock(sk);
25761da177e4SLinus Torvalds 	local_bh_enable();
25778873c064SEric Dumazet 	release_sock(sk);
25781da177e4SLinus Torvalds 	sock_put(sk);
25791da177e4SLinus Torvalds }
25804bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_close);
25811da177e4SLinus Torvalds 
25821da177e4SLinus Torvalds /* These states need RST on ABORT according to RFC793 */
25831da177e4SLinus Torvalds 
2584a2a385d6SEric Dumazet static inline bool tcp_need_reset(int state)
25851da177e4SLinus Torvalds {
25861da177e4SLinus Torvalds 	return (1 << state) &
25871da177e4SLinus Torvalds 	       (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT | TCPF_FIN_WAIT1 |
25881da177e4SLinus Torvalds 		TCPF_FIN_WAIT2 | TCPF_SYN_RECV);
25891da177e4SLinus Torvalds }
25901da177e4SLinus Torvalds 
259175c119afSEric Dumazet static void tcp_rtx_queue_purge(struct sock *sk)
259275c119afSEric Dumazet {
259375c119afSEric Dumazet 	struct rb_node *p = rb_first(&sk->tcp_rtx_queue);
259475c119afSEric Dumazet 
25952bec445fSEric Dumazet 	tcp_sk(sk)->highest_sack = NULL;
259675c119afSEric Dumazet 	while (p) {
259775c119afSEric Dumazet 		struct sk_buff *skb = rb_to_skb(p);
259875c119afSEric Dumazet 
259975c119afSEric Dumazet 		p = rb_next(p);
260075c119afSEric Dumazet 		/* Since we are deleting whole queue, no need to
260175c119afSEric Dumazet 		 * list_del(&skb->tcp_tsorted_anchor)
260275c119afSEric Dumazet 		 */
260375c119afSEric Dumazet 		tcp_rtx_queue_unlink(skb, sk);
260475c119afSEric Dumazet 		sk_wmem_free_skb(sk, skb);
260575c119afSEric Dumazet 	}
260675c119afSEric Dumazet }
260775c119afSEric Dumazet 
2608ac3f09baSEric Dumazet void tcp_write_queue_purge(struct sock *sk)
2609ac3f09baSEric Dumazet {
2610ac3f09baSEric Dumazet 	struct sk_buff *skb;
2611ac3f09baSEric Dumazet 
2612ac3f09baSEric Dumazet 	tcp_chrono_stop(sk, TCP_CHRONO_BUSY);
2613ac3f09baSEric Dumazet 	while ((skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) {
2614ac3f09baSEric Dumazet 		tcp_skb_tsorted_anchor_cleanup(skb);
2615ac3f09baSEric Dumazet 		sk_wmem_free_skb(sk, skb);
2616ac3f09baSEric Dumazet 	}
261775c119afSEric Dumazet 	tcp_rtx_queue_purge(sk);
2618472c2e07SEric Dumazet 	skb = sk->sk_tx_skb_cache;
2619472c2e07SEric Dumazet 	if (skb) {
2620472c2e07SEric Dumazet 		__kfree_skb(skb);
2621472c2e07SEric Dumazet 		sk->sk_tx_skb_cache = NULL;
2622472c2e07SEric Dumazet 	}
2623ac3f09baSEric Dumazet 	INIT_LIST_HEAD(&tcp_sk(sk)->tsorted_sent_queue);
2624ac3f09baSEric Dumazet 	sk_mem_reclaim(sk);
2625ac3f09baSEric Dumazet 	tcp_clear_all_retrans_hints(tcp_sk(sk));
2626bffd168cSSoheil Hassas Yeganeh 	tcp_sk(sk)->packets_out = 0;
262704c03114SEric Dumazet 	inet_csk(sk)->icsk_backoff = 0;
2628ac3f09baSEric Dumazet }
2629ac3f09baSEric Dumazet 
26301da177e4SLinus Torvalds int tcp_disconnect(struct sock *sk, int flags)
26311da177e4SLinus Torvalds {
26321da177e4SLinus Torvalds 	struct inet_sock *inet = inet_sk(sk);
2633463c84b9SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
26341da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
26351da177e4SLinus Torvalds 	int old_state = sk->sk_state;
26360f317464SEric Dumazet 	u32 seq;
26371da177e4SLinus Torvalds 
26381da177e4SLinus Torvalds 	if (old_state != TCP_CLOSE)
26391da177e4SLinus Torvalds 		tcp_set_state(sk, TCP_CLOSE);
26401da177e4SLinus Torvalds 
26411da177e4SLinus Torvalds 	/* ABORT function of RFC793 */
26421da177e4SLinus Torvalds 	if (old_state == TCP_LISTEN) {
26430a5578cfSArnaldo Carvalho de Melo 		inet_csk_listen_stop(sk);
2644ee995283SPavel Emelyanov 	} else if (unlikely(tp->repair)) {
2645ee995283SPavel Emelyanov 		sk->sk_err = ECONNABORTED;
26461da177e4SLinus Torvalds 	} else if (tcp_need_reset(old_state) ||
26471da177e4SLinus Torvalds 		   (tp->snd_nxt != tp->write_seq &&
26481da177e4SLinus Torvalds 		    (1 << old_state) & (TCPF_CLOSING | TCPF_LAST_ACK))) {
2649caa20d9aSStephen Hemminger 		/* The last check adjusts for discrepancy of Linux wrt. RFC
26501da177e4SLinus Torvalds 		 * states
26511da177e4SLinus Torvalds 		 */
26521da177e4SLinus Torvalds 		tcp_send_active_reset(sk, gfp_any());
26531da177e4SLinus Torvalds 		sk->sk_err = ECONNRESET;
26541da177e4SLinus Torvalds 	} else if (old_state == TCP_SYN_SENT)
26551da177e4SLinus Torvalds 		sk->sk_err = ECONNRESET;
26561da177e4SLinus Torvalds 
26571da177e4SLinus Torvalds 	tcp_clear_xmit_timers(sk);
26581da177e4SLinus Torvalds 	__skb_queue_purge(&sk->sk_receive_queue);
26598b27dae5SEric Dumazet 	if (sk->sk_rx_skb_cache) {
26608b27dae5SEric Dumazet 		__kfree_skb(sk->sk_rx_skb_cache);
26618b27dae5SEric Dumazet 		sk->sk_rx_skb_cache = NULL;
26628b27dae5SEric Dumazet 	}
26637db48e98SEric Dumazet 	WRITE_ONCE(tp->copied_seq, tp->rcv_nxt);
26646508b678SEric Dumazet 	tp->urg_data = 0;
2665fe067e8aSDavid S. Miller 	tcp_write_queue_purge(sk);
2666cf1ef3f0SWei Wang 	tcp_fastopen_active_disable_ofo_check(sk);
26679f5afeaeSYaogong Wang 	skb_rbtree_purge(&tp->out_of_order_queue);
26681da177e4SLinus Torvalds 
2669c720c7e8SEric Dumazet 	inet->inet_dport = 0;
26701da177e4SLinus Torvalds 
26711da177e4SLinus Torvalds 	if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK))
26721da177e4SLinus Torvalds 		inet_reset_saddr(sk);
26731da177e4SLinus Torvalds 
26741da177e4SLinus Torvalds 	sk->sk_shutdown = 0;
26751da177e4SLinus Torvalds 	sock_reset_flag(sk, SOCK_DONE);
2676740b0f18SEric Dumazet 	tp->srtt_us = 0;
2677b9e2e689SEric Dumazet 	tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT);
26783f6c65d6SWei Wang 	tp->rcv_rtt_last_tsecr = 0;
26790f317464SEric Dumazet 
26800f317464SEric Dumazet 	seq = tp->write_seq + tp->max_window + 2;
26810f317464SEric Dumazet 	if (!seq)
26820f317464SEric Dumazet 		seq = 1;
26830f317464SEric Dumazet 	WRITE_ONCE(tp->write_seq, seq);
26840f317464SEric Dumazet 
2685463c84b9SArnaldo Carvalho de Melo 	icsk->icsk_backoff = 0;
26866687e988SArnaldo Carvalho de Melo 	icsk->icsk_probes_out = 0;
26876a408147SEric Dumazet 	icsk->icsk_rto = TCP_TIMEOUT_INIT;
2688ca584ba0SMartin KaFai Lau 	icsk->icsk_rto_min = TCP_RTO_MIN;
26892b8ee4f0SMartin KaFai Lau 	icsk->icsk_delack_max = TCP_DELACK_MAX;
26900b6a05c1SIlpo Järvinen 	tp->snd_ssthresh = TCP_INFINITE_SSTHRESH;
26913a9a57f6SEric Dumazet 	tp->snd_cwnd = TCP_INIT_CWND;
26921da177e4SLinus Torvalds 	tp->snd_cwnd_cnt = 0;
26931fdf475aSEric Dumazet 	tp->window_clamp = 0;
26942fbdd562SEric Dumazet 	tp->delivered = 0;
2695e21db6f6SYuchung Cheng 	tp->delivered_ce = 0;
2696ce69e563SChristoph Paasch 	if (icsk->icsk_ca_ops->release)
2697ce69e563SChristoph Paasch 		icsk->icsk_ca_ops->release(sk);
2698ce69e563SChristoph Paasch 	memset(icsk->icsk_ca_priv, 0, sizeof(icsk->icsk_ca_priv));
26996687e988SArnaldo Carvalho de Melo 	tcp_set_ca_state(sk, TCP_CA_Open);
2700d4761754SYousuk Seung 	tp->is_sack_reneg = 0;
27011da177e4SLinus Torvalds 	tcp_clear_retrans(tp);
2702c13c48c0SEric Dumazet 	tp->total_retrans = 0;
2703463c84b9SArnaldo Carvalho de Melo 	inet_csk_delack_init(sk);
2704499350a5SWei Wang 	/* Initialize rcv_mss to TCP_MIN_MSS to avoid division by 0
2705499350a5SWei Wang 	 * issue in __tcp_select_window()
2706499350a5SWei Wang 	 */
2707499350a5SWei Wang 	icsk->icsk_ack.rcv_mss = TCP_MIN_MSS;
2708b40b4f79SSrinivas Aji 	memset(&tp->rx_opt, 0, sizeof(tp->rx_opt));
27091da177e4SLinus Torvalds 	__sk_dst_reset(sk);
2710d747a7a5SWANG Cong 	dst_release(sk->sk_rx_dst);
2711d747a7a5SWANG Cong 	sk->sk_rx_dst = NULL;
271217c3060bSEric Dumazet 	tcp_saved_syn_free(tp);
27135d9f4262SEric Dumazet 	tp->compressed_ack = 0;
2714784f8344SEric Dumazet 	tp->segs_in = 0;
2715784f8344SEric Dumazet 	tp->segs_out = 0;
2716ba113c3aSWei Wang 	tp->bytes_sent = 0;
2717e858faf5SChristoph Paasch 	tp->bytes_acked = 0;
2718e858faf5SChristoph Paasch 	tp->bytes_received = 0;
2719fb31c9b9SWei Wang 	tp->bytes_retrans = 0;
2720db7ffee6SEric Dumazet 	tp->data_segs_in = 0;
2721db7ffee6SEric Dumazet 	tp->data_segs_out = 0;
27227788174eSYuchung Cheng 	tp->duplicate_sack[0].start_seq = 0;
27237788174eSYuchung Cheng 	tp->duplicate_sack[0].end_seq = 0;
27247e10b655SWei Wang 	tp->dsack_dups = 0;
27257ec65372SWei Wang 	tp->reord_seen = 0;
27265c701549SEric Dumazet 	tp->retrans_out = 0;
27275c701549SEric Dumazet 	tp->sacked_out = 0;
27285c701549SEric Dumazet 	tp->tlp_high_seq = 0;
27295c701549SEric Dumazet 	tp->last_oow_ack_time = 0;
27306cda8b74SEric Dumazet 	/* There's a bubble in the pipe until at least the first ACK. */
27316cda8b74SEric Dumazet 	tp->app_limited = ~0U;
2732792c4354SEric Dumazet 	tp->rack.mstamp = 0;
2733792c4354SEric Dumazet 	tp->rack.advanced = 0;
2734792c4354SEric Dumazet 	tp->rack.reo_wnd_steps = 1;
2735792c4354SEric Dumazet 	tp->rack.last_delivered = 0;
2736792c4354SEric Dumazet 	tp->rack.reo_wnd_persist = 0;
2737792c4354SEric Dumazet 	tp->rack.dsack_seen = 0;
27386bcdc40dSEric Dumazet 	tp->syn_data_acked = 0;
27396bcdc40dSEric Dumazet 	tp->rx_opt.saw_tstamp = 0;
27406bcdc40dSEric Dumazet 	tp->rx_opt.dsack = 0;
27416bcdc40dSEric Dumazet 	tp->rx_opt.num_sacks = 0;
2742f9af2dbbSThomas Higdon 	tp->rcv_ooopack = 0;
27436cda8b74SEric Dumazet 
27441da177e4SLinus Torvalds 
27457db92362SWei Wang 	/* Clean up fastopen related fields */
27467db92362SWei Wang 	tcp_free_fastopen_req(tp);
27477db92362SWei Wang 	inet->defer_connect = 0;
274848027478SJason Baron 	tp->fastopen_client_fail = 0;
27497db92362SWei Wang 
2750c720c7e8SEric Dumazet 	WARN_ON(inet->inet_num && !icsk->icsk_bind_hash);
27511da177e4SLinus Torvalds 
27529b42d55aSLi RongQing 	if (sk->sk_frag.page) {
27539b42d55aSLi RongQing 		put_page(sk->sk_frag.page);
27549b42d55aSLi RongQing 		sk->sk_frag.page = NULL;
27559b42d55aSLi RongQing 		sk->sk_frag.offset = 0;
27569b42d55aSLi RongQing 	}
27579b42d55aSLi RongQing 
27581da177e4SLinus Torvalds 	sk->sk_error_report(sk);
2759a01512b1SYueHaibing 	return 0;
27601da177e4SLinus Torvalds }
27614bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_disconnect);
27621da177e4SLinus Torvalds 
2763a2a385d6SEric Dumazet static inline bool tcp_can_repair_sock(const struct sock *sk)
2764ee995283SPavel Emelyanov {
276552e804c6SEric W. Biederman 	return ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN) &&
2766319b0534SAndrey Vagin 		(sk->sk_state != TCP_LISTEN);
2767ee995283SPavel Emelyanov }
2768ee995283SPavel Emelyanov 
2769d38d2b00SChristoph Hellwig static int tcp_repair_set_window(struct tcp_sock *tp, sockptr_t optbuf, int len)
2770b1ed4c4fSAndrey Vagin {
2771b1ed4c4fSAndrey Vagin 	struct tcp_repair_window opt;
2772b1ed4c4fSAndrey Vagin 
2773b1ed4c4fSAndrey Vagin 	if (!tp->repair)
2774b1ed4c4fSAndrey Vagin 		return -EPERM;
2775b1ed4c4fSAndrey Vagin 
2776b1ed4c4fSAndrey Vagin 	if (len != sizeof(opt))
2777b1ed4c4fSAndrey Vagin 		return -EINVAL;
2778b1ed4c4fSAndrey Vagin 
2779d38d2b00SChristoph Hellwig 	if (copy_from_sockptr(&opt, optbuf, sizeof(opt)))
2780b1ed4c4fSAndrey Vagin 		return -EFAULT;
2781b1ed4c4fSAndrey Vagin 
2782b1ed4c4fSAndrey Vagin 	if (opt.max_window < opt.snd_wnd)
2783b1ed4c4fSAndrey Vagin 		return -EINVAL;
2784b1ed4c4fSAndrey Vagin 
2785b1ed4c4fSAndrey Vagin 	if (after(opt.snd_wl1, tp->rcv_nxt + opt.rcv_wnd))
2786b1ed4c4fSAndrey Vagin 		return -EINVAL;
2787b1ed4c4fSAndrey Vagin 
2788b1ed4c4fSAndrey Vagin 	if (after(opt.rcv_wup, tp->rcv_nxt))
2789b1ed4c4fSAndrey Vagin 		return -EINVAL;
2790b1ed4c4fSAndrey Vagin 
2791b1ed4c4fSAndrey Vagin 	tp->snd_wl1	= opt.snd_wl1;
2792b1ed4c4fSAndrey Vagin 	tp->snd_wnd	= opt.snd_wnd;
2793b1ed4c4fSAndrey Vagin 	tp->max_window	= opt.max_window;
2794b1ed4c4fSAndrey Vagin 
2795b1ed4c4fSAndrey Vagin 	tp->rcv_wnd	= opt.rcv_wnd;
2796b1ed4c4fSAndrey Vagin 	tp->rcv_wup	= opt.rcv_wup;
2797b1ed4c4fSAndrey Vagin 
2798b1ed4c4fSAndrey Vagin 	return 0;
2799b1ed4c4fSAndrey Vagin }
2800b1ed4c4fSAndrey Vagin 
2801d38d2b00SChristoph Hellwig static int tcp_repair_options_est(struct sock *sk, sockptr_t optbuf,
2802d38d2b00SChristoph Hellwig 		unsigned int len)
2803b139ba4eSPavel Emelyanov {
280415e56515SDouglas Caetano dos Santos 	struct tcp_sock *tp = tcp_sk(sk);
2805de248a75SPavel Emelyanov 	struct tcp_repair_opt opt;
2806d3c48151SChristoph Hellwig 	size_t offset = 0;
2807b139ba4eSPavel Emelyanov 
2808de248a75SPavel Emelyanov 	while (len >= sizeof(opt)) {
2809d3c48151SChristoph Hellwig 		if (copy_from_sockptr_offset(&opt, optbuf, offset, sizeof(opt)))
2810b139ba4eSPavel Emelyanov 			return -EFAULT;
2811b139ba4eSPavel Emelyanov 
2812d3c48151SChristoph Hellwig 		offset += sizeof(opt);
2813de248a75SPavel Emelyanov 		len -= sizeof(opt);
2814b139ba4eSPavel Emelyanov 
2815de248a75SPavel Emelyanov 		switch (opt.opt_code) {
2816de248a75SPavel Emelyanov 		case TCPOPT_MSS:
2817de248a75SPavel Emelyanov 			tp->rx_opt.mss_clamp = opt.opt_val;
281815e56515SDouglas Caetano dos Santos 			tcp_mtup_init(sk);
2819b139ba4eSPavel Emelyanov 			break;
2820de248a75SPavel Emelyanov 		case TCPOPT_WINDOW:
2821bc26ccd8SAndrey Vagin 			{
2822bc26ccd8SAndrey Vagin 				u16 snd_wscale = opt.opt_val & 0xFFFF;
2823bc26ccd8SAndrey Vagin 				u16 rcv_wscale = opt.opt_val >> 16;
2824bc26ccd8SAndrey Vagin 
2825589c49cbSGao Feng 				if (snd_wscale > TCP_MAX_WSCALE || rcv_wscale > TCP_MAX_WSCALE)
2826b139ba4eSPavel Emelyanov 					return -EFBIG;
2827b139ba4eSPavel Emelyanov 
2828bc26ccd8SAndrey Vagin 				tp->rx_opt.snd_wscale = snd_wscale;
2829bc26ccd8SAndrey Vagin 				tp->rx_opt.rcv_wscale = rcv_wscale;
2830bc26ccd8SAndrey Vagin 				tp->rx_opt.wscale_ok = 1;
2831bc26ccd8SAndrey Vagin 			}
2832b139ba4eSPavel Emelyanov 			break;
2833b139ba4eSPavel Emelyanov 		case TCPOPT_SACK_PERM:
2834de248a75SPavel Emelyanov 			if (opt.opt_val != 0)
2835de248a75SPavel Emelyanov 				return -EINVAL;
2836de248a75SPavel Emelyanov 
2837b139ba4eSPavel Emelyanov 			tp->rx_opt.sack_ok |= TCP_SACK_SEEN;
2838b139ba4eSPavel Emelyanov 			break;
2839b139ba4eSPavel Emelyanov 		case TCPOPT_TIMESTAMP:
2840de248a75SPavel Emelyanov 			if (opt.opt_val != 0)
2841de248a75SPavel Emelyanov 				return -EINVAL;
2842de248a75SPavel Emelyanov 
2843b139ba4eSPavel Emelyanov 			tp->rx_opt.tstamp_ok = 1;
2844b139ba4eSPavel Emelyanov 			break;
2845b139ba4eSPavel Emelyanov 		}
2846b139ba4eSPavel Emelyanov 	}
2847b139ba4eSPavel Emelyanov 
2848b139ba4eSPavel Emelyanov 	return 0;
2849b139ba4eSPavel Emelyanov }
2850b139ba4eSPavel Emelyanov 
2851a842fe14SEric Dumazet DEFINE_STATIC_KEY_FALSE(tcp_tx_delay_enabled);
2852a842fe14SEric Dumazet EXPORT_SYMBOL(tcp_tx_delay_enabled);
2853a842fe14SEric Dumazet 
2854a842fe14SEric Dumazet static void tcp_enable_tx_delay(void)
2855a842fe14SEric Dumazet {
2856a842fe14SEric Dumazet 	if (!static_branch_unlikely(&tcp_tx_delay_enabled)) {
2857a842fe14SEric Dumazet 		static int __tcp_tx_delay_enabled = 0;
2858a842fe14SEric Dumazet 
2859a842fe14SEric Dumazet 		if (cmpxchg(&__tcp_tx_delay_enabled, 0, 1) == 0) {
2860a842fe14SEric Dumazet 			static_branch_enable(&tcp_tx_delay_enabled);
2861a842fe14SEric Dumazet 			pr_info("TCP_TX_DELAY enabled\n");
2862a842fe14SEric Dumazet 		}
2863a842fe14SEric Dumazet 	}
2864a842fe14SEric Dumazet }
2865a842fe14SEric Dumazet 
2866db10538aSChristoph Hellwig /* When set indicates to always queue non-full frames.  Later the user clears
2867db10538aSChristoph Hellwig  * this option and we transmit any pending partial frames in the queue.  This is
2868db10538aSChristoph Hellwig  * meant to be used alongside sendfile() to get properly filled frames when the
2869db10538aSChristoph Hellwig  * user (for example) must write out headers with a write() call first and then
2870db10538aSChristoph Hellwig  * use sendfile to send out the data parts.
2871db10538aSChristoph Hellwig  *
2872db10538aSChristoph Hellwig  * TCP_CORK can be set together with TCP_NODELAY and it is stronger than
2873db10538aSChristoph Hellwig  * TCP_NODELAY.
2874db10538aSChristoph Hellwig  */
2875db10538aSChristoph Hellwig static void __tcp_sock_set_cork(struct sock *sk, bool on)
2876db10538aSChristoph Hellwig {
2877db10538aSChristoph Hellwig 	struct tcp_sock *tp = tcp_sk(sk);
2878db10538aSChristoph Hellwig 
2879db10538aSChristoph Hellwig 	if (on) {
2880db10538aSChristoph Hellwig 		tp->nonagle |= TCP_NAGLE_CORK;
2881db10538aSChristoph Hellwig 	} else {
2882db10538aSChristoph Hellwig 		tp->nonagle &= ~TCP_NAGLE_CORK;
2883db10538aSChristoph Hellwig 		if (tp->nonagle & TCP_NAGLE_OFF)
2884db10538aSChristoph Hellwig 			tp->nonagle |= TCP_NAGLE_PUSH;
2885db10538aSChristoph Hellwig 		tcp_push_pending_frames(sk);
2886db10538aSChristoph Hellwig 	}
2887db10538aSChristoph Hellwig }
2888db10538aSChristoph Hellwig 
2889db10538aSChristoph Hellwig void tcp_sock_set_cork(struct sock *sk, bool on)
2890db10538aSChristoph Hellwig {
2891db10538aSChristoph Hellwig 	lock_sock(sk);
2892db10538aSChristoph Hellwig 	__tcp_sock_set_cork(sk, on);
2893db10538aSChristoph Hellwig 	release_sock(sk);
2894db10538aSChristoph Hellwig }
2895db10538aSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_cork);
2896db10538aSChristoph Hellwig 
289712abc5eeSChristoph Hellwig /* TCP_NODELAY is weaker than TCP_CORK, so that this option on corked socket is
289812abc5eeSChristoph Hellwig  * remembered, but it is not activated until cork is cleared.
289912abc5eeSChristoph Hellwig  *
290012abc5eeSChristoph Hellwig  * However, when TCP_NODELAY is set we make an explicit push, which overrides
290112abc5eeSChristoph Hellwig  * even TCP_CORK for currently queued segments.
290212abc5eeSChristoph Hellwig  */
290312abc5eeSChristoph Hellwig static void __tcp_sock_set_nodelay(struct sock *sk, bool on)
290412abc5eeSChristoph Hellwig {
290512abc5eeSChristoph Hellwig 	if (on) {
290612abc5eeSChristoph Hellwig 		tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF|TCP_NAGLE_PUSH;
290712abc5eeSChristoph Hellwig 		tcp_push_pending_frames(sk);
290812abc5eeSChristoph Hellwig 	} else {
290912abc5eeSChristoph Hellwig 		tcp_sk(sk)->nonagle &= ~TCP_NAGLE_OFF;
291012abc5eeSChristoph Hellwig 	}
291112abc5eeSChristoph Hellwig }
291212abc5eeSChristoph Hellwig 
291312abc5eeSChristoph Hellwig void tcp_sock_set_nodelay(struct sock *sk)
291412abc5eeSChristoph Hellwig {
291512abc5eeSChristoph Hellwig 	lock_sock(sk);
291612abc5eeSChristoph Hellwig 	__tcp_sock_set_nodelay(sk, true);
291712abc5eeSChristoph Hellwig 	release_sock(sk);
291812abc5eeSChristoph Hellwig }
291912abc5eeSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_nodelay);
292012abc5eeSChristoph Hellwig 
2921ddd061b8SChristoph Hellwig static void __tcp_sock_set_quickack(struct sock *sk, int val)
2922ddd061b8SChristoph Hellwig {
2923ddd061b8SChristoph Hellwig 	if (!val) {
2924ddd061b8SChristoph Hellwig 		inet_csk_enter_pingpong_mode(sk);
2925ddd061b8SChristoph Hellwig 		return;
2926ddd061b8SChristoph Hellwig 	}
2927ddd061b8SChristoph Hellwig 
2928ddd061b8SChristoph Hellwig 	inet_csk_exit_pingpong_mode(sk);
2929ddd061b8SChristoph Hellwig 	if ((1 << sk->sk_state) & (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT) &&
2930ddd061b8SChristoph Hellwig 	    inet_csk_ack_scheduled(sk)) {
2931ddd061b8SChristoph Hellwig 		inet_csk(sk)->icsk_ack.pending |= ICSK_ACK_PUSHED;
2932ddd061b8SChristoph Hellwig 		tcp_cleanup_rbuf(sk, 1);
2933ddd061b8SChristoph Hellwig 		if (!(val & 1))
2934ddd061b8SChristoph Hellwig 			inet_csk_enter_pingpong_mode(sk);
2935ddd061b8SChristoph Hellwig 	}
2936ddd061b8SChristoph Hellwig }
2937ddd061b8SChristoph Hellwig 
2938ddd061b8SChristoph Hellwig void tcp_sock_set_quickack(struct sock *sk, int val)
2939ddd061b8SChristoph Hellwig {
2940ddd061b8SChristoph Hellwig 	lock_sock(sk);
2941ddd061b8SChristoph Hellwig 	__tcp_sock_set_quickack(sk, val);
2942ddd061b8SChristoph Hellwig 	release_sock(sk);
2943ddd061b8SChristoph Hellwig }
2944ddd061b8SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_quickack);
2945ddd061b8SChristoph Hellwig 
2946557eadfcSChristoph Hellwig int tcp_sock_set_syncnt(struct sock *sk, int val)
2947557eadfcSChristoph Hellwig {
2948557eadfcSChristoph Hellwig 	if (val < 1 || val > MAX_TCP_SYNCNT)
2949557eadfcSChristoph Hellwig 		return -EINVAL;
2950557eadfcSChristoph Hellwig 
2951557eadfcSChristoph Hellwig 	lock_sock(sk);
2952557eadfcSChristoph Hellwig 	inet_csk(sk)->icsk_syn_retries = val;
2953557eadfcSChristoph Hellwig 	release_sock(sk);
2954557eadfcSChristoph Hellwig 	return 0;
2955557eadfcSChristoph Hellwig }
2956557eadfcSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_syncnt);
2957557eadfcSChristoph Hellwig 
2958c488aeadSChristoph Hellwig void tcp_sock_set_user_timeout(struct sock *sk, u32 val)
2959c488aeadSChristoph Hellwig {
2960c488aeadSChristoph Hellwig 	lock_sock(sk);
2961c488aeadSChristoph Hellwig 	inet_csk(sk)->icsk_user_timeout = val;
2962c488aeadSChristoph Hellwig 	release_sock(sk);
2963c488aeadSChristoph Hellwig }
2964c488aeadSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_user_timeout);
2965c488aeadSChristoph Hellwig 
2966aad4a0a9SDmitry Yakunin int tcp_sock_set_keepidle_locked(struct sock *sk, int val)
296771c48eb8SChristoph Hellwig {
296871c48eb8SChristoph Hellwig 	struct tcp_sock *tp = tcp_sk(sk);
296971c48eb8SChristoph Hellwig 
297071c48eb8SChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPIDLE)
297171c48eb8SChristoph Hellwig 		return -EINVAL;
297271c48eb8SChristoph Hellwig 
297371c48eb8SChristoph Hellwig 	tp->keepalive_time = val * HZ;
297471c48eb8SChristoph Hellwig 	if (sock_flag(sk, SOCK_KEEPOPEN) &&
297571c48eb8SChristoph Hellwig 	    !((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN))) {
297671c48eb8SChristoph Hellwig 		u32 elapsed = keepalive_time_elapsed(tp);
297771c48eb8SChristoph Hellwig 
297871c48eb8SChristoph Hellwig 		if (tp->keepalive_time > elapsed)
297971c48eb8SChristoph Hellwig 			elapsed = tp->keepalive_time - elapsed;
298071c48eb8SChristoph Hellwig 		else
298171c48eb8SChristoph Hellwig 			elapsed = 0;
298271c48eb8SChristoph Hellwig 		inet_csk_reset_keepalive_timer(sk, elapsed);
298371c48eb8SChristoph Hellwig 	}
298471c48eb8SChristoph Hellwig 
298571c48eb8SChristoph Hellwig 	return 0;
298671c48eb8SChristoph Hellwig }
298771c48eb8SChristoph Hellwig 
298871c48eb8SChristoph Hellwig int tcp_sock_set_keepidle(struct sock *sk, int val)
298971c48eb8SChristoph Hellwig {
299071c48eb8SChristoph Hellwig 	int err;
299171c48eb8SChristoph Hellwig 
299271c48eb8SChristoph Hellwig 	lock_sock(sk);
2993aad4a0a9SDmitry Yakunin 	err = tcp_sock_set_keepidle_locked(sk, val);
299471c48eb8SChristoph Hellwig 	release_sock(sk);
299571c48eb8SChristoph Hellwig 	return err;
299671c48eb8SChristoph Hellwig }
299771c48eb8SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepidle);
299871c48eb8SChristoph Hellwig 
2999d41ecaacSChristoph Hellwig int tcp_sock_set_keepintvl(struct sock *sk, int val)
3000d41ecaacSChristoph Hellwig {
3001d41ecaacSChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPINTVL)
3002d41ecaacSChristoph Hellwig 		return -EINVAL;
3003d41ecaacSChristoph Hellwig 
3004d41ecaacSChristoph Hellwig 	lock_sock(sk);
3005d41ecaacSChristoph Hellwig 	tcp_sk(sk)->keepalive_intvl = val * HZ;
3006d41ecaacSChristoph Hellwig 	release_sock(sk);
3007d41ecaacSChristoph Hellwig 	return 0;
3008d41ecaacSChristoph Hellwig }
3009d41ecaacSChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepintvl);
3010d41ecaacSChristoph Hellwig 
3011480aeb96SChristoph Hellwig int tcp_sock_set_keepcnt(struct sock *sk, int val)
3012480aeb96SChristoph Hellwig {
3013480aeb96SChristoph Hellwig 	if (val < 1 || val > MAX_TCP_KEEPCNT)
3014480aeb96SChristoph Hellwig 		return -EINVAL;
3015480aeb96SChristoph Hellwig 
3016480aeb96SChristoph Hellwig 	lock_sock(sk);
3017480aeb96SChristoph Hellwig 	tcp_sk(sk)->keepalive_probes = val;
3018480aeb96SChristoph Hellwig 	release_sock(sk);
3019480aeb96SChristoph Hellwig 	return 0;
3020480aeb96SChristoph Hellwig }
3021480aeb96SChristoph Hellwig EXPORT_SYMBOL(tcp_sock_set_keepcnt);
3022480aeb96SChristoph Hellwig 
30231da177e4SLinus Torvalds /*
30241da177e4SLinus Torvalds  *	Socket option code for TCP.
30251da177e4SLinus Torvalds  */
3026d38d2b00SChristoph Hellwig static int do_tcp_setsockopt(struct sock *sk, int level, int optname,
3027d38d2b00SChristoph Hellwig 		sockptr_t optval, unsigned int optlen)
30281da177e4SLinus Torvalds {
30291da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
3030463c84b9SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
30311e579caaSNikolay Borisov 	struct net *net = sock_net(sk);
30321da177e4SLinus Torvalds 	int val;
30331da177e4SLinus Torvalds 	int err = 0;
30341da177e4SLinus Torvalds 
3035e56fb50fSWilliam Allen Simpson 	/* These are data/string values, all the others are ints */
3036e56fb50fSWilliam Allen Simpson 	switch (optname) {
3037e56fb50fSWilliam Allen Simpson 	case TCP_CONGESTION: {
30385f8ef48dSStephen Hemminger 		char name[TCP_CA_NAME_MAX];
30395f8ef48dSStephen Hemminger 
30405f8ef48dSStephen Hemminger 		if (optlen < 1)
30415f8ef48dSStephen Hemminger 			return -EINVAL;
30425f8ef48dSStephen Hemminger 
3043d38d2b00SChristoph Hellwig 		val = strncpy_from_sockptr(name, optval,
30444fdb78d3SAndrew Morton 					min_t(long, TCP_CA_NAME_MAX-1, optlen));
30455f8ef48dSStephen Hemminger 		if (val < 0)
30465f8ef48dSStephen Hemminger 			return -EFAULT;
30475f8ef48dSStephen Hemminger 		name[val] = 0;
30485f8ef48dSStephen Hemminger 
30495f8ef48dSStephen Hemminger 		lock_sock(sk);
30508d650cdeSEric Dumazet 		err = tcp_set_congestion_control(sk, name, true, true,
30518d650cdeSEric Dumazet 						 ns_capable(sock_net(sk)->user_ns,
30528d650cdeSEric Dumazet 							    CAP_NET_ADMIN));
30535f8ef48dSStephen Hemminger 		release_sock(sk);
30545f8ef48dSStephen Hemminger 		return err;
30555f8ef48dSStephen Hemminger 	}
3056734942ccSDave Watson 	case TCP_ULP: {
3057734942ccSDave Watson 		char name[TCP_ULP_NAME_MAX];
3058734942ccSDave Watson 
3059734942ccSDave Watson 		if (optlen < 1)
3060734942ccSDave Watson 			return -EINVAL;
3061734942ccSDave Watson 
3062d38d2b00SChristoph Hellwig 		val = strncpy_from_sockptr(name, optval,
3063734942ccSDave Watson 					min_t(long, TCP_ULP_NAME_MAX - 1,
3064734942ccSDave Watson 					      optlen));
3065734942ccSDave Watson 		if (val < 0)
3066734942ccSDave Watson 			return -EFAULT;
3067734942ccSDave Watson 		name[val] = 0;
3068734942ccSDave Watson 
3069734942ccSDave Watson 		lock_sock(sk);
3070734942ccSDave Watson 		err = tcp_set_ulp(sk, name);
3071734942ccSDave Watson 		release_sock(sk);
3072734942ccSDave Watson 		return err;
3073734942ccSDave Watson 	}
30741fba70e5SYuchung Cheng 	case TCP_FASTOPEN_KEY: {
30750f1ce023SJason Baron 		__u8 key[TCP_FASTOPEN_KEY_BUF_LENGTH];
30760f1ce023SJason Baron 		__u8 *backup_key = NULL;
30771fba70e5SYuchung Cheng 
30780f1ce023SJason Baron 		/* Allow a backup key as well to facilitate key rotation
30790f1ce023SJason Baron 		 * First key is the active one.
30800f1ce023SJason Baron 		 */
30810f1ce023SJason Baron 		if (optlen != TCP_FASTOPEN_KEY_LENGTH &&
30820f1ce023SJason Baron 		    optlen != TCP_FASTOPEN_KEY_BUF_LENGTH)
30831fba70e5SYuchung Cheng 			return -EINVAL;
30841fba70e5SYuchung Cheng 
3085d38d2b00SChristoph Hellwig 		if (copy_from_sockptr(key, optval, optlen))
30861fba70e5SYuchung Cheng 			return -EFAULT;
30871fba70e5SYuchung Cheng 
30880f1ce023SJason Baron 		if (optlen == TCP_FASTOPEN_KEY_BUF_LENGTH)
30890f1ce023SJason Baron 			backup_key = key + TCP_FASTOPEN_KEY_LENGTH;
30900f1ce023SJason Baron 
3091438ac880SArd Biesheuvel 		return tcp_fastopen_reset_cipher(net, sk, key, backup_key);
30921fba70e5SYuchung Cheng 	}
3093e56fb50fSWilliam Allen Simpson 	default:
3094e56fb50fSWilliam Allen Simpson 		/* fallthru */
3095e56fb50fSWilliam Allen Simpson 		break;
3096ccbd6a5aSJoe Perches 	}
30975f8ef48dSStephen Hemminger 
30981da177e4SLinus Torvalds 	if (optlen < sizeof(int))
30991da177e4SLinus Torvalds 		return -EINVAL;
31001da177e4SLinus Torvalds 
3101d38d2b00SChristoph Hellwig 	if (copy_from_sockptr(&val, optval, sizeof(val)))
31021da177e4SLinus Torvalds 		return -EFAULT;
31031da177e4SLinus Torvalds 
31041da177e4SLinus Torvalds 	lock_sock(sk);
31051da177e4SLinus Torvalds 
31061da177e4SLinus Torvalds 	switch (optname) {
31071da177e4SLinus Torvalds 	case TCP_MAXSEG:
31081da177e4SLinus Torvalds 		/* Values greater than interface MTU won't take effect. However
31091da177e4SLinus Torvalds 		 * at the point when this call is done we typically don't yet
3110a777f715SRohit Chavan 		 * know which interface is going to be used
3111a777f715SRohit Chavan 		 */
3112cfc62d87SGao Feng 		if (val && (val < TCP_MIN_MSS || val > MAX_TCP_WINDOW)) {
31131da177e4SLinus Torvalds 			err = -EINVAL;
31141da177e4SLinus Torvalds 			break;
31151da177e4SLinus Torvalds 		}
31161da177e4SLinus Torvalds 		tp->rx_opt.user_mss = val;
31171da177e4SLinus Torvalds 		break;
31181da177e4SLinus Torvalds 
31191da177e4SLinus Torvalds 	case TCP_NODELAY:
312012abc5eeSChristoph Hellwig 		__tcp_sock_set_nodelay(sk, val);
31211da177e4SLinus Torvalds 		break;
31221da177e4SLinus Torvalds 
312336e31b0aSAndreas Petlund 	case TCP_THIN_LINEAR_TIMEOUTS:
312436e31b0aSAndreas Petlund 		if (val < 0 || val > 1)
312536e31b0aSAndreas Petlund 			err = -EINVAL;
312636e31b0aSAndreas Petlund 		else
312736e31b0aSAndreas Petlund 			tp->thin_lto = val;
312836e31b0aSAndreas Petlund 		break;
312936e31b0aSAndreas Petlund 
31307e380175SAndreas Petlund 	case TCP_THIN_DUPACK:
31317e380175SAndreas Petlund 		if (val < 0 || val > 1)
31327e380175SAndreas Petlund 			err = -EINVAL;
31337e380175SAndreas Petlund 		break;
31347e380175SAndreas Petlund 
3135ee995283SPavel Emelyanov 	case TCP_REPAIR:
3136ee995283SPavel Emelyanov 		if (!tcp_can_repair_sock(sk))
3137ee995283SPavel Emelyanov 			err = -EPERM;
313831048d7aSStefan Baranoff 		else if (val == TCP_REPAIR_ON) {
3139ee995283SPavel Emelyanov 			tp->repair = 1;
3140ee995283SPavel Emelyanov 			sk->sk_reuse = SK_FORCE_REUSE;
3141ee995283SPavel Emelyanov 			tp->repair_queue = TCP_NO_QUEUE;
314231048d7aSStefan Baranoff 		} else if (val == TCP_REPAIR_OFF) {
3143ee995283SPavel Emelyanov 			tp->repair = 0;
3144ee995283SPavel Emelyanov 			sk->sk_reuse = SK_NO_REUSE;
3145ee995283SPavel Emelyanov 			tcp_send_window_probe(sk);
314631048d7aSStefan Baranoff 		} else if (val == TCP_REPAIR_OFF_NO_WP) {
314731048d7aSStefan Baranoff 			tp->repair = 0;
314831048d7aSStefan Baranoff 			sk->sk_reuse = SK_NO_REUSE;
3149ee995283SPavel Emelyanov 		} else
3150ee995283SPavel Emelyanov 			err = -EINVAL;
3151ee995283SPavel Emelyanov 
3152ee995283SPavel Emelyanov 		break;
3153ee995283SPavel Emelyanov 
3154ee995283SPavel Emelyanov 	case TCP_REPAIR_QUEUE:
3155ee995283SPavel Emelyanov 		if (!tp->repair)
3156ee995283SPavel Emelyanov 			err = -EPERM;
3157bf2acc94SEric Dumazet 		else if ((unsigned int)val < TCP_QUEUES_NR)
3158ee995283SPavel Emelyanov 			tp->repair_queue = val;
3159ee995283SPavel Emelyanov 		else
3160ee995283SPavel Emelyanov 			err = -EINVAL;
3161ee995283SPavel Emelyanov 		break;
3162ee995283SPavel Emelyanov 
3163ee995283SPavel Emelyanov 	case TCP_QUEUE_SEQ:
3164ee995283SPavel Emelyanov 		if (sk->sk_state != TCP_CLOSE)
3165ee995283SPavel Emelyanov 			err = -EPERM;
3166ee995283SPavel Emelyanov 		else if (tp->repair_queue == TCP_SEND_QUEUE)
31670f317464SEric Dumazet 			WRITE_ONCE(tp->write_seq, val);
31686cd6cbf5SEric Dumazet 		else if (tp->repair_queue == TCP_RECV_QUEUE) {
3169dba7d9b8SEric Dumazet 			WRITE_ONCE(tp->rcv_nxt, val);
31706cd6cbf5SEric Dumazet 			WRITE_ONCE(tp->copied_seq, val);
31716cd6cbf5SEric Dumazet 		}
3172ee995283SPavel Emelyanov 		else
3173ee995283SPavel Emelyanov 			err = -EINVAL;
3174ee995283SPavel Emelyanov 		break;
3175ee995283SPavel Emelyanov 
3176b139ba4eSPavel Emelyanov 	case TCP_REPAIR_OPTIONS:
3177b139ba4eSPavel Emelyanov 		if (!tp->repair)
3178b139ba4eSPavel Emelyanov 			err = -EINVAL;
3179b139ba4eSPavel Emelyanov 		else if (sk->sk_state == TCP_ESTABLISHED)
3180d38d2b00SChristoph Hellwig 			err = tcp_repair_options_est(sk, optval, optlen);
3181b139ba4eSPavel Emelyanov 		else
3182b139ba4eSPavel Emelyanov 			err = -EPERM;
3183b139ba4eSPavel Emelyanov 		break;
3184b139ba4eSPavel Emelyanov 
31851da177e4SLinus Torvalds 	case TCP_CORK:
3186db10538aSChristoph Hellwig 		__tcp_sock_set_cork(sk, val);
31871da177e4SLinus Torvalds 		break;
31881da177e4SLinus Torvalds 
31891da177e4SLinus Torvalds 	case TCP_KEEPIDLE:
3190aad4a0a9SDmitry Yakunin 		err = tcp_sock_set_keepidle_locked(sk, val);
31911da177e4SLinus Torvalds 		break;
31921da177e4SLinus Torvalds 	case TCP_KEEPINTVL:
31931da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_KEEPINTVL)
31941da177e4SLinus Torvalds 			err = -EINVAL;
31951da177e4SLinus Torvalds 		else
31961da177e4SLinus Torvalds 			tp->keepalive_intvl = val * HZ;
31971da177e4SLinus Torvalds 		break;
31981da177e4SLinus Torvalds 	case TCP_KEEPCNT:
31991da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_KEEPCNT)
32001da177e4SLinus Torvalds 			err = -EINVAL;
32011da177e4SLinus Torvalds 		else
32021da177e4SLinus Torvalds 			tp->keepalive_probes = val;
32031da177e4SLinus Torvalds 		break;
32041da177e4SLinus Torvalds 	case TCP_SYNCNT:
32051da177e4SLinus Torvalds 		if (val < 1 || val > MAX_TCP_SYNCNT)
32061da177e4SLinus Torvalds 			err = -EINVAL;
32071da177e4SLinus Torvalds 		else
3208463c84b9SArnaldo Carvalho de Melo 			icsk->icsk_syn_retries = val;
32091da177e4SLinus Torvalds 		break;
32101da177e4SLinus Torvalds 
3211cd8ae852SEric Dumazet 	case TCP_SAVE_SYN:
3212267cf9faSMartin KaFai Lau 		/* 0: disable, 1: enable, 2: start from ether_header */
3213267cf9faSMartin KaFai Lau 		if (val < 0 || val > 2)
3214cd8ae852SEric Dumazet 			err = -EINVAL;
3215cd8ae852SEric Dumazet 		else
3216cd8ae852SEric Dumazet 			tp->save_syn = val;
3217cd8ae852SEric Dumazet 		break;
3218cd8ae852SEric Dumazet 
32191da177e4SLinus Torvalds 	case TCP_LINGER2:
32201da177e4SLinus Torvalds 		if (val < 0)
32211da177e4SLinus Torvalds 			tp->linger2 = -1;
3222f0628c52SCambda Zhu 		else if (val > TCP_FIN_TIMEOUT_MAX / HZ)
3223f0628c52SCambda Zhu 			tp->linger2 = TCP_FIN_TIMEOUT_MAX;
32241da177e4SLinus Torvalds 		else
32251da177e4SLinus Torvalds 			tp->linger2 = val * HZ;
32261da177e4SLinus Torvalds 		break;
32271da177e4SLinus Torvalds 
32281da177e4SLinus Torvalds 	case TCP_DEFER_ACCEPT:
3229b103cf34SJulian Anastasov 		/* Translate value in seconds to number of retransmits */
3230b103cf34SJulian Anastasov 		icsk->icsk_accept_queue.rskq_defer_accept =
3231b103cf34SJulian Anastasov 			secs_to_retrans(val, TCP_TIMEOUT_INIT / HZ,
3232b103cf34SJulian Anastasov 					TCP_RTO_MAX / HZ);
32331da177e4SLinus Torvalds 		break;
32341da177e4SLinus Torvalds 
32351da177e4SLinus Torvalds 	case TCP_WINDOW_CLAMP:
32361da177e4SLinus Torvalds 		if (!val) {
32371da177e4SLinus Torvalds 			if (sk->sk_state != TCP_CLOSE) {
32381da177e4SLinus Torvalds 				err = -EINVAL;
32391da177e4SLinus Torvalds 				break;
32401da177e4SLinus Torvalds 			}
32411da177e4SLinus Torvalds 			tp->window_clamp = 0;
32421da177e4SLinus Torvalds 		} else
32431da177e4SLinus Torvalds 			tp->window_clamp = val < SOCK_MIN_RCVBUF / 2 ?
32441da177e4SLinus Torvalds 						SOCK_MIN_RCVBUF / 2 : val;
32451da177e4SLinus Torvalds 		break;
32461da177e4SLinus Torvalds 
32471da177e4SLinus Torvalds 	case TCP_QUICKACK:
3248ddd061b8SChristoph Hellwig 		__tcp_sock_set_quickack(sk, val);
32491da177e4SLinus Torvalds 		break;
32501da177e4SLinus Torvalds 
3251cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG
3252cfb6eeb4SYOSHIFUJI Hideaki 	case TCP_MD5SIG:
32538917a777SIvan Delalande 	case TCP_MD5SIG_EXT:
3254d38d2b00SChristoph Hellwig 		err = tp->af_specific->md5_parse(sk, optname, optval, optlen);
3255cfb6eeb4SYOSHIFUJI Hideaki 		break;
3256cfb6eeb4SYOSHIFUJI Hideaki #endif
3257dca43c75SJerry Chu 	case TCP_USER_TIMEOUT:
3258b248230cSYuchung Cheng 		/* Cap the max time in ms TCP will retry or probe the window
3259dca43c75SJerry Chu 		 * before giving up and aborting (ETIMEDOUT) a connection.
3260dca43c75SJerry Chu 		 */
326142493570SHangbin Liu 		if (val < 0)
326242493570SHangbin Liu 			err = -EINVAL;
326342493570SHangbin Liu 		else
32649bcc66e1SJon Maxwell 			icsk->icsk_user_timeout = val;
3265dca43c75SJerry Chu 		break;
32668336886fSJerry Chu 
32678336886fSJerry Chu 	case TCP_FASTOPEN:
32688336886fSJerry Chu 		if (val >= 0 && ((1 << sk->sk_state) & (TCPF_CLOSE |
3269dfea2aa6SChristoph Paasch 		    TCPF_LISTEN))) {
327043713848SHaishuang Yan 			tcp_fastopen_init_key_once(net);
3271dfea2aa6SChristoph Paasch 
32720536fcc0SEric Dumazet 			fastopen_queue_tune(sk, val);
3273dfea2aa6SChristoph Paasch 		} else {
32748336886fSJerry Chu 			err = -EINVAL;
3275dfea2aa6SChristoph Paasch 		}
32768336886fSJerry Chu 		break;
327719f6d3f3SWei Wang 	case TCP_FASTOPEN_CONNECT:
327819f6d3f3SWei Wang 		if (val > 1 || val < 0) {
327919f6d3f3SWei Wang 			err = -EINVAL;
3280e1cfcbe8SHaishuang Yan 		} else if (net->ipv4.sysctl_tcp_fastopen & TFO_CLIENT_ENABLE) {
328119f6d3f3SWei Wang 			if (sk->sk_state == TCP_CLOSE)
328219f6d3f3SWei Wang 				tp->fastopen_connect = val;
328319f6d3f3SWei Wang 			else
328419f6d3f3SWei Wang 				err = -EINVAL;
328519f6d3f3SWei Wang 		} else {
328619f6d3f3SWei Wang 			err = -EOPNOTSUPP;
328719f6d3f3SWei Wang 		}
328819f6d3f3SWei Wang 		break;
328971c02379SChristoph Paasch 	case TCP_FASTOPEN_NO_COOKIE:
329071c02379SChristoph Paasch 		if (val > 1 || val < 0)
329171c02379SChristoph Paasch 			err = -EINVAL;
329271c02379SChristoph Paasch 		else if (!((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)))
329371c02379SChristoph Paasch 			err = -EINVAL;
329471c02379SChristoph Paasch 		else
329571c02379SChristoph Paasch 			tp->fastopen_no_cookie = val;
329671c02379SChristoph Paasch 		break;
329793be6ce0SAndrey Vagin 	case TCP_TIMESTAMP:
329893be6ce0SAndrey Vagin 		if (!tp->repair)
329993be6ce0SAndrey Vagin 			err = -EPERM;
330093be6ce0SAndrey Vagin 		else
33019a568de4SEric Dumazet 			tp->tsoffset = val - tcp_time_stamp_raw();
330293be6ce0SAndrey Vagin 		break;
3303b1ed4c4fSAndrey Vagin 	case TCP_REPAIR_WINDOW:
3304b1ed4c4fSAndrey Vagin 		err = tcp_repair_set_window(tp, optval, optlen);
3305b1ed4c4fSAndrey Vagin 		break;
3306c9bee3b7SEric Dumazet 	case TCP_NOTSENT_LOWAT:
3307c9bee3b7SEric Dumazet 		tp->notsent_lowat = val;
3308c9bee3b7SEric Dumazet 		sk->sk_write_space(sk);
3309c9bee3b7SEric Dumazet 		break;
3310b75eba76SSoheil Hassas Yeganeh 	case TCP_INQ:
3311b75eba76SSoheil Hassas Yeganeh 		if (val > 1 || val < 0)
3312b75eba76SSoheil Hassas Yeganeh 			err = -EINVAL;
3313b75eba76SSoheil Hassas Yeganeh 		else
3314b75eba76SSoheil Hassas Yeganeh 			tp->recvmsg_inq = val;
3315b75eba76SSoheil Hassas Yeganeh 		break;
3316a842fe14SEric Dumazet 	case TCP_TX_DELAY:
3317a842fe14SEric Dumazet 		if (val)
3318a842fe14SEric Dumazet 			tcp_enable_tx_delay();
3319a842fe14SEric Dumazet 		tp->tcp_tx_delay = val;
3320a842fe14SEric Dumazet 		break;
33211da177e4SLinus Torvalds 	default:
33221da177e4SLinus Torvalds 		err = -ENOPROTOOPT;
33231da177e4SLinus Torvalds 		break;
33243ff50b79SStephen Hemminger 	}
33253ff50b79SStephen Hemminger 
33261da177e4SLinus Torvalds 	release_sock(sk);
33271da177e4SLinus Torvalds 	return err;
33281da177e4SLinus Torvalds }
33291da177e4SLinus Torvalds 
3330a7b75c5aSChristoph Hellwig int tcp_setsockopt(struct sock *sk, int level, int optname, sockptr_t optval,
3331b7058842SDavid S. Miller 		   unsigned int optlen)
33323fdadf7dSDmitry Mishin {
3333cf533ea5SEric Dumazet 	const struct inet_connection_sock *icsk = inet_csk(sk);
33343fdadf7dSDmitry Mishin 
33353fdadf7dSDmitry Mishin 	if (level != SOL_TCP)
33363fdadf7dSDmitry Mishin 		return icsk->icsk_af_ops->setsockopt(sk, level, optname,
33373fdadf7dSDmitry Mishin 						     optval, optlen);
3338a7b75c5aSChristoph Hellwig 	return do_tcp_setsockopt(sk, level, optname, optval, optlen);
33393fdadf7dSDmitry Mishin }
33404bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_setsockopt);
33413fdadf7dSDmitry Mishin 
3342efd90174SFrancis Yan static void tcp_get_info_chrono_stats(const struct tcp_sock *tp,
3343efd90174SFrancis Yan 				      struct tcp_info *info)
3344efd90174SFrancis Yan {
3345efd90174SFrancis Yan 	u64 stats[__TCP_CHRONO_MAX], total = 0;
3346efd90174SFrancis Yan 	enum tcp_chrono i;
3347efd90174SFrancis Yan 
3348efd90174SFrancis Yan 	for (i = TCP_CHRONO_BUSY; i < __TCP_CHRONO_MAX; ++i) {
3349efd90174SFrancis Yan 		stats[i] = tp->chrono_stat[i - 1];
3350efd90174SFrancis Yan 		if (i == tp->chrono_type)
3351628174ccSEric Dumazet 			stats[i] += tcp_jiffies32 - tp->chrono_start;
3352efd90174SFrancis Yan 		stats[i] *= USEC_PER_SEC / HZ;
3353efd90174SFrancis Yan 		total += stats[i];
3354efd90174SFrancis Yan 	}
3355efd90174SFrancis Yan 
3356efd90174SFrancis Yan 	info->tcpi_busy_time = total;
3357efd90174SFrancis Yan 	info->tcpi_rwnd_limited = stats[TCP_CHRONO_RWND_LIMITED];
3358efd90174SFrancis Yan 	info->tcpi_sndbuf_limited = stats[TCP_CHRONO_SNDBUF_LIMITED];
3359efd90174SFrancis Yan }
3360efd90174SFrancis Yan 
33611da177e4SLinus Torvalds /* Return information about state of tcp endpoint in API format. */
33620df48c26SEric Dumazet void tcp_get_info(struct sock *sk, struct tcp_info *info)
33631da177e4SLinus Torvalds {
336435ac838aSCraig Gallek 	const struct tcp_sock *tp = tcp_sk(sk); /* iff sk_type == SOCK_STREAM */
3365463c84b9SArnaldo Carvalho de Melo 	const struct inet_connection_sock *icsk = inet_csk(sk);
336676a9ebe8SEric Dumazet 	unsigned long rate;
33670263598cSWei Wang 	u32 now;
3368ff5d7497SEric Dumazet 	u64 rate64;
336967db3e4bSEric Dumazet 	bool slow;
33701da177e4SLinus Torvalds 
33711da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
337235ac838aSCraig Gallek 	if (sk->sk_type != SOCK_STREAM)
337335ac838aSCraig Gallek 		return;
33741da177e4SLinus Torvalds 
3375986ffdfdSYafang Shao 	info->tcpi_state = inet_sk_state_load(sk);
337600fd38d9SEric Dumazet 
3377ccbf3bfaSEric Dumazet 	/* Report meaningful fields for all TCP states, including listeners */
3378ccbf3bfaSEric Dumazet 	rate = READ_ONCE(sk->sk_pacing_rate);
337976a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3380f522a5fcSEric Dumazet 	info->tcpi_pacing_rate = rate64;
3381ccbf3bfaSEric Dumazet 
3382ccbf3bfaSEric Dumazet 	rate = READ_ONCE(sk->sk_max_pacing_rate);
338376a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3384f522a5fcSEric Dumazet 	info->tcpi_max_pacing_rate = rate64;
3385ccbf3bfaSEric Dumazet 
3386ccbf3bfaSEric Dumazet 	info->tcpi_reordering = tp->reordering;
3387ccbf3bfaSEric Dumazet 	info->tcpi_snd_cwnd = tp->snd_cwnd;
3388ccbf3bfaSEric Dumazet 
3389ccbf3bfaSEric Dumazet 	if (info->tcpi_state == TCP_LISTEN) {
3390ccbf3bfaSEric Dumazet 		/* listeners aliased fields :
3391ccbf3bfaSEric Dumazet 		 * tcpi_unacked -> Number of children ready for accept()
3392ccbf3bfaSEric Dumazet 		 * tcpi_sacked  -> max backlog
3393ccbf3bfaSEric Dumazet 		 */
3394288efe86SEric Dumazet 		info->tcpi_unacked = READ_ONCE(sk->sk_ack_backlog);
3395099ecf59SEric Dumazet 		info->tcpi_sacked = READ_ONCE(sk->sk_max_ack_backlog);
3396ccbf3bfaSEric Dumazet 		return;
3397ccbf3bfaSEric Dumazet 	}
3398b369e7fdSEric Dumazet 
3399b369e7fdSEric Dumazet 	slow = lock_sock_fast(sk);
3400b369e7fdSEric Dumazet 
34016687e988SArnaldo Carvalho de Melo 	info->tcpi_ca_state = icsk->icsk_ca_state;
3402463c84b9SArnaldo Carvalho de Melo 	info->tcpi_retransmits = icsk->icsk_retransmits;
34036687e988SArnaldo Carvalho de Melo 	info->tcpi_probes = icsk->icsk_probes_out;
3404463c84b9SArnaldo Carvalho de Melo 	info->tcpi_backoff = icsk->icsk_backoff;
34051da177e4SLinus Torvalds 
34061da177e4SLinus Torvalds 	if (tp->rx_opt.tstamp_ok)
34071da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_TIMESTAMPS;
3408e60402d0SIlpo Järvinen 	if (tcp_is_sack(tp))
34091da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_SACK;
34101da177e4SLinus Torvalds 	if (tp->rx_opt.wscale_ok) {
34111da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_WSCALE;
34121da177e4SLinus Torvalds 		info->tcpi_snd_wscale = tp->rx_opt.snd_wscale;
34131da177e4SLinus Torvalds 		info->tcpi_rcv_wscale = tp->rx_opt.rcv_wscale;
34141da177e4SLinus Torvalds 	}
34151da177e4SLinus Torvalds 
34161da177e4SLinus Torvalds 	if (tp->ecn_flags & TCP_ECN_OK)
34171da177e4SLinus Torvalds 		info->tcpi_options |= TCPI_OPT_ECN;
3418b5c5693bSEric Dumazet 	if (tp->ecn_flags & TCP_ECN_SEEN)
3419b5c5693bSEric Dumazet 		info->tcpi_options |= TCPI_OPT_ECN_SEEN;
34206f73601eSYuchung Cheng 	if (tp->syn_data_acked)
34216f73601eSYuchung Cheng 		info->tcpi_options |= TCPI_OPT_SYN_DATA;
34221da177e4SLinus Torvalds 
3423463c84b9SArnaldo Carvalho de Melo 	info->tcpi_rto = jiffies_to_usecs(icsk->icsk_rto);
3424463c84b9SArnaldo Carvalho de Melo 	info->tcpi_ato = jiffies_to_usecs(icsk->icsk_ack.ato);
3425c1b4a7e6SDavid S. Miller 	info->tcpi_snd_mss = tp->mss_cache;
3426463c84b9SArnaldo Carvalho de Melo 	info->tcpi_rcv_mss = icsk->icsk_ack.rcv_mss;
34271da177e4SLinus Torvalds 
34281da177e4SLinus Torvalds 	info->tcpi_unacked = tp->packets_out;
34291da177e4SLinus Torvalds 	info->tcpi_sacked = tp->sacked_out;
3430ccbf3bfaSEric Dumazet 
34311da177e4SLinus Torvalds 	info->tcpi_lost = tp->lost_out;
34321da177e4SLinus Torvalds 	info->tcpi_retrans = tp->retrans_out;
34331da177e4SLinus Torvalds 
3434d635fbe2SEric Dumazet 	now = tcp_jiffies32;
34351da177e4SLinus Torvalds 	info->tcpi_last_data_sent = jiffies_to_msecs(now - tp->lsndtime);
3436463c84b9SArnaldo Carvalho de Melo 	info->tcpi_last_data_recv = jiffies_to_msecs(now - icsk->icsk_ack.lrcvtime);
34371da177e4SLinus Torvalds 	info->tcpi_last_ack_recv = jiffies_to_msecs(now - tp->rcv_tstamp);
34381da177e4SLinus Torvalds 
3439d83d8461SArnaldo Carvalho de Melo 	info->tcpi_pmtu = icsk->icsk_pmtu_cookie;
34401da177e4SLinus Torvalds 	info->tcpi_rcv_ssthresh = tp->rcv_ssthresh;
3441740b0f18SEric Dumazet 	info->tcpi_rtt = tp->srtt_us >> 3;
3442740b0f18SEric Dumazet 	info->tcpi_rttvar = tp->mdev_us >> 2;
34431da177e4SLinus Torvalds 	info->tcpi_snd_ssthresh = tp->snd_ssthresh;
34441da177e4SLinus Torvalds 	info->tcpi_advmss = tp->advmss;
34451da177e4SLinus Torvalds 
3446645f4c6fSEric Dumazet 	info->tcpi_rcv_rtt = tp->rcv_rtt_est.rtt_us >> 3;
34471da177e4SLinus Torvalds 	info->tcpi_rcv_space = tp->rcvq_space.space;
34481da177e4SLinus Torvalds 
34491da177e4SLinus Torvalds 	info->tcpi_total_retrans = tp->total_retrans;
3450977cb0ecSEric Dumazet 
3451f522a5fcSEric Dumazet 	info->tcpi_bytes_acked = tp->bytes_acked;
3452f522a5fcSEric Dumazet 	info->tcpi_bytes_received = tp->bytes_received;
345367db3e4bSEric Dumazet 	info->tcpi_notsent_bytes = max_t(int, 0, tp->write_seq - tp->snd_nxt);
3454efd90174SFrancis Yan 	tcp_get_info_chrono_stats(tp, info);
345567db3e4bSEric Dumazet 
34562efd055cSMarcelo Ricardo Leitner 	info->tcpi_segs_out = tp->segs_out;
34572efd055cSMarcelo Ricardo Leitner 	info->tcpi_segs_in = tp->segs_in;
3458cd9b2660SEric Dumazet 
3459cd9b2660SEric Dumazet 	info->tcpi_min_rtt = tcp_min_rtt(tp);
3460a44d6eacSMartin KaFai Lau 	info->tcpi_data_segs_in = tp->data_segs_in;
3461a44d6eacSMartin KaFai Lau 	info->tcpi_data_segs_out = tp->data_segs_out;
3462eb8329e0SYuchung Cheng 
3463eb8329e0SYuchung Cheng 	info->tcpi_delivery_rate_app_limited = tp->rate_app_limited ? 1 : 0;
34640263598cSWei Wang 	rate64 = tcp_compute_delivery_rate(tp);
34650263598cSWei Wang 	if (rate64)
3466f522a5fcSEric Dumazet 		info->tcpi_delivery_rate = rate64;
3467feb5f2ecSYuchung Cheng 	info->tcpi_delivered = tp->delivered;
3468feb5f2ecSYuchung Cheng 	info->tcpi_delivered_ce = tp->delivered_ce;
3469ba113c3aSWei Wang 	info->tcpi_bytes_sent = tp->bytes_sent;
3470fb31c9b9SWei Wang 	info->tcpi_bytes_retrans = tp->bytes_retrans;
34717e10b655SWei Wang 	info->tcpi_dsack_dups = tp->dsack_dups;
34727ec65372SWei Wang 	info->tcpi_reord_seen = tp->reord_seen;
3473f9af2dbbSThomas Higdon 	info->tcpi_rcv_ooopack = tp->rcv_ooopack;
34748f7baad7SThomas Higdon 	info->tcpi_snd_wnd = tp->snd_wnd;
347548027478SJason Baron 	info->tcpi_fastopen_client_fail = tp->fastopen_client_fail;
3476b369e7fdSEric Dumazet 	unlock_sock_fast(sk, slow);
34771da177e4SLinus Torvalds }
34781da177e4SLinus Torvalds EXPORT_SYMBOL_GPL(tcp_get_info);
34791da177e4SLinus Torvalds 
3480984988aaSWei Wang static size_t tcp_opt_stats_get_size(void)
3481984988aaSWei Wang {
3482984988aaSWei Wang 	return
3483984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BUSY */
3484984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_RWND_LIMITED */
3485984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_SNDBUF_LIMITED */
3486984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DATA_SEGS_OUT */
3487984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_TOTAL_RETRANS */
3488984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_PACING_RATE */
3489984988aaSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DELIVERY_RATE */
3490984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_CWND */
3491984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_REORDERING */
3492984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_MIN_RTT */
3493984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_RECUR_RETRANS */
3494984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_DELIVERY_RATE_APP_LMT */
3495984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SNDQ_SIZE */
3496984988aaSWei Wang 		nla_total_size(sizeof(u8)) + /* TCP_NLA_CA_STATE */
3497984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_SSTHRESH */
3498984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED */
3499984988aaSWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED_CE */
3500ba113c3aSWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_SENT */
3501fb31c9b9SWei Wang 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_RETRANS */
35027e10b655SWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_DSACK_DUPS */
35037ec65372SWei Wang 		nla_total_size(sizeof(u32)) + /* TCP_NLA_REORD_SEEN */
3504e8bd8fcaSYousuk Seung 		nla_total_size(sizeof(u32)) + /* TCP_NLA_SRTT */
350532efcc06SAbdul Kabbani 		nla_total_size(sizeof(u16)) + /* TCP_NLA_TIMEOUT_REHASH */
3506e08ab0b3SYousuk Seung 		nla_total_size(sizeof(u32)) + /* TCP_NLA_BYTES_NOTSENT */
350748040793SYousuk Seung 		nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_EDT */
3508984988aaSWei Wang 		0;
3509984988aaSWei Wang }
3510984988aaSWei Wang 
351148040793SYousuk Seung struct sk_buff *tcp_get_timestamping_opt_stats(const struct sock *sk,
351248040793SYousuk Seung 					       const struct sk_buff *orig_skb)
35131c885808SFrancis Yan {
35141c885808SFrancis Yan 	const struct tcp_sock *tp = tcp_sk(sk);
35151c885808SFrancis Yan 	struct sk_buff *stats;
35161c885808SFrancis Yan 	struct tcp_info info;
351776a9ebe8SEric Dumazet 	unsigned long rate;
3518bb7c19f9SWei Wang 	u64 rate64;
35191c885808SFrancis Yan 
3520984988aaSWei Wang 	stats = alloc_skb(tcp_opt_stats_get_size(), GFP_ATOMIC);
35211c885808SFrancis Yan 	if (!stats)
35221c885808SFrancis Yan 		return NULL;
35231c885808SFrancis Yan 
35241c885808SFrancis Yan 	tcp_get_info_chrono_stats(tp, &info);
35251c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_BUSY,
35261c885808SFrancis Yan 			  info.tcpi_busy_time, TCP_NLA_PAD);
35271c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_RWND_LIMITED,
35281c885808SFrancis Yan 			  info.tcpi_rwnd_limited, TCP_NLA_PAD);
35291c885808SFrancis Yan 	nla_put_u64_64bit(stats, TCP_NLA_SNDBUF_LIMITED,
35301c885808SFrancis Yan 			  info.tcpi_sndbuf_limited, TCP_NLA_PAD);
35317e98102fSYuchung Cheng 	nla_put_u64_64bit(stats, TCP_NLA_DATA_SEGS_OUT,
35327e98102fSYuchung Cheng 			  tp->data_segs_out, TCP_NLA_PAD);
35337e98102fSYuchung Cheng 	nla_put_u64_64bit(stats, TCP_NLA_TOTAL_RETRANS,
35347e98102fSYuchung Cheng 			  tp->total_retrans, TCP_NLA_PAD);
3535bb7c19f9SWei Wang 
3536bb7c19f9SWei Wang 	rate = READ_ONCE(sk->sk_pacing_rate);
353776a9ebe8SEric Dumazet 	rate64 = (rate != ~0UL) ? rate : ~0ULL;
3538bb7c19f9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_PACING_RATE, rate64, TCP_NLA_PAD);
3539bb7c19f9SWei Wang 
3540bb7c19f9SWei Wang 	rate64 = tcp_compute_delivery_rate(tp);
3541bb7c19f9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_DELIVERY_RATE, rate64, TCP_NLA_PAD);
3542bb7c19f9SWei Wang 
3543bb7c19f9SWei Wang 	nla_put_u32(stats, TCP_NLA_SND_CWND, tp->snd_cwnd);
3544bb7c19f9SWei Wang 	nla_put_u32(stats, TCP_NLA_REORDERING, tp->reordering);
3545bb7c19f9SWei Wang 	nla_put_u32(stats, TCP_NLA_MIN_RTT, tcp_min_rtt(tp));
3546bb7c19f9SWei Wang 
3547bb7c19f9SWei Wang 	nla_put_u8(stats, TCP_NLA_RECUR_RETRANS, inet_csk(sk)->icsk_retransmits);
3548bb7c19f9SWei Wang 	nla_put_u8(stats, TCP_NLA_DELIVERY_RATE_APP_LMT, !!tp->rate_app_limited);
35497156d194SYousuk Seung 	nla_put_u32(stats, TCP_NLA_SND_SSTHRESH, tp->snd_ssthresh);
3550feb5f2ecSYuchung Cheng 	nla_put_u32(stats, TCP_NLA_DELIVERED, tp->delivered);
3551feb5f2ecSYuchung Cheng 	nla_put_u32(stats, TCP_NLA_DELIVERED_CE, tp->delivered_ce);
355287ecc95dSPriyaranjan Jha 
355387ecc95dSPriyaranjan Jha 	nla_put_u32(stats, TCP_NLA_SNDQ_SIZE, tp->write_seq - tp->snd_una);
3554be631892SPriyaranjan Jha 	nla_put_u8(stats, TCP_NLA_CA_STATE, inet_csk(sk)->icsk_ca_state);
3555feb5f2ecSYuchung Cheng 
3556ba113c3aSWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_BYTES_SENT, tp->bytes_sent,
3557ba113c3aSWei Wang 			  TCP_NLA_PAD);
3558fb31c9b9SWei Wang 	nla_put_u64_64bit(stats, TCP_NLA_BYTES_RETRANS, tp->bytes_retrans,
3559fb31c9b9SWei Wang 			  TCP_NLA_PAD);
35607e10b655SWei Wang 	nla_put_u32(stats, TCP_NLA_DSACK_DUPS, tp->dsack_dups);
35617ec65372SWei Wang 	nla_put_u32(stats, TCP_NLA_REORD_SEEN, tp->reord_seen);
3562e8bd8fcaSYousuk Seung 	nla_put_u32(stats, TCP_NLA_SRTT, tp->srtt_us >> 3);
356332efcc06SAbdul Kabbani 	nla_put_u16(stats, TCP_NLA_TIMEOUT_REHASH, tp->timeout_rehash);
3564e08ab0b3SYousuk Seung 	nla_put_u32(stats, TCP_NLA_BYTES_NOTSENT,
3565e08ab0b3SYousuk Seung 		    max_t(int, 0, tp->write_seq - tp->snd_nxt));
356648040793SYousuk Seung 	nla_put_u64_64bit(stats, TCP_NLA_EDT, orig_skb->skb_mstamp_ns,
356748040793SYousuk Seung 			  TCP_NLA_PAD);
3568ba113c3aSWei Wang 
35691c885808SFrancis Yan 	return stats;
35701c885808SFrancis Yan }
35711c885808SFrancis Yan 
35723fdadf7dSDmitry Mishin static int do_tcp_getsockopt(struct sock *sk, int level,
35733fdadf7dSDmitry Mishin 		int optname, char __user *optval, int __user *optlen)
35741da177e4SLinus Torvalds {
3575295f7324SArnaldo Carvalho de Melo 	struct inet_connection_sock *icsk = inet_csk(sk);
35761da177e4SLinus Torvalds 	struct tcp_sock *tp = tcp_sk(sk);
35776fa25166SNikolay Borisov 	struct net *net = sock_net(sk);
35781da177e4SLinus Torvalds 	int val, len;
35791da177e4SLinus Torvalds 
35801da177e4SLinus Torvalds 	if (get_user(len, optlen))
35811da177e4SLinus Torvalds 		return -EFAULT;
35821da177e4SLinus Torvalds 
35831da177e4SLinus Torvalds 	len = min_t(unsigned int, len, sizeof(int));
35841da177e4SLinus Torvalds 
35851da177e4SLinus Torvalds 	if (len < 0)
35861da177e4SLinus Torvalds 		return -EINVAL;
35871da177e4SLinus Torvalds 
35881da177e4SLinus Torvalds 	switch (optname) {
35891da177e4SLinus Torvalds 	case TCP_MAXSEG:
3590c1b4a7e6SDavid S. Miller 		val = tp->mss_cache;
35911da177e4SLinus Torvalds 		if (!val && ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)))
35921da177e4SLinus Torvalds 			val = tp->rx_opt.user_mss;
35935e6a3ce6SPavel Emelyanov 		if (tp->repair)
35945e6a3ce6SPavel Emelyanov 			val = tp->rx_opt.mss_clamp;
35951da177e4SLinus Torvalds 		break;
35961da177e4SLinus Torvalds 	case TCP_NODELAY:
35971da177e4SLinus Torvalds 		val = !!(tp->nonagle&TCP_NAGLE_OFF);
35981da177e4SLinus Torvalds 		break;
35991da177e4SLinus Torvalds 	case TCP_CORK:
36001da177e4SLinus Torvalds 		val = !!(tp->nonagle&TCP_NAGLE_CORK);
36011da177e4SLinus Torvalds 		break;
36021da177e4SLinus Torvalds 	case TCP_KEEPIDLE:
3603df19a626SEric Dumazet 		val = keepalive_time_when(tp) / HZ;
36041da177e4SLinus Torvalds 		break;
36051da177e4SLinus Torvalds 	case TCP_KEEPINTVL:
3606df19a626SEric Dumazet 		val = keepalive_intvl_when(tp) / HZ;
36071da177e4SLinus Torvalds 		break;
36081da177e4SLinus Torvalds 	case TCP_KEEPCNT:
3609df19a626SEric Dumazet 		val = keepalive_probes(tp);
36101da177e4SLinus Torvalds 		break;
36111da177e4SLinus Torvalds 	case TCP_SYNCNT:
36126fa25166SNikolay Borisov 		val = icsk->icsk_syn_retries ? : net->ipv4.sysctl_tcp_syn_retries;
36131da177e4SLinus Torvalds 		break;
36141da177e4SLinus Torvalds 	case TCP_LINGER2:
36151da177e4SLinus Torvalds 		val = tp->linger2;
36161da177e4SLinus Torvalds 		if (val >= 0)
36171e579caaSNikolay Borisov 			val = (val ? : net->ipv4.sysctl_tcp_fin_timeout) / HZ;
36181da177e4SLinus Torvalds 		break;
36191da177e4SLinus Torvalds 	case TCP_DEFER_ACCEPT:
3620b103cf34SJulian Anastasov 		val = retrans_to_secs(icsk->icsk_accept_queue.rskq_defer_accept,
3621b103cf34SJulian Anastasov 				      TCP_TIMEOUT_INIT / HZ, TCP_RTO_MAX / HZ);
36221da177e4SLinus Torvalds 		break;
36231da177e4SLinus Torvalds 	case TCP_WINDOW_CLAMP:
36241da177e4SLinus Torvalds 		val = tp->window_clamp;
36251da177e4SLinus Torvalds 		break;
36261da177e4SLinus Torvalds 	case TCP_INFO: {
36271da177e4SLinus Torvalds 		struct tcp_info info;
36281da177e4SLinus Torvalds 
36291da177e4SLinus Torvalds 		if (get_user(len, optlen))
36301da177e4SLinus Torvalds 			return -EFAULT;
36311da177e4SLinus Torvalds 
36321da177e4SLinus Torvalds 		tcp_get_info(sk, &info);
36331da177e4SLinus Torvalds 
36341da177e4SLinus Torvalds 		len = min_t(unsigned int, len, sizeof(info));
36351da177e4SLinus Torvalds 		if (put_user(len, optlen))
36361da177e4SLinus Torvalds 			return -EFAULT;
36371da177e4SLinus Torvalds 		if (copy_to_user(optval, &info, len))
36381da177e4SLinus Torvalds 			return -EFAULT;
36391da177e4SLinus Torvalds 		return 0;
36401da177e4SLinus Torvalds 	}
36416e9250f5SEric Dumazet 	case TCP_CC_INFO: {
36426e9250f5SEric Dumazet 		const struct tcp_congestion_ops *ca_ops;
36436e9250f5SEric Dumazet 		union tcp_cc_info info;
36446e9250f5SEric Dumazet 		size_t sz = 0;
36456e9250f5SEric Dumazet 		int attr;
36466e9250f5SEric Dumazet 
36476e9250f5SEric Dumazet 		if (get_user(len, optlen))
36486e9250f5SEric Dumazet 			return -EFAULT;
36496e9250f5SEric Dumazet 
36506e9250f5SEric Dumazet 		ca_ops = icsk->icsk_ca_ops;
36516e9250f5SEric Dumazet 		if (ca_ops && ca_ops->get_info)
36526e9250f5SEric Dumazet 			sz = ca_ops->get_info(sk, ~0U, &attr, &info);
36536e9250f5SEric Dumazet 
36546e9250f5SEric Dumazet 		len = min_t(unsigned int, len, sz);
36556e9250f5SEric Dumazet 		if (put_user(len, optlen))
36566e9250f5SEric Dumazet 			return -EFAULT;
36576e9250f5SEric Dumazet 		if (copy_to_user(optval, &info, len))
36586e9250f5SEric Dumazet 			return -EFAULT;
36596e9250f5SEric Dumazet 		return 0;
36606e9250f5SEric Dumazet 	}
36611da177e4SLinus Torvalds 	case TCP_QUICKACK:
366231954cd8SWei Wang 		val = !inet_csk_in_pingpong_mode(sk);
36631da177e4SLinus Torvalds 		break;
36645f8ef48dSStephen Hemminger 
36655f8ef48dSStephen Hemminger 	case TCP_CONGESTION:
36665f8ef48dSStephen Hemminger 		if (get_user(len, optlen))
36675f8ef48dSStephen Hemminger 			return -EFAULT;
36685f8ef48dSStephen Hemminger 		len = min_t(unsigned int, len, TCP_CA_NAME_MAX);
36695f8ef48dSStephen Hemminger 		if (put_user(len, optlen))
36705f8ef48dSStephen Hemminger 			return -EFAULT;
36716687e988SArnaldo Carvalho de Melo 		if (copy_to_user(optval, icsk->icsk_ca_ops->name, len))
36725f8ef48dSStephen Hemminger 			return -EFAULT;
36735f8ef48dSStephen Hemminger 		return 0;
3674e56fb50fSWilliam Allen Simpson 
3675734942ccSDave Watson 	case TCP_ULP:
3676734942ccSDave Watson 		if (get_user(len, optlen))
3677734942ccSDave Watson 			return -EFAULT;
3678734942ccSDave Watson 		len = min_t(unsigned int, len, TCP_ULP_NAME_MAX);
3679d97af30fSDave Watson 		if (!icsk->icsk_ulp_ops) {
3680d97af30fSDave Watson 			if (put_user(0, optlen))
3681d97af30fSDave Watson 				return -EFAULT;
3682d97af30fSDave Watson 			return 0;
3683d97af30fSDave Watson 		}
3684734942ccSDave Watson 		if (put_user(len, optlen))
3685734942ccSDave Watson 			return -EFAULT;
3686734942ccSDave Watson 		if (copy_to_user(optval, icsk->icsk_ulp_ops->name, len))
3687734942ccSDave Watson 			return -EFAULT;
3688734942ccSDave Watson 		return 0;
3689734942ccSDave Watson 
36901fba70e5SYuchung Cheng 	case TCP_FASTOPEN_KEY: {
3691f19008e6SJason Baron 		u64 key[TCP_FASTOPEN_KEY_BUF_LENGTH / sizeof(u64)];
3692f19008e6SJason Baron 		unsigned int key_len;
36931fba70e5SYuchung Cheng 
36941fba70e5SYuchung Cheng 		if (get_user(len, optlen))
36951fba70e5SYuchung Cheng 			return -EFAULT;
36961fba70e5SYuchung Cheng 
3697f19008e6SJason Baron 		key_len = tcp_fastopen_get_cipher(net, icsk, key) *
36980f1ce023SJason Baron 				TCP_FASTOPEN_KEY_LENGTH;
36990f1ce023SJason Baron 		len = min_t(unsigned int, len, key_len);
37001fba70e5SYuchung Cheng 		if (put_user(len, optlen))
37011fba70e5SYuchung Cheng 			return -EFAULT;
37021fba70e5SYuchung Cheng 		if (copy_to_user(optval, key, len))
37031fba70e5SYuchung Cheng 			return -EFAULT;
37041fba70e5SYuchung Cheng 		return 0;
37051fba70e5SYuchung Cheng 	}
37063c0fef0bSJosh Hunt 	case TCP_THIN_LINEAR_TIMEOUTS:
37073c0fef0bSJosh Hunt 		val = tp->thin_lto;
37083c0fef0bSJosh Hunt 		break;
37094a7f6009SYuchung Cheng 
37103c0fef0bSJosh Hunt 	case TCP_THIN_DUPACK:
37114a7f6009SYuchung Cheng 		val = 0;
37123c0fef0bSJosh Hunt 		break;
3713dca43c75SJerry Chu 
3714ee995283SPavel Emelyanov 	case TCP_REPAIR:
3715ee995283SPavel Emelyanov 		val = tp->repair;
3716ee995283SPavel Emelyanov 		break;
3717ee995283SPavel Emelyanov 
3718ee995283SPavel Emelyanov 	case TCP_REPAIR_QUEUE:
3719ee995283SPavel Emelyanov 		if (tp->repair)
3720ee995283SPavel Emelyanov 			val = tp->repair_queue;
3721ee995283SPavel Emelyanov 		else
3722ee995283SPavel Emelyanov 			return -EINVAL;
3723ee995283SPavel Emelyanov 		break;
3724ee995283SPavel Emelyanov 
3725b1ed4c4fSAndrey Vagin 	case TCP_REPAIR_WINDOW: {
3726b1ed4c4fSAndrey Vagin 		struct tcp_repair_window opt;
3727b1ed4c4fSAndrey Vagin 
3728b1ed4c4fSAndrey Vagin 		if (get_user(len, optlen))
3729b1ed4c4fSAndrey Vagin 			return -EFAULT;
3730b1ed4c4fSAndrey Vagin 
3731b1ed4c4fSAndrey Vagin 		if (len != sizeof(opt))
3732b1ed4c4fSAndrey Vagin 			return -EINVAL;
3733b1ed4c4fSAndrey Vagin 
3734b1ed4c4fSAndrey Vagin 		if (!tp->repair)
3735b1ed4c4fSAndrey Vagin 			return -EPERM;
3736b1ed4c4fSAndrey Vagin 
3737b1ed4c4fSAndrey Vagin 		opt.snd_wl1	= tp->snd_wl1;
3738b1ed4c4fSAndrey Vagin 		opt.snd_wnd	= tp->snd_wnd;
3739b1ed4c4fSAndrey Vagin 		opt.max_window	= tp->max_window;
3740b1ed4c4fSAndrey Vagin 		opt.rcv_wnd	= tp->rcv_wnd;
3741b1ed4c4fSAndrey Vagin 		opt.rcv_wup	= tp->rcv_wup;
3742b1ed4c4fSAndrey Vagin 
3743b1ed4c4fSAndrey Vagin 		if (copy_to_user(optval, &opt, len))
3744b1ed4c4fSAndrey Vagin 			return -EFAULT;
3745b1ed4c4fSAndrey Vagin 		return 0;
3746b1ed4c4fSAndrey Vagin 	}
3747ee995283SPavel Emelyanov 	case TCP_QUEUE_SEQ:
3748ee995283SPavel Emelyanov 		if (tp->repair_queue == TCP_SEND_QUEUE)
3749ee995283SPavel Emelyanov 			val = tp->write_seq;
3750ee995283SPavel Emelyanov 		else if (tp->repair_queue == TCP_RECV_QUEUE)
3751ee995283SPavel Emelyanov 			val = tp->rcv_nxt;
3752ee995283SPavel Emelyanov 		else
3753ee995283SPavel Emelyanov 			return -EINVAL;
3754ee995283SPavel Emelyanov 		break;
3755ee995283SPavel Emelyanov 
3756dca43c75SJerry Chu 	case TCP_USER_TIMEOUT:
37579bcc66e1SJon Maxwell 		val = icsk->icsk_user_timeout;
3758dca43c75SJerry Chu 		break;
37591536e285SKenjiro Nakayama 
37601536e285SKenjiro Nakayama 	case TCP_FASTOPEN:
37610536fcc0SEric Dumazet 		val = icsk->icsk_accept_queue.fastopenq.max_qlen;
37621536e285SKenjiro Nakayama 		break;
37631536e285SKenjiro Nakayama 
376419f6d3f3SWei Wang 	case TCP_FASTOPEN_CONNECT:
376519f6d3f3SWei Wang 		val = tp->fastopen_connect;
376619f6d3f3SWei Wang 		break;
376719f6d3f3SWei Wang 
376871c02379SChristoph Paasch 	case TCP_FASTOPEN_NO_COOKIE:
376971c02379SChristoph Paasch 		val = tp->fastopen_no_cookie;
377071c02379SChristoph Paasch 		break;
377171c02379SChristoph Paasch 
3772a842fe14SEric Dumazet 	case TCP_TX_DELAY:
3773a842fe14SEric Dumazet 		val = tp->tcp_tx_delay;
3774a842fe14SEric Dumazet 		break;
3775a842fe14SEric Dumazet 
377693be6ce0SAndrey Vagin 	case TCP_TIMESTAMP:
37779a568de4SEric Dumazet 		val = tcp_time_stamp_raw() + tp->tsoffset;
377893be6ce0SAndrey Vagin 		break;
3779c9bee3b7SEric Dumazet 	case TCP_NOTSENT_LOWAT:
3780c9bee3b7SEric Dumazet 		val = tp->notsent_lowat;
3781c9bee3b7SEric Dumazet 		break;
3782b75eba76SSoheil Hassas Yeganeh 	case TCP_INQ:
3783b75eba76SSoheil Hassas Yeganeh 		val = tp->recvmsg_inq;
3784b75eba76SSoheil Hassas Yeganeh 		break;
3785cd8ae852SEric Dumazet 	case TCP_SAVE_SYN:
3786cd8ae852SEric Dumazet 		val = tp->save_syn;
3787cd8ae852SEric Dumazet 		break;
3788cd8ae852SEric Dumazet 	case TCP_SAVED_SYN: {
3789cd8ae852SEric Dumazet 		if (get_user(len, optlen))
3790cd8ae852SEric Dumazet 			return -EFAULT;
3791cd8ae852SEric Dumazet 
3792cd8ae852SEric Dumazet 		lock_sock(sk);
3793cd8ae852SEric Dumazet 		if (tp->saved_syn) {
379470a217f1SMartin KaFai Lau 			if (len < tcp_saved_syn_len(tp->saved_syn)) {
379570a217f1SMartin KaFai Lau 				if (put_user(tcp_saved_syn_len(tp->saved_syn),
379670a217f1SMartin KaFai Lau 					     optlen)) {
3797aea0929eSEric B Munson 					release_sock(sk);
3798aea0929eSEric B Munson 					return -EFAULT;
3799aea0929eSEric B Munson 				}
3800aea0929eSEric B Munson 				release_sock(sk);
3801aea0929eSEric B Munson 				return -EINVAL;
3802aea0929eSEric B Munson 			}
380370a217f1SMartin KaFai Lau 			len = tcp_saved_syn_len(tp->saved_syn);
3804cd8ae852SEric Dumazet 			if (put_user(len, optlen)) {
3805cd8ae852SEric Dumazet 				release_sock(sk);
3806cd8ae852SEric Dumazet 				return -EFAULT;
3807cd8ae852SEric Dumazet 			}
380870a217f1SMartin KaFai Lau 			if (copy_to_user(optval, tp->saved_syn->data, len)) {
3809cd8ae852SEric Dumazet 				release_sock(sk);
3810cd8ae852SEric Dumazet 				return -EFAULT;
3811cd8ae852SEric Dumazet 			}
3812cd8ae852SEric Dumazet 			tcp_saved_syn_free(tp);
3813cd8ae852SEric Dumazet 			release_sock(sk);
3814cd8ae852SEric Dumazet 		} else {
3815cd8ae852SEric Dumazet 			release_sock(sk);
3816cd8ae852SEric Dumazet 			len = 0;
3817cd8ae852SEric Dumazet 			if (put_user(len, optlen))
3818cd8ae852SEric Dumazet 				return -EFAULT;
3819cd8ae852SEric Dumazet 		}
3820cd8ae852SEric Dumazet 		return 0;
3821cd8ae852SEric Dumazet 	}
382205255b82SEric Dumazet #ifdef CONFIG_MMU
382305255b82SEric Dumazet 	case TCP_ZEROCOPY_RECEIVE: {
382405255b82SEric Dumazet 		struct tcp_zerocopy_receive zc;
382505255b82SEric Dumazet 		int err;
382605255b82SEric Dumazet 
382705255b82SEric Dumazet 		if (get_user(len, optlen))
382805255b82SEric Dumazet 			return -EFAULT;
3829c8856c05SArjun Roy 		if (len < offsetofend(struct tcp_zerocopy_receive, length))
383005255b82SEric Dumazet 			return -EINVAL;
38310b7f41f6SArjun Roy 		if (len > sizeof(zc)) {
3832c8856c05SArjun Roy 			len = sizeof(zc);
38330b7f41f6SArjun Roy 			if (put_user(len, optlen))
38340b7f41f6SArjun Roy 				return -EFAULT;
38350b7f41f6SArjun Roy 		}
383605255b82SEric Dumazet 		if (copy_from_user(&zc, optval, len))
383705255b82SEric Dumazet 			return -EFAULT;
383805255b82SEric Dumazet 		lock_sock(sk);
383905255b82SEric Dumazet 		err = tcp_zerocopy_receive(sk, &zc);
384005255b82SEric Dumazet 		release_sock(sk);
384133946518SArjun Roy 		if (len == sizeof(zc))
384233946518SArjun Roy 			goto zerocopy_rcv_sk_err;
3843c8856c05SArjun Roy 		switch (len) {
384433946518SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, err):
384533946518SArjun Roy 			goto zerocopy_rcv_sk_err;
3846c8856c05SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, inq):
3847c8856c05SArjun Roy 			goto zerocopy_rcv_inq;
3848c8856c05SArjun Roy 		case offsetofend(struct tcp_zerocopy_receive, length):
3849c8856c05SArjun Roy 		default:
3850c8856c05SArjun Roy 			goto zerocopy_rcv_out;
3851c8856c05SArjun Roy 		}
385233946518SArjun Roy zerocopy_rcv_sk_err:
385333946518SArjun Roy 		if (!err)
385433946518SArjun Roy 			zc.err = sock_error(sk);
3855c8856c05SArjun Roy zerocopy_rcv_inq:
3856c8856c05SArjun Roy 		zc.inq = tcp_inq_hint(sk);
3857c8856c05SArjun Roy zerocopy_rcv_out:
385805255b82SEric Dumazet 		if (!err && copy_to_user(optval, &zc, len))
385905255b82SEric Dumazet 			err = -EFAULT;
386005255b82SEric Dumazet 		return err;
386105255b82SEric Dumazet 	}
386205255b82SEric Dumazet #endif
38631da177e4SLinus Torvalds 	default:
38641da177e4SLinus Torvalds 		return -ENOPROTOOPT;
38653ff50b79SStephen Hemminger 	}
38661da177e4SLinus Torvalds 
38671da177e4SLinus Torvalds 	if (put_user(len, optlen))
38681da177e4SLinus Torvalds 		return -EFAULT;
38691da177e4SLinus Torvalds 	if (copy_to_user(optval, &val, len))
38701da177e4SLinus Torvalds 		return -EFAULT;
38711da177e4SLinus Torvalds 	return 0;
38721da177e4SLinus Torvalds }
38731da177e4SLinus Torvalds 
38743fdadf7dSDmitry Mishin int tcp_getsockopt(struct sock *sk, int level, int optname, char __user *optval,
38753fdadf7dSDmitry Mishin 		   int __user *optlen)
38763fdadf7dSDmitry Mishin {
38773fdadf7dSDmitry Mishin 	struct inet_connection_sock *icsk = inet_csk(sk);
38783fdadf7dSDmitry Mishin 
38793fdadf7dSDmitry Mishin 	if (level != SOL_TCP)
38803fdadf7dSDmitry Mishin 		return icsk->icsk_af_ops->getsockopt(sk, level, optname,
38813fdadf7dSDmitry Mishin 						     optval, optlen);
38823fdadf7dSDmitry Mishin 	return do_tcp_getsockopt(sk, level, optname, optval, optlen);
38833fdadf7dSDmitry Mishin }
38844bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_getsockopt);
38853fdadf7dSDmitry Mishin 
3886cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG
3887349ce993SEric Dumazet static DEFINE_PER_CPU(struct tcp_md5sig_pool, tcp_md5sig_pool);
388871cea17eSEric Dumazet static DEFINE_MUTEX(tcp_md5sig_mutex);
3889349ce993SEric Dumazet static bool tcp_md5sig_pool_populated = false;
3890cfb6eeb4SYOSHIFUJI Hideaki 
389171cea17eSEric Dumazet static void __tcp_alloc_md5sig_pool(void)
3892cfb6eeb4SYOSHIFUJI Hideaki {
3893cf80e0e4SHerbert Xu 	struct crypto_ahash *hash;
3894cfb6eeb4SYOSHIFUJI Hideaki 	int cpu;
3895cfb6eeb4SYOSHIFUJI Hideaki 
3896cf80e0e4SHerbert Xu 	hash = crypto_alloc_ahash("md5", 0, CRYPTO_ALG_ASYNC);
38971eea84b7SInsu Yun 	if (IS_ERR(hash))
3898349ce993SEric Dumazet 		return;
3899cf80e0e4SHerbert Xu 
3900cf80e0e4SHerbert Xu 	for_each_possible_cpu(cpu) {
390119689e38SEric Dumazet 		void *scratch = per_cpu(tcp_md5sig_pool, cpu).scratch;
3902cf80e0e4SHerbert Xu 		struct ahash_request *req;
3903cf80e0e4SHerbert Xu 
390419689e38SEric Dumazet 		if (!scratch) {
390519689e38SEric Dumazet 			scratch = kmalloc_node(sizeof(union tcp_md5sum_block) +
390619689e38SEric Dumazet 					       sizeof(struct tcphdr),
390719689e38SEric Dumazet 					       GFP_KERNEL,
390819689e38SEric Dumazet 					       cpu_to_node(cpu));
390919689e38SEric Dumazet 			if (!scratch)
391019689e38SEric Dumazet 				return;
391119689e38SEric Dumazet 			per_cpu(tcp_md5sig_pool, cpu).scratch = scratch;
391219689e38SEric Dumazet 		}
3913cf80e0e4SHerbert Xu 		if (per_cpu(tcp_md5sig_pool, cpu).md5_req)
3914cf80e0e4SHerbert Xu 			continue;
3915cf80e0e4SHerbert Xu 
3916cf80e0e4SHerbert Xu 		req = ahash_request_alloc(hash, GFP_KERNEL);
3917cf80e0e4SHerbert Xu 		if (!req)
3918cf80e0e4SHerbert Xu 			return;
3919cf80e0e4SHerbert Xu 
3920cf80e0e4SHerbert Xu 		ahash_request_set_callback(req, 0, NULL, NULL);
3921cf80e0e4SHerbert Xu 
3922cf80e0e4SHerbert Xu 		per_cpu(tcp_md5sig_pool, cpu).md5_req = req;
3923349ce993SEric Dumazet 	}
3924349ce993SEric Dumazet 	/* before setting tcp_md5sig_pool_populated, we must commit all writes
3925349ce993SEric Dumazet 	 * to memory. See smp_rmb() in tcp_get_md5sig_pool()
392671cea17eSEric Dumazet 	 */
392771cea17eSEric Dumazet 	smp_wmb();
3928349ce993SEric Dumazet 	tcp_md5sig_pool_populated = true;
3929cfb6eeb4SYOSHIFUJI Hideaki }
3930cfb6eeb4SYOSHIFUJI Hideaki 
393171cea17eSEric Dumazet bool tcp_alloc_md5sig_pool(void)
3932cfb6eeb4SYOSHIFUJI Hideaki {
3933349ce993SEric Dumazet 	if (unlikely(!tcp_md5sig_pool_populated)) {
393471cea17eSEric Dumazet 		mutex_lock(&tcp_md5sig_mutex);
3935cfb6eeb4SYOSHIFUJI Hideaki 
39366015c71eSEric Dumazet 		if (!tcp_md5sig_pool_populated) {
393771cea17eSEric Dumazet 			__tcp_alloc_md5sig_pool();
39386015c71eSEric Dumazet 			if (tcp_md5sig_pool_populated)
3939921f9a0fSEric Dumazet 				static_branch_inc(&tcp_md5_needed);
39406015c71eSEric Dumazet 		}
3941cfb6eeb4SYOSHIFUJI Hideaki 
394271cea17eSEric Dumazet 		mutex_unlock(&tcp_md5sig_mutex);
3943cfb6eeb4SYOSHIFUJI Hideaki 	}
3944349ce993SEric Dumazet 	return tcp_md5sig_pool_populated;
3945cfb6eeb4SYOSHIFUJI Hideaki }
3946cfb6eeb4SYOSHIFUJI Hideaki EXPORT_SYMBOL(tcp_alloc_md5sig_pool);
3947cfb6eeb4SYOSHIFUJI Hideaki 
394835790c04SEric Dumazet 
394935790c04SEric Dumazet /**
395035790c04SEric Dumazet  *	tcp_get_md5sig_pool - get md5sig_pool for this user
395135790c04SEric Dumazet  *
395235790c04SEric Dumazet  *	We use percpu structure, so if we succeed, we exit with preemption
395335790c04SEric Dumazet  *	and BH disabled, to make sure another thread or softirq handling
395435790c04SEric Dumazet  *	wont try to get same context.
395535790c04SEric Dumazet  */
395635790c04SEric Dumazet struct tcp_md5sig_pool *tcp_get_md5sig_pool(void)
3957cfb6eeb4SYOSHIFUJI Hideaki {
395835790c04SEric Dumazet 	local_bh_disable();
395935790c04SEric Dumazet 
3960349ce993SEric Dumazet 	if (tcp_md5sig_pool_populated) {
3961349ce993SEric Dumazet 		/* coupled with smp_wmb() in __tcp_alloc_md5sig_pool() */
3962349ce993SEric Dumazet 		smp_rmb();
3963349ce993SEric Dumazet 		return this_cpu_ptr(&tcp_md5sig_pool);
3964349ce993SEric Dumazet 	}
396535790c04SEric Dumazet 	local_bh_enable();
396635790c04SEric Dumazet 	return NULL;
3967cfb6eeb4SYOSHIFUJI Hideaki }
396835790c04SEric Dumazet EXPORT_SYMBOL(tcp_get_md5sig_pool);
3969cfb6eeb4SYOSHIFUJI Hideaki 
397049a72dfbSAdam Langley int tcp_md5_hash_skb_data(struct tcp_md5sig_pool *hp,
3971cf533ea5SEric Dumazet 			  const struct sk_buff *skb, unsigned int header_len)
397249a72dfbSAdam Langley {
397349a72dfbSAdam Langley 	struct scatterlist sg;
397449a72dfbSAdam Langley 	const struct tcphdr *tp = tcp_hdr(skb);
3975cf80e0e4SHerbert Xu 	struct ahash_request *req = hp->md5_req;
397695c96174SEric Dumazet 	unsigned int i;
397795c96174SEric Dumazet 	const unsigned int head_data_len = skb_headlen(skb) > header_len ?
397849a72dfbSAdam Langley 					   skb_headlen(skb) - header_len : 0;
397949a72dfbSAdam Langley 	const struct skb_shared_info *shi = skb_shinfo(skb);
3980d7fd1b57SEric Dumazet 	struct sk_buff *frag_iter;
398149a72dfbSAdam Langley 
398249a72dfbSAdam Langley 	sg_init_table(&sg, 1);
398349a72dfbSAdam Langley 
398449a72dfbSAdam Langley 	sg_set_buf(&sg, ((u8 *) tp) + header_len, head_data_len);
3985cf80e0e4SHerbert Xu 	ahash_request_set_crypt(req, &sg, NULL, head_data_len);
3986cf80e0e4SHerbert Xu 	if (crypto_ahash_update(req))
398749a72dfbSAdam Langley 		return 1;
398849a72dfbSAdam Langley 
398949a72dfbSAdam Langley 	for (i = 0; i < shi->nr_frags; ++i) {
3990d8e18a51SMatthew Wilcox (Oracle) 		const skb_frag_t *f = &shi->frags[i];
3991b54c9d5bSJonathan Lemon 		unsigned int offset = skb_frag_off(f);
399254d27fcbSEric Dumazet 		struct page *page = skb_frag_page(f) + (offset >> PAGE_SHIFT);
399354d27fcbSEric Dumazet 
399454d27fcbSEric Dumazet 		sg_set_page(&sg, page, skb_frag_size(f),
399554d27fcbSEric Dumazet 			    offset_in_page(offset));
3996cf80e0e4SHerbert Xu 		ahash_request_set_crypt(req, &sg, NULL, skb_frag_size(f));
3997cf80e0e4SHerbert Xu 		if (crypto_ahash_update(req))
399849a72dfbSAdam Langley 			return 1;
399949a72dfbSAdam Langley 	}
400049a72dfbSAdam Langley 
4001d7fd1b57SEric Dumazet 	skb_walk_frags(skb, frag_iter)
4002d7fd1b57SEric Dumazet 		if (tcp_md5_hash_skb_data(hp, frag_iter, 0))
4003d7fd1b57SEric Dumazet 			return 1;
4004d7fd1b57SEric Dumazet 
400549a72dfbSAdam Langley 	return 0;
400649a72dfbSAdam Langley }
400749a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_skb_data);
400849a72dfbSAdam Langley 
4009cf533ea5SEric Dumazet int tcp_md5_hash_key(struct tcp_md5sig_pool *hp, const struct tcp_md5sig_key *key)
401049a72dfbSAdam Langley {
4011e6ced831SEric Dumazet 	u8 keylen = READ_ONCE(key->keylen); /* paired with WRITE_ONCE() in tcp_md5_do_add */
401249a72dfbSAdam Langley 	struct scatterlist sg;
401349a72dfbSAdam Langley 
40146a2febecSEric Dumazet 	sg_init_one(&sg, key->key, keylen);
40156a2febecSEric Dumazet 	ahash_request_set_crypt(hp->md5_req, &sg, NULL, keylen);
4016e6ced831SEric Dumazet 
4017e6ced831SEric Dumazet 	/* We use data_race() because tcp_md5_do_add() might change key->key under us */
4018e6ced831SEric Dumazet 	return data_race(crypto_ahash_update(hp->md5_req));
401949a72dfbSAdam Langley }
402049a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_key);
402149a72dfbSAdam Langley 
4022cfb6eeb4SYOSHIFUJI Hideaki #endif
4023cfb6eeb4SYOSHIFUJI Hideaki 
40244ac02babSAndi Kleen void tcp_done(struct sock *sk)
40254ac02babSAndi Kleen {
4026d983ea6fSEric Dumazet 	struct request_sock *req;
40278336886fSJerry Chu 
4028cab209e5SEric Dumazet 	/* We might be called with a new socket, after
4029cab209e5SEric Dumazet 	 * inet_csk_prepare_forced_close() has been called
4030cab209e5SEric Dumazet 	 * so we can not use lockdep_sock_is_held(sk)
4031cab209e5SEric Dumazet 	 */
4032cab209e5SEric Dumazet 	req = rcu_dereference_protected(tcp_sk(sk)->fastopen_rsk, 1);
40334ac02babSAndi Kleen 
40344ac02babSAndi Kleen 	if (sk->sk_state == TCP_SYN_SENT || sk->sk_state == TCP_SYN_RECV)
4035c10d9310SEric Dumazet 		TCP_INC_STATS(sock_net(sk), TCP_MIB_ATTEMPTFAILS);
40364ac02babSAndi Kleen 
40374ac02babSAndi Kleen 	tcp_set_state(sk, TCP_CLOSE);
40384ac02babSAndi Kleen 	tcp_clear_xmit_timers(sk);
403900db4124SIan Morris 	if (req)
40408336886fSJerry Chu 		reqsk_fastopen_remove(sk, req, false);
40414ac02babSAndi Kleen 
40424ac02babSAndi Kleen 	sk->sk_shutdown = SHUTDOWN_MASK;
40434ac02babSAndi Kleen 
40444ac02babSAndi Kleen 	if (!sock_flag(sk, SOCK_DEAD))
40454ac02babSAndi Kleen 		sk->sk_state_change(sk);
40464ac02babSAndi Kleen 	else
40474ac02babSAndi Kleen 		inet_csk_destroy_sock(sk);
40484ac02babSAndi Kleen }
40494ac02babSAndi Kleen EXPORT_SYMBOL_GPL(tcp_done);
40504ac02babSAndi Kleen 
4051c1e64e29SLorenzo Colitti int tcp_abort(struct sock *sk, int err)
4052c1e64e29SLorenzo Colitti {
4053c1e64e29SLorenzo Colitti 	if (!sk_fullsock(sk)) {
405407f6f4a3SEric Dumazet 		if (sk->sk_state == TCP_NEW_SYN_RECV) {
405507f6f4a3SEric Dumazet 			struct request_sock *req = inet_reqsk(sk);
405607f6f4a3SEric Dumazet 
405707f6f4a3SEric Dumazet 			local_bh_disable();
4058acc2cf4eSLorenzo Colitti 			inet_csk_reqsk_queue_drop(req->rsk_listener, req);
405907f6f4a3SEric Dumazet 			local_bh_enable();
406007f6f4a3SEric Dumazet 			return 0;
406107f6f4a3SEric Dumazet 		}
4062c1e64e29SLorenzo Colitti 		return -EOPNOTSUPP;
4063c1e64e29SLorenzo Colitti 	}
4064c1e64e29SLorenzo Colitti 
4065c1e64e29SLorenzo Colitti 	/* Don't race with userspace socket closes such as tcp_close. */
4066c1e64e29SLorenzo Colitti 	lock_sock(sk);
4067c1e64e29SLorenzo Colitti 
40682010b93eSLorenzo Colitti 	if (sk->sk_state == TCP_LISTEN) {
40692010b93eSLorenzo Colitti 		tcp_set_state(sk, TCP_CLOSE);
40702010b93eSLorenzo Colitti 		inet_csk_listen_stop(sk);
40712010b93eSLorenzo Colitti 	}
40722010b93eSLorenzo Colitti 
4073c1e64e29SLorenzo Colitti 	/* Don't race with BH socket closes such as inet_csk_listen_stop. */
4074c1e64e29SLorenzo Colitti 	local_bh_disable();
4075c1e64e29SLorenzo Colitti 	bh_lock_sock(sk);
4076c1e64e29SLorenzo Colitti 
4077c1e64e29SLorenzo Colitti 	if (!sock_flag(sk, SOCK_DEAD)) {
4078c1e64e29SLorenzo Colitti 		sk->sk_err = err;
4079c1e64e29SLorenzo Colitti 		/* This barrier is coupled with smp_rmb() in tcp_poll() */
4080c1e64e29SLorenzo Colitti 		smp_wmb();
4081c1e64e29SLorenzo Colitti 		sk->sk_error_report(sk);
4082c1e64e29SLorenzo Colitti 		if (tcp_need_reset(sk->sk_state))
4083c1e64e29SLorenzo Colitti 			tcp_send_active_reset(sk, GFP_ATOMIC);
4084c1e64e29SLorenzo Colitti 		tcp_done(sk);
4085c1e64e29SLorenzo Colitti 	}
4086c1e64e29SLorenzo Colitti 
4087c1e64e29SLorenzo Colitti 	bh_unlock_sock(sk);
4088c1e64e29SLorenzo Colitti 	local_bh_enable();
4089e05836acSSoheil Hassas Yeganeh 	tcp_write_queue_purge(sk);
4090c1e64e29SLorenzo Colitti 	release_sock(sk);
4091c1e64e29SLorenzo Colitti 	return 0;
4092c1e64e29SLorenzo Colitti }
4093c1e64e29SLorenzo Colitti EXPORT_SYMBOL_GPL(tcp_abort);
4094c1e64e29SLorenzo Colitti 
40955f8ef48dSStephen Hemminger extern struct tcp_congestion_ops tcp_reno;
40961da177e4SLinus Torvalds 
40971da177e4SLinus Torvalds static __initdata unsigned long thash_entries;
40981da177e4SLinus Torvalds static int __init set_thash_entries(char *str)
40991da177e4SLinus Torvalds {
4100413c27d8SEldad Zack 	ssize_t ret;
4101413c27d8SEldad Zack 
41021da177e4SLinus Torvalds 	if (!str)
41031da177e4SLinus Torvalds 		return 0;
4104413c27d8SEldad Zack 
4105413c27d8SEldad Zack 	ret = kstrtoul(str, 0, &thash_entries);
4106413c27d8SEldad Zack 	if (ret)
4107413c27d8SEldad Zack 		return 0;
4108413c27d8SEldad Zack 
41091da177e4SLinus Torvalds 	return 1;
41101da177e4SLinus Torvalds }
41111da177e4SLinus Torvalds __setup("thash_entries=", set_thash_entries);
41121da177e4SLinus Torvalds 
411347d7a88cSFabian Frederick static void __init tcp_init_mem(void)
41144acb4190SGlauber Costa {
4115b66e91ccSEric Dumazet 	unsigned long limit = nr_free_buffer_pages() / 16;
4116b66e91ccSEric Dumazet 
41174acb4190SGlauber Costa 	limit = max(limit, 128UL);
4118b66e91ccSEric Dumazet 	sysctl_tcp_mem[0] = limit / 4 * 3;		/* 4.68 % */
4119b66e91ccSEric Dumazet 	sysctl_tcp_mem[1] = limit;			/* 6.25 % */
4120b66e91ccSEric Dumazet 	sysctl_tcp_mem[2] = sysctl_tcp_mem[0] * 2;	/* 9.37 % */
41214acb4190SGlauber Costa }
41224acb4190SGlauber Costa 
41231da177e4SLinus Torvalds void __init tcp_init(void)
41241da177e4SLinus Torvalds {
4125b49960a0SEric Dumazet 	int max_rshare, max_wshare, cnt;
4126b2d3ea4aSEric Dumazet 	unsigned long limit;
4127074b8517SDimitri Sivanich 	unsigned int i;
41281da177e4SLinus Torvalds 
41293b4929f6SEric Dumazet 	BUILD_BUG_ON(TCP_MIN_SND_MSS <= MAX_TCP_OPTION_SPACE);
4130b2d3ea4aSEric Dumazet 	BUILD_BUG_ON(sizeof(struct tcp_skb_cb) >
4131c593642cSPankaj Bharadiya 		     sizeof_field(struct sk_buff, cb));
41321da177e4SLinus Torvalds 
4133908c7f19STejun Heo 	percpu_counter_init(&tcp_sockets_allocated, 0, GFP_KERNEL);
4134908c7f19STejun Heo 	percpu_counter_init(&tcp_orphan_count, 0, GFP_KERNEL);
41351946e672SHaishuang Yan 	inet_hashinfo_init(&tcp_hashinfo);
413627da6d37SMartin KaFai Lau 	inet_hashinfo2_init(&tcp_hashinfo, "tcp_listen_portaddr_hash",
413727da6d37SMartin KaFai Lau 			    thash_entries, 21,  /* one slot per 2 MB*/
413827da6d37SMartin KaFai Lau 			    0, 64 * 1024);
41396e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.bind_bucket_cachep =
41406e04e021SArnaldo Carvalho de Melo 		kmem_cache_create("tcp_bind_bucket",
41416e04e021SArnaldo Carvalho de Melo 				  sizeof(struct inet_bind_bucket), 0,
414220c2df83SPaul Mundt 				  SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL);
41431da177e4SLinus Torvalds 
41441da177e4SLinus Torvalds 	/* Size and allocate the main established and bind bucket
41451da177e4SLinus Torvalds 	 * hash tables.
41461da177e4SLinus Torvalds 	 *
41471da177e4SLinus Torvalds 	 * The methodology is similar to that of the buffer cache.
41481da177e4SLinus Torvalds 	 */
41496e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.ehash =
41501da177e4SLinus Torvalds 		alloc_large_system_hash("TCP established",
41510f7ff927SArnaldo Carvalho de Melo 					sizeof(struct inet_ehash_bucket),
41521da177e4SLinus Torvalds 					thash_entries,
4153fd90b29dSEric Dumazet 					17, /* one slot per 128 KB of memory */
41549e950efaSJohn Heffner 					0,
41551da177e4SLinus Torvalds 					NULL,
4156f373b53bSEric Dumazet 					&tcp_hashinfo.ehash_mask,
415731fe62b9STim Bird 					0,
41580ccfe618SJean Delvare 					thash_entries ? 0 : 512 * 1024);
415905dbc7b5SEric Dumazet 	for (i = 0; i <= tcp_hashinfo.ehash_mask; i++)
41603ab5aee7SEric Dumazet 		INIT_HLIST_NULLS_HEAD(&tcp_hashinfo.ehash[i].chain, i);
416105dbc7b5SEric Dumazet 
4162230140cfSEric Dumazet 	if (inet_ehash_locks_alloc(&tcp_hashinfo))
4163230140cfSEric Dumazet 		panic("TCP: failed to alloc ehash_locks");
41646e04e021SArnaldo Carvalho de Melo 	tcp_hashinfo.bhash =
41651da177e4SLinus Torvalds 		alloc_large_system_hash("TCP bind",
41660f7ff927SArnaldo Carvalho de Melo 					sizeof(struct inet_bind_hashbucket),
4167f373b53bSEric Dumazet 					tcp_hashinfo.ehash_mask + 1,
4168fd90b29dSEric Dumazet 					17, /* one slot per 128 KB of memory */
41699e950efaSJohn Heffner 					0,
41706e04e021SArnaldo Carvalho de Melo 					&tcp_hashinfo.bhash_size,
41711da177e4SLinus Torvalds 					NULL,
417231fe62b9STim Bird 					0,
41731da177e4SLinus Torvalds 					64 * 1024);
4174074b8517SDimitri Sivanich 	tcp_hashinfo.bhash_size = 1U << tcp_hashinfo.bhash_size;
41756e04e021SArnaldo Carvalho de Melo 	for (i = 0; i < tcp_hashinfo.bhash_size; i++) {
41766e04e021SArnaldo Carvalho de Melo 		spin_lock_init(&tcp_hashinfo.bhash[i].lock);
41776e04e021SArnaldo Carvalho de Melo 		INIT_HLIST_HEAD(&tcp_hashinfo.bhash[i].chain);
41781da177e4SLinus Torvalds 	}
41791da177e4SLinus Torvalds 
4180c5ed63d6SEric Dumazet 
4181c5ed63d6SEric Dumazet 	cnt = tcp_hashinfo.ehash_mask + 1;
4182c5ed63d6SEric Dumazet 	sysctl_tcp_max_orphans = cnt / 2;
41831da177e4SLinus Torvalds 
4184a4fe34bfSEric W. Biederman 	tcp_init_mem();
4185c43b874dSJason Wang 	/* Set per-socket limits to no more than 1/128 the pressure threshold */
41865fb84b14SEric Dumazet 	limit = nr_free_buffer_pages() << (PAGE_SHIFT - 7);
4187b49960a0SEric Dumazet 	max_wshare = min(4UL*1024*1024, limit);
4188b49960a0SEric Dumazet 	max_rshare = min(6UL*1024*1024, limit);
41897b4f4b5eSJohn Heffner 
4190356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[0] = SK_MEM_QUANTUM;
4191356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[1] = 16*1024;
4192356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_wmem[2] = max(64*1024, max_wshare);
41937b4f4b5eSJohn Heffner 
4194356d1833SEric Dumazet 	init_net.ipv4.sysctl_tcp_rmem[0] = SK_MEM_QUANTUM;
4195a337531bSYuchung Cheng 	init_net.ipv4.sysctl_tcp_rmem[1] = 131072;
4196a337531bSYuchung Cheng 	init_net.ipv4.sysctl_tcp_rmem[2] = max(131072, max_rshare);
41971da177e4SLinus Torvalds 
4198afd46503SJoe Perches 	pr_info("Hash tables configured (established %u bind %u)\n",
4199f373b53bSEric Dumazet 		tcp_hashinfo.ehash_mask + 1, tcp_hashinfo.bhash_size);
4200317a76f9SStephen Hemminger 
42011946e672SHaishuang Yan 	tcp_v4_init();
420251c5d0c4SDavid S. Miller 	tcp_metrics_init();
420355d8694fSFlorian Westphal 	BUG_ON(tcp_register_congestion_control(&tcp_reno) != 0);
420446d3ceabSEric Dumazet 	tcp_tasklet_init();
4205f870fa0bSMat Martineau 	mptcp_init();
42061da177e4SLinus Torvalds }
4207