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