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> 2741da177e4SLinus Torvalds #include <net/xfrm.h> 2751da177e4SLinus Torvalds #include <net/ip.h> 2769c55e01cSJens Axboe #include <net/sock.h> 2771da177e4SLinus Torvalds 2787c0f6ba6SLinus Torvalds #include <linux/uaccess.h> 2791da177e4SLinus Torvalds #include <asm/ioctls.h> 280076bb0c8SEliezer Tamir #include <net/busy_poll.h> 2811da177e4SLinus Torvalds 282dd24c001SEric Dumazet struct percpu_counter tcp_orphan_count; 2830a5578cfSArnaldo Carvalho de Melo EXPORT_SYMBOL_GPL(tcp_orphan_count); 2840a5578cfSArnaldo Carvalho de Melo 285a4fe34bfSEric W. Biederman long sysctl_tcp_mem[3] __read_mostly; 286a4fe34bfSEric W. Biederman EXPORT_SYMBOL(sysctl_tcp_mem); 2871da177e4SLinus Torvalds 2888d987e5cSEric Dumazet atomic_long_t tcp_memory_allocated; /* Current allocated memory. */ 2891da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_memory_allocated); 2901748376bSEric Dumazet 29160e2a778SUrsula Braun #if IS_ENABLED(CONFIG_SMC) 29260e2a778SUrsula Braun DEFINE_STATIC_KEY_FALSE(tcp_have_smc); 29360e2a778SUrsula Braun EXPORT_SYMBOL(tcp_have_smc); 29460e2a778SUrsula Braun #endif 29560e2a778SUrsula Braun 2961748376bSEric Dumazet /* 2971748376bSEric Dumazet * Current number of TCP sockets. 2981748376bSEric Dumazet */ 2991748376bSEric Dumazet struct percpu_counter tcp_sockets_allocated; 3001da177e4SLinus Torvalds EXPORT_SYMBOL(tcp_sockets_allocated); 3011da177e4SLinus Torvalds 3021da177e4SLinus Torvalds /* 3039c55e01cSJens Axboe * TCP splice context 3049c55e01cSJens Axboe */ 3059c55e01cSJens Axboe struct tcp_splice_state { 3069c55e01cSJens Axboe struct pipe_inode_info *pipe; 3079c55e01cSJens Axboe size_t len; 3089c55e01cSJens Axboe unsigned int flags; 3099c55e01cSJens Axboe }; 3109c55e01cSJens Axboe 3119c55e01cSJens Axboe /* 3121da177e4SLinus Torvalds * Pressure flag: try to collapse. 3131da177e4SLinus Torvalds * Technical note: it is used by multiple contexts non atomically. 3143ab224beSHideo Aoki * All the __sk_mem_schedule() is of this nature: accounting 3151da177e4SLinus Torvalds * is strict, actions are advisory and have some latency. 3161da177e4SLinus Torvalds */ 31706044751SEric Dumazet unsigned long tcp_memory_pressure __read_mostly; 31806044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_memory_pressure); 3191da177e4SLinus Torvalds 3205c52ba17SPavel Emelyanov void tcp_enter_memory_pressure(struct sock *sk) 3211da177e4SLinus Torvalds { 32206044751SEric Dumazet unsigned long val; 32306044751SEric Dumazet 32406044751SEric Dumazet if (tcp_memory_pressure) 32506044751SEric Dumazet return; 32606044751SEric Dumazet val = jiffies; 32706044751SEric Dumazet 32806044751SEric Dumazet if (!val) 32906044751SEric Dumazet val--; 33006044751SEric Dumazet if (!cmpxchg(&tcp_memory_pressure, 0, val)) 3314e673444SPavel Emelyanov NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURES); 3321da177e4SLinus Torvalds } 33306044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_enter_memory_pressure); 33406044751SEric Dumazet 33506044751SEric Dumazet void tcp_leave_memory_pressure(struct sock *sk) 33606044751SEric Dumazet { 33706044751SEric Dumazet unsigned long val; 33806044751SEric Dumazet 33906044751SEric Dumazet if (!tcp_memory_pressure) 34006044751SEric Dumazet return; 34106044751SEric Dumazet val = xchg(&tcp_memory_pressure, 0); 34206044751SEric Dumazet if (val) 34306044751SEric Dumazet NET_ADD_STATS(sock_net(sk), LINUX_MIB_TCPMEMORYPRESSURESCHRONO, 34406044751SEric Dumazet jiffies_to_msecs(jiffies - val)); 3451da177e4SLinus Torvalds } 34606044751SEric Dumazet EXPORT_SYMBOL_GPL(tcp_leave_memory_pressure); 3471da177e4SLinus Torvalds 348b103cf34SJulian Anastasov /* Convert seconds to retransmits based on initial and max timeout */ 349b103cf34SJulian Anastasov static u8 secs_to_retrans(int seconds, int timeout, int rto_max) 350b103cf34SJulian Anastasov { 351b103cf34SJulian Anastasov u8 res = 0; 352b103cf34SJulian Anastasov 353b103cf34SJulian Anastasov if (seconds > 0) { 354b103cf34SJulian Anastasov int period = timeout; 355b103cf34SJulian Anastasov 356b103cf34SJulian Anastasov res = 1; 357b103cf34SJulian Anastasov while (seconds > period && res < 255) { 358b103cf34SJulian Anastasov res++; 359b103cf34SJulian Anastasov timeout <<= 1; 360b103cf34SJulian Anastasov if (timeout > rto_max) 361b103cf34SJulian Anastasov timeout = rto_max; 362b103cf34SJulian Anastasov period += timeout; 363b103cf34SJulian Anastasov } 364b103cf34SJulian Anastasov } 365b103cf34SJulian Anastasov return res; 366b103cf34SJulian Anastasov } 367b103cf34SJulian Anastasov 368b103cf34SJulian Anastasov /* Convert retransmits to seconds based on initial and max timeout */ 369b103cf34SJulian Anastasov static int retrans_to_secs(u8 retrans, int timeout, int rto_max) 370b103cf34SJulian Anastasov { 371b103cf34SJulian Anastasov int period = 0; 372b103cf34SJulian Anastasov 373b103cf34SJulian Anastasov if (retrans > 0) { 374b103cf34SJulian Anastasov period = timeout; 375b103cf34SJulian Anastasov while (--retrans) { 376b103cf34SJulian Anastasov timeout <<= 1; 377b103cf34SJulian Anastasov if (timeout > rto_max) 378b103cf34SJulian Anastasov timeout = rto_max; 379b103cf34SJulian Anastasov period += timeout; 380b103cf34SJulian Anastasov } 381b103cf34SJulian Anastasov } 382b103cf34SJulian Anastasov return period; 383b103cf34SJulian Anastasov } 384b103cf34SJulian Anastasov 3850263598cSWei Wang static u64 tcp_compute_delivery_rate(const struct tcp_sock *tp) 3860263598cSWei Wang { 3870263598cSWei Wang u32 rate = READ_ONCE(tp->rate_delivered); 3880263598cSWei Wang u32 intv = READ_ONCE(tp->rate_interval_us); 3890263598cSWei Wang u64 rate64 = 0; 3900263598cSWei Wang 3910263598cSWei Wang if (rate && intv) { 3920263598cSWei Wang rate64 = (u64)rate * tp->mss_cache * USEC_PER_SEC; 3930263598cSWei Wang do_div(rate64, intv); 3940263598cSWei Wang } 3950263598cSWei Wang return rate64; 3960263598cSWei Wang } 3970263598cSWei Wang 398900f65d3SNeal Cardwell /* Address-family independent initialization for a tcp_sock. 399900f65d3SNeal Cardwell * 400900f65d3SNeal Cardwell * NOTE: A lot of things set to zero explicitly by call to 401900f65d3SNeal Cardwell * sk_alloc() so need not be done here. 402900f65d3SNeal Cardwell */ 403900f65d3SNeal Cardwell void tcp_init_sock(struct sock *sk) 404900f65d3SNeal Cardwell { 405900f65d3SNeal Cardwell struct inet_connection_sock *icsk = inet_csk(sk); 406900f65d3SNeal Cardwell struct tcp_sock *tp = tcp_sk(sk); 407900f65d3SNeal Cardwell 4089f5afeaeSYaogong Wang tp->out_of_order_queue = RB_ROOT; 40975c119afSEric Dumazet sk->tcp_rtx_queue = RB_ROOT; 410900f65d3SNeal Cardwell tcp_init_xmit_timers(sk); 41146d3ceabSEric Dumazet INIT_LIST_HEAD(&tp->tsq_node); 412e2080072SEric Dumazet INIT_LIST_HEAD(&tp->tsorted_sent_queue); 413900f65d3SNeal Cardwell 414900f65d3SNeal Cardwell icsk->icsk_rto = TCP_TIMEOUT_INIT; 415740b0f18SEric Dumazet tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT); 416ac9517fcSEric Dumazet minmax_reset(&tp->rtt_min, tcp_jiffies32, ~0U); 417900f65d3SNeal Cardwell 418900f65d3SNeal Cardwell /* So many TCP implementations out there (incorrectly) count the 419900f65d3SNeal Cardwell * initial SYN frame in their delayed-ACK and congestion control 420900f65d3SNeal Cardwell * algorithms that we must have the following bandaid to talk 421900f65d3SNeal Cardwell * efficiently to them. -DaveM 422900f65d3SNeal Cardwell */ 423900f65d3SNeal Cardwell tp->snd_cwnd = TCP_INIT_CWND; 424900f65d3SNeal Cardwell 425d7722e85SSoheil Hassas Yeganeh /* There's a bubble in the pipe until at least the first ACK. */ 426d7722e85SSoheil Hassas Yeganeh tp->app_limited = ~0U; 427d7722e85SSoheil Hassas Yeganeh 428900f65d3SNeal Cardwell /* See draft-stevens-tcpca-spec-01 for discussion of the 429900f65d3SNeal Cardwell * initialization of these values. 430900f65d3SNeal Cardwell */ 431900f65d3SNeal Cardwell tp->snd_ssthresh = TCP_INFINITE_SSTHRESH; 432900f65d3SNeal Cardwell tp->snd_cwnd_clamp = ~0; 433900f65d3SNeal Cardwell tp->mss_cache = TCP_MSS_DEFAULT; 434900f65d3SNeal Cardwell 4351043e25fSNikolay Borisov tp->reordering = sock_net(sk)->ipv4.sysctl_tcp_reordering; 43655d8694fSFlorian Westphal tcp_assign_congestion_control(sk); 437900f65d3SNeal Cardwell 438ceaa1fefSAndrey Vagin tp->tsoffset = 0; 4391f255691SPriyaranjan Jha tp->rack.reo_wnd_steps = 1; 440ceaa1fefSAndrey Vagin 441900f65d3SNeal Cardwell sk->sk_state = TCP_CLOSE; 442900f65d3SNeal Cardwell 443900f65d3SNeal Cardwell sk->sk_write_space = sk_stream_write_space; 444900f65d3SNeal Cardwell sock_set_flag(sk, SOCK_USE_WRITE_QUEUE); 445900f65d3SNeal Cardwell 446900f65d3SNeal Cardwell icsk->icsk_sync_mss = tcp_sync_mss; 447900f65d3SNeal Cardwell 448356d1833SEric Dumazet sk->sk_sndbuf = sock_net(sk)->ipv4.sysctl_tcp_wmem[1]; 449356d1833SEric Dumazet sk->sk_rcvbuf = sock_net(sk)->ipv4.sysctl_tcp_rmem[1]; 450900f65d3SNeal Cardwell 451900f65d3SNeal Cardwell sk_sockets_allocated_inc(sk); 4520a6b2a1dSEric Dumazet sk->sk_route_forced_caps = NETIF_F_GSO; 453900f65d3SNeal Cardwell } 454900f65d3SNeal Cardwell EXPORT_SYMBOL(tcp_init_sock); 455900f65d3SNeal Cardwell 4564e8cc228SEric Dumazet static void tcp_tx_timestamp(struct sock *sk, u16 tsflags) 4574ed2d765SWillem de Bruijn { 4584e8cc228SEric Dumazet struct sk_buff *skb = tcp_write_queue_tail(sk); 4594e8cc228SEric Dumazet 460ad02c4f5SSoheil Hassas Yeganeh if (tsflags && skb) { 4614ed2d765SWillem de Bruijn struct skb_shared_info *shinfo = skb_shinfo(skb); 4626b084928SSoheil Hassas Yeganeh struct tcp_skb_cb *tcb = TCP_SKB_CB(skb); 4634ed2d765SWillem de Bruijn 464c14ac945SSoheil Hassas Yeganeh sock_tx_timestamp(sk, tsflags, &shinfo->tx_flags); 4650a2cf20cSSoheil Hassas Yeganeh if (tsflags & SOF_TIMESTAMPING_TX_ACK) 4660a2cf20cSSoheil Hassas Yeganeh tcb->txstamp_ack = 1; 4670a2cf20cSSoheil Hassas Yeganeh if (tsflags & SOF_TIMESTAMPING_TX_RECORD_MASK) 4684ed2d765SWillem de Bruijn shinfo->tskey = TCP_SKB_CB(skb)->seq + skb->len - 1; 4694ed2d765SWillem de Bruijn } 470f066e2b0SWillem de Bruijn } 4714ed2d765SWillem de Bruijn 4728934ce2fSJohn Fastabend static inline bool tcp_stream_is_readable(const struct tcp_sock *tp, 4738934ce2fSJohn Fastabend int target, struct sock *sk) 4748934ce2fSJohn Fastabend { 4758934ce2fSJohn Fastabend return (tp->rcv_nxt - tp->copied_seq >= target) || 4768934ce2fSJohn Fastabend (sk->sk_prot->stream_memory_read ? 4778934ce2fSJohn Fastabend sk->sk_prot->stream_memory_read(sk) : false); 4788934ce2fSJohn Fastabend } 4798934ce2fSJohn Fastabend 4801da177e4SLinus Torvalds /* 481a11e1d43SLinus Torvalds * Wait for a TCP event. 482a11e1d43SLinus Torvalds * 483a11e1d43SLinus Torvalds * Note that we don't need to lock the socket, as the upper poll layers 484a11e1d43SLinus Torvalds * take care of normal races (between the test and the event) and we don't 485a11e1d43SLinus Torvalds * go look at any of the socket buffers directly. 4861da177e4SLinus Torvalds */ 487a11e1d43SLinus Torvalds __poll_t tcp_poll(struct file *file, struct socket *sock, poll_table *wait) 4881da177e4SLinus Torvalds { 489a11e1d43SLinus Torvalds __poll_t mask; 4901da177e4SLinus Torvalds struct sock *sk = sock->sk; 491cf533ea5SEric Dumazet const struct tcp_sock *tp = tcp_sk(sk); 49200fd38d9SEric Dumazet int state; 4931da177e4SLinus Torvalds 49489ab066dSKarsten Graul sock_poll_wait(file, sock, wait); 495a11e1d43SLinus Torvalds 496986ffdfdSYafang Shao state = inet_sk_state_load(sk); 49700fd38d9SEric Dumazet if (state == TCP_LISTEN) 498dc40c7bcSArnaldo Carvalho de Melo return inet_csk_listen_poll(sk); 4991da177e4SLinus Torvalds 500a11e1d43SLinus Torvalds /* Socket is not locked. We are protected from async events 501a11e1d43SLinus Torvalds * by poll logic and correct handling of state changes 502a11e1d43SLinus Torvalds * made by other threads is impossible in any case. 503a11e1d43SLinus Torvalds */ 504a11e1d43SLinus Torvalds 505a11e1d43SLinus Torvalds mask = 0; 506a11e1d43SLinus Torvalds 5071da177e4SLinus Torvalds /* 508a9a08845SLinus Torvalds * EPOLLHUP is certainly not done right. But poll() doesn't 5091da177e4SLinus Torvalds * have a notion of HUP in just one direction, and for a 5101da177e4SLinus Torvalds * socket the read side is more interesting. 5111da177e4SLinus Torvalds * 512a9a08845SLinus Torvalds * Some poll() documentation says that EPOLLHUP is incompatible 513a9a08845SLinus Torvalds * with the EPOLLOUT/POLLWR flags, so somebody should check this 5141da177e4SLinus Torvalds * all. But careful, it tends to be safer to return too many 5151da177e4SLinus Torvalds * bits than too few, and you can easily break real applications 5161da177e4SLinus Torvalds * if you don't tell them that something has hung up! 5171da177e4SLinus Torvalds * 5181da177e4SLinus Torvalds * Check-me. 5191da177e4SLinus Torvalds * 520a9a08845SLinus Torvalds * Check number 1. EPOLLHUP is _UNMASKABLE_ event (see UNIX98 and 5211da177e4SLinus Torvalds * our fs/select.c). It means that after we received EOF, 5221da177e4SLinus Torvalds * poll always returns immediately, making impossible poll() on write() 523a9a08845SLinus Torvalds * in state CLOSE_WAIT. One solution is evident --- to set EPOLLHUP 5241da177e4SLinus Torvalds * if and only if shutdown has been made in both directions. 5251da177e4SLinus Torvalds * Actually, it is interesting to look how Solaris and DUX 526a9a08845SLinus Torvalds * solve this dilemma. I would prefer, if EPOLLHUP were maskable, 5271da177e4SLinus Torvalds * then we could set it on SND_SHUTDOWN. BTW examples given 5281da177e4SLinus Torvalds * in Stevens' books assume exactly this behaviour, it explains 529a9a08845SLinus Torvalds * why EPOLLHUP is incompatible with EPOLLOUT. --ANK 5301da177e4SLinus Torvalds * 5311da177e4SLinus Torvalds * NOTE. Check for TCP_CLOSE is added. The goal is to prevent 5321da177e4SLinus Torvalds * blocking on fresh not-connected or disconnected socket. --ANK 5331da177e4SLinus Torvalds */ 53400fd38d9SEric Dumazet if (sk->sk_shutdown == SHUTDOWN_MASK || state == TCP_CLOSE) 535a9a08845SLinus Torvalds mask |= EPOLLHUP; 5361da177e4SLinus Torvalds if (sk->sk_shutdown & RCV_SHUTDOWN) 537a9a08845SLinus Torvalds mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP; 5381da177e4SLinus Torvalds 5398336886fSJerry Chu /* Connected or passive Fast Open socket? */ 54000fd38d9SEric Dumazet if (state != TCP_SYN_SENT && 54100fd38d9SEric Dumazet (state != TCP_SYN_RECV || tp->fastopen_rsk)) { 542c7004482SDavid S. Miller int target = sock_rcvlowat(sk, 0, INT_MAX); 543c7004482SDavid S. Miller 544c7004482SDavid S. Miller if (tp->urg_seq == tp->copied_seq && 545c7004482SDavid S. Miller !sock_flag(sk, SOCK_URGINLINE) && 546c7004482SDavid S. Miller tp->urg_data) 547b634f875SAlexandra Kossovsky target++; 548c7004482SDavid S. Miller 5498934ce2fSJohn Fastabend if (tcp_stream_is_readable(tp, target, sk)) 550a9a08845SLinus Torvalds mask |= EPOLLIN | EPOLLRDNORM; 5511da177e4SLinus Torvalds 5521da177e4SLinus Torvalds if (!(sk->sk_shutdown & SEND_SHUTDOWN)) { 55364dc6130SEric Dumazet if (sk_stream_is_writeable(sk)) { 554a9a08845SLinus Torvalds mask |= EPOLLOUT | EPOLLWRNORM; 5551da177e4SLinus Torvalds } else { /* send SIGIO later */ 5569cd3e072SEric Dumazet sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk); 5571da177e4SLinus Torvalds set_bit(SOCK_NOSPACE, &sk->sk_socket->flags); 5581da177e4SLinus Torvalds 5591da177e4SLinus Torvalds /* Race breaker. If space is freed after 5601da177e4SLinus Torvalds * wspace test but before the flags are set, 5613c715127Sjbaron@akamai.com * IO signal will be lost. Memory barrier 5623c715127Sjbaron@akamai.com * pairs with the input side. 5631da177e4SLinus Torvalds */ 5643c715127Sjbaron@akamai.com smp_mb__after_atomic(); 56564dc6130SEric Dumazet if (sk_stream_is_writeable(sk)) 566a9a08845SLinus Torvalds mask |= EPOLLOUT | EPOLLWRNORM; 5671da177e4SLinus Torvalds } 568d84ba638SKOSAKI Motohiro } else 569a9a08845SLinus Torvalds mask |= EPOLLOUT | EPOLLWRNORM; 5701da177e4SLinus Torvalds 5711da177e4SLinus Torvalds if (tp->urg_data & TCP_URG_VALID) 572a9a08845SLinus Torvalds mask |= EPOLLPRI; 573d68be71eSDavide Caratti } else if (state == TCP_SYN_SENT && inet_sk(sk)->defer_connect) { 57419f6d3f3SWei Wang /* Active TCP fastopen socket with defer_connect 575a9a08845SLinus Torvalds * Return EPOLLOUT so application can call write() 57619f6d3f3SWei Wang * in order for kernel to generate SYN+data 57719f6d3f3SWei Wang */ 578a9a08845SLinus Torvalds mask |= EPOLLOUT | EPOLLWRNORM; 5791da177e4SLinus Torvalds } 580a4d25803STom Marshall /* This barrier is coupled with smp_wmb() in tcp_reset() */ 581a4d25803STom Marshall smp_rmb(); 5824ed2d765SWillem de Bruijn if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue)) 583a9a08845SLinus Torvalds mask |= EPOLLERR; 584a4d25803STom Marshall 5851da177e4SLinus Torvalds return mask; 5861da177e4SLinus Torvalds } 587a11e1d43SLinus Torvalds EXPORT_SYMBOL(tcp_poll); 5881da177e4SLinus Torvalds 5891da177e4SLinus Torvalds int tcp_ioctl(struct sock *sk, int cmd, unsigned long arg) 5901da177e4SLinus Torvalds { 5911da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 5921da177e4SLinus Torvalds int answ; 5930e71c55cSEric Dumazet bool slow; 5941da177e4SLinus Torvalds 5951da177e4SLinus Torvalds switch (cmd) { 5961da177e4SLinus Torvalds case SIOCINQ: 5971da177e4SLinus Torvalds if (sk->sk_state == TCP_LISTEN) 5981da177e4SLinus Torvalds return -EINVAL; 5991da177e4SLinus Torvalds 6000e71c55cSEric Dumazet slow = lock_sock_fast(sk); 601473bd239STom Herbert answ = tcp_inq(sk); 6020e71c55cSEric Dumazet unlock_sock_fast(sk, slow); 6031da177e4SLinus Torvalds break; 6041da177e4SLinus Torvalds case SIOCATMARK: 6051da177e4SLinus Torvalds answ = tp->urg_data && tp->urg_seq == tp->copied_seq; 6061da177e4SLinus Torvalds break; 6071da177e4SLinus Torvalds case SIOCOUTQ: 6081da177e4SLinus Torvalds if (sk->sk_state == TCP_LISTEN) 6091da177e4SLinus Torvalds return -EINVAL; 6101da177e4SLinus Torvalds 6111da177e4SLinus Torvalds if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) 6121da177e4SLinus Torvalds answ = 0; 6131da177e4SLinus Torvalds else 6141da177e4SLinus Torvalds answ = tp->write_seq - tp->snd_una; 6151da177e4SLinus Torvalds break; 6162f4e1b39SMario Schuknecht case SIOCOUTQNSD: 6172f4e1b39SMario Schuknecht if (sk->sk_state == TCP_LISTEN) 6182f4e1b39SMario Schuknecht return -EINVAL; 6192f4e1b39SMario Schuknecht 6202f4e1b39SMario Schuknecht if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) 6212f4e1b39SMario Schuknecht answ = 0; 6222f4e1b39SMario Schuknecht else 6232f4e1b39SMario Schuknecht answ = tp->write_seq - tp->snd_nxt; 6242f4e1b39SMario Schuknecht break; 6251da177e4SLinus Torvalds default: 6261da177e4SLinus Torvalds return -ENOIOCTLCMD; 6273ff50b79SStephen Hemminger } 6281da177e4SLinus Torvalds 6291da177e4SLinus Torvalds return put_user(answ, (int __user *)arg); 6301da177e4SLinus Torvalds } 6314bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_ioctl); 6321da177e4SLinus Torvalds 6331da177e4SLinus Torvalds static inline void tcp_mark_push(struct tcp_sock *tp, struct sk_buff *skb) 6341da177e4SLinus Torvalds { 6354de075e0SEric Dumazet TCP_SKB_CB(skb)->tcp_flags |= TCPHDR_PSH; 6361da177e4SLinus Torvalds tp->pushed_seq = tp->write_seq; 6371da177e4SLinus Torvalds } 6381da177e4SLinus Torvalds 639a2a385d6SEric Dumazet static inline bool forced_push(const struct tcp_sock *tp) 6401da177e4SLinus Torvalds { 6411da177e4SLinus Torvalds return after(tp->write_seq, tp->pushed_seq + (tp->max_window >> 1)); 6421da177e4SLinus Torvalds } 6431da177e4SLinus Torvalds 644f4a775d1SEric Dumazet static void skb_entail(struct sock *sk, struct sk_buff *skb) 6451da177e4SLinus Torvalds { 6469e412ba7SIlpo Järvinen struct tcp_sock *tp = tcp_sk(sk); 647352d4800SArnaldo Carvalho de Melo struct tcp_skb_cb *tcb = TCP_SKB_CB(skb); 648352d4800SArnaldo Carvalho de Melo 6491da177e4SLinus Torvalds skb->csum = 0; 650352d4800SArnaldo Carvalho de Melo tcb->seq = tcb->end_seq = tp->write_seq; 6514de075e0SEric Dumazet tcb->tcp_flags = TCPHDR_ACK; 652352d4800SArnaldo Carvalho de Melo tcb->sacked = 0; 653f4a775d1SEric Dumazet __skb_header_release(skb); 654fe067e8aSDavid S. Miller tcp_add_write_queue_tail(sk, skb); 6553ab224beSHideo Aoki sk->sk_wmem_queued += skb->truesize; 6563ab224beSHideo Aoki sk_mem_charge(sk, skb->truesize); 65789ebd197SDavid S. Miller if (tp->nonagle & TCP_NAGLE_PUSH) 6581da177e4SLinus Torvalds tp->nonagle &= ~TCP_NAGLE_PUSH; 6596f021c62SEric Dumazet 6606f021c62SEric Dumazet tcp_slow_start_after_idle_check(sk); 6611da177e4SLinus Torvalds } 6621da177e4SLinus Torvalds 663afeca340SKrishna Kumar static inline void tcp_mark_urg(struct tcp_sock *tp, int flags) 6641da177e4SLinus Torvalds { 66533f5f57eSIlpo Järvinen if (flags & MSG_OOB) 6661da177e4SLinus Torvalds tp->snd_up = tp->write_seq; 6671da177e4SLinus Torvalds } 6681da177e4SLinus Torvalds 669f54b3111SEric Dumazet /* If a not yet filled skb is pushed, do not send it if 670a181ceb5SEric Dumazet * we have data packets in Qdisc or NIC queues : 671f54b3111SEric Dumazet * Because TX completion will happen shortly, it gives a chance 672f54b3111SEric Dumazet * to coalesce future sendmsg() payload into this skb, without 673f54b3111SEric Dumazet * need for a timer, and with no latency trade off. 674f54b3111SEric Dumazet * As packets containing data payload have a bigger truesize 675a181ceb5SEric Dumazet * than pure acks (dataless) packets, the last checks prevent 676a181ceb5SEric Dumazet * autocorking if we only have an ACK in Qdisc/NIC queues, 677a181ceb5SEric Dumazet * or if TX completion was delayed after we processed ACK packet. 678f54b3111SEric Dumazet */ 679f54b3111SEric Dumazet static bool tcp_should_autocork(struct sock *sk, struct sk_buff *skb, 680f54b3111SEric Dumazet int size_goal) 6811da177e4SLinus Torvalds { 682f54b3111SEric Dumazet return skb->len < size_goal && 683790f00e1SEric Dumazet sock_net(sk)->ipv4.sysctl_tcp_autocorking && 684114f39feSEric Dumazet !tcp_rtx_queue_empty(sk) && 68514afee4bSReshetova, Elena refcount_read(&sk->sk_wmem_alloc) > skb->truesize; 686f54b3111SEric Dumazet } 6879e412ba7SIlpo Järvinen 688f54b3111SEric Dumazet static void tcp_push(struct sock *sk, int flags, int mss_now, 689f54b3111SEric Dumazet int nonagle, int size_goal) 690f54b3111SEric Dumazet { 691f54b3111SEric Dumazet struct tcp_sock *tp = tcp_sk(sk); 692f54b3111SEric Dumazet struct sk_buff *skb; 693f54b3111SEric Dumazet 694f54b3111SEric Dumazet skb = tcp_write_queue_tail(sk); 69575c119afSEric Dumazet if (!skb) 69675c119afSEric Dumazet return; 6971da177e4SLinus Torvalds if (!(flags & MSG_MORE) || forced_push(tp)) 698f54b3111SEric Dumazet tcp_mark_push(tp, skb); 699afeca340SKrishna Kumar 700afeca340SKrishna Kumar tcp_mark_urg(tp, flags); 701f54b3111SEric Dumazet 702f54b3111SEric Dumazet if (tcp_should_autocork(sk, skb, size_goal)) { 703f54b3111SEric Dumazet 704f54b3111SEric Dumazet /* avoid atomic op if TSQ_THROTTLED bit is already set */ 7057aa5470cSEric Dumazet if (!test_bit(TSQ_THROTTLED, &sk->sk_tsq_flags)) { 706f54b3111SEric Dumazet NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPAUTOCORKING); 7077aa5470cSEric Dumazet set_bit(TSQ_THROTTLED, &sk->sk_tsq_flags); 7081da177e4SLinus Torvalds } 709a181ceb5SEric Dumazet /* It is possible TX completion already happened 710a181ceb5SEric Dumazet * before we set TSQ_THROTTLED. 711a181ceb5SEric Dumazet */ 71214afee4bSReshetova, Elena if (refcount_read(&sk->sk_wmem_alloc) > skb->truesize) 713f54b3111SEric Dumazet return; 714f54b3111SEric Dumazet } 715f54b3111SEric Dumazet 716f54b3111SEric Dumazet if (flags & MSG_MORE) 717f54b3111SEric Dumazet nonagle = TCP_NAGLE_CORK; 718f54b3111SEric Dumazet 719f54b3111SEric Dumazet __tcp_push_pending_frames(sk, mss_now, nonagle); 7201da177e4SLinus Torvalds } 7211da177e4SLinus Torvalds 7226ff7751dSAdrian Bunk static int tcp_splice_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb, 7239c55e01cSJens Axboe unsigned int offset, size_t len) 7249c55e01cSJens Axboe { 7259c55e01cSJens Axboe struct tcp_splice_state *tss = rd_desc->arg.data; 72633966dd0SWilly Tarreau int ret; 7279c55e01cSJens Axboe 728a60e3cc7SHannes Frederic Sowa ret = skb_splice_bits(skb, skb->sk, offset, tss->pipe, 72925869262SAl Viro min(rd_desc->count, len), tss->flags); 73033966dd0SWilly Tarreau if (ret > 0) 73133966dd0SWilly Tarreau rd_desc->count -= ret; 73233966dd0SWilly Tarreau return ret; 7339c55e01cSJens Axboe } 7349c55e01cSJens Axboe 7359c55e01cSJens Axboe static int __tcp_splice_read(struct sock *sk, struct tcp_splice_state *tss) 7369c55e01cSJens Axboe { 7379c55e01cSJens Axboe /* Store TCP splice context information in read_descriptor_t. */ 7389c55e01cSJens Axboe read_descriptor_t rd_desc = { 7399c55e01cSJens Axboe .arg.data = tss, 74033966dd0SWilly Tarreau .count = tss->len, 7419c55e01cSJens Axboe }; 7429c55e01cSJens Axboe 7439c55e01cSJens Axboe return tcp_read_sock(sk, &rd_desc, tcp_splice_data_recv); 7449c55e01cSJens Axboe } 7459c55e01cSJens Axboe 7469c55e01cSJens Axboe /** 7479c55e01cSJens Axboe * tcp_splice_read - splice data from TCP socket to a pipe 7489c55e01cSJens Axboe * @sock: socket to splice from 7499c55e01cSJens Axboe * @ppos: position (not valid) 7509c55e01cSJens Axboe * @pipe: pipe to splice to 7519c55e01cSJens Axboe * @len: number of bytes to splice 7529c55e01cSJens Axboe * @flags: splice modifier flags 7539c55e01cSJens Axboe * 7549c55e01cSJens Axboe * Description: 7559c55e01cSJens Axboe * Will read pages from given socket and fill them into a pipe. 7569c55e01cSJens Axboe * 7579c55e01cSJens Axboe **/ 7589c55e01cSJens Axboe ssize_t tcp_splice_read(struct socket *sock, loff_t *ppos, 7599c55e01cSJens Axboe struct pipe_inode_info *pipe, size_t len, 7609c55e01cSJens Axboe unsigned int flags) 7619c55e01cSJens Axboe { 7629c55e01cSJens Axboe struct sock *sk = sock->sk; 7639c55e01cSJens Axboe struct tcp_splice_state tss = { 7649c55e01cSJens Axboe .pipe = pipe, 7659c55e01cSJens Axboe .len = len, 7669c55e01cSJens Axboe .flags = flags, 7679c55e01cSJens Axboe }; 7689c55e01cSJens Axboe long timeo; 7699c55e01cSJens Axboe ssize_t spliced; 7709c55e01cSJens Axboe int ret; 7719c55e01cSJens Axboe 7723a047bf8SChangli Gao sock_rps_record_flow(sk); 7739c55e01cSJens Axboe /* 7749c55e01cSJens Axboe * We can't seek on a socket input 7759c55e01cSJens Axboe */ 7769c55e01cSJens Axboe if (unlikely(*ppos)) 7779c55e01cSJens Axboe return -ESPIPE; 7789c55e01cSJens Axboe 7799c55e01cSJens Axboe ret = spliced = 0; 7809c55e01cSJens Axboe 7819c55e01cSJens Axboe lock_sock(sk); 7829c55e01cSJens Axboe 78342324c62SEric Dumazet timeo = sock_rcvtimeo(sk, sock->file->f_flags & O_NONBLOCK); 7849c55e01cSJens Axboe while (tss.len) { 7859c55e01cSJens Axboe ret = __tcp_splice_read(sk, &tss); 7869c55e01cSJens Axboe if (ret < 0) 7879c55e01cSJens Axboe break; 7889c55e01cSJens Axboe else if (!ret) { 7899c55e01cSJens Axboe if (spliced) 7909c55e01cSJens Axboe break; 7919c55e01cSJens Axboe if (sock_flag(sk, SOCK_DONE)) 7929c55e01cSJens Axboe break; 7939c55e01cSJens Axboe if (sk->sk_err) { 7949c55e01cSJens Axboe ret = sock_error(sk); 7959c55e01cSJens Axboe break; 7969c55e01cSJens Axboe } 7979c55e01cSJens Axboe if (sk->sk_shutdown & RCV_SHUTDOWN) 7989c55e01cSJens Axboe break; 7999c55e01cSJens Axboe if (sk->sk_state == TCP_CLOSE) { 8009c55e01cSJens Axboe /* 8019c55e01cSJens Axboe * This occurs when user tries to read 8029c55e01cSJens Axboe * from never connected socket. 8039c55e01cSJens Axboe */ 8049c55e01cSJens Axboe ret = -ENOTCONN; 8059c55e01cSJens Axboe break; 8069c55e01cSJens Axboe } 8079c55e01cSJens Axboe if (!timeo) { 8089c55e01cSJens Axboe ret = -EAGAIN; 8099c55e01cSJens Axboe break; 8109c55e01cSJens Axboe } 811ccf7abb9SEric Dumazet /* if __tcp_splice_read() got nothing while we have 812ccf7abb9SEric Dumazet * an skb in receive queue, we do not want to loop. 813ccf7abb9SEric Dumazet * This might happen with URG data. 814ccf7abb9SEric Dumazet */ 815ccf7abb9SEric Dumazet if (!skb_queue_empty(&sk->sk_receive_queue)) 816ccf7abb9SEric Dumazet break; 817dfbafc99SSabrina Dubroca sk_wait_data(sk, &timeo, NULL); 8189c55e01cSJens Axboe if (signal_pending(current)) { 8199c55e01cSJens Axboe ret = sock_intr_errno(timeo); 8209c55e01cSJens Axboe break; 8219c55e01cSJens Axboe } 8229c55e01cSJens Axboe continue; 8239c55e01cSJens Axboe } 8249c55e01cSJens Axboe tss.len -= ret; 8259c55e01cSJens Axboe spliced += ret; 8269c55e01cSJens Axboe 82733966dd0SWilly Tarreau if (!timeo) 82833966dd0SWilly Tarreau break; 8299c55e01cSJens Axboe release_sock(sk); 8309c55e01cSJens Axboe lock_sock(sk); 8319c55e01cSJens Axboe 8329c55e01cSJens Axboe if (sk->sk_err || sk->sk_state == TCP_CLOSE || 83333966dd0SWilly Tarreau (sk->sk_shutdown & RCV_SHUTDOWN) || 8349c55e01cSJens Axboe signal_pending(current)) 8359c55e01cSJens Axboe break; 8369c55e01cSJens Axboe } 8379c55e01cSJens Axboe 8389c55e01cSJens Axboe release_sock(sk); 8399c55e01cSJens Axboe 8409c55e01cSJens Axboe if (spliced) 8419c55e01cSJens Axboe return spliced; 8429c55e01cSJens Axboe 8439c55e01cSJens Axboe return ret; 8449c55e01cSJens Axboe } 8454bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_splice_read); 8469c55e01cSJens Axboe 847eb934478SEric Dumazet struct sk_buff *sk_stream_alloc_skb(struct sock *sk, int size, gfp_t gfp, 848eb934478SEric Dumazet bool force_schedule) 849f561d0f2SPavel Emelyanov { 850f561d0f2SPavel Emelyanov struct sk_buff *skb; 851f561d0f2SPavel Emelyanov 8524f661542SEric Dumazet if (likely(!size)) { 853472c2e07SEric Dumazet skb = sk->sk_tx_skb_cache; 854858f5017SEric Dumazet if (skb) { 855d7cc399eSEric Dumazet skb->truesize = SKB_TRUESIZE(skb_end_offset(skb)); 856472c2e07SEric Dumazet sk->sk_tx_skb_cache = NULL; 857472c2e07SEric Dumazet pskb_trim(skb, 0); 858472c2e07SEric Dumazet INIT_LIST_HEAD(&skb->tcp_tsorted_anchor); 859eb70a1aeSEric Dumazet skb_shinfo(skb)->tx_flags = 0; 860eb70a1aeSEric Dumazet memset(TCP_SKB_CB(skb), 0, sizeof(struct tcp_skb_cb)); 861472c2e07SEric Dumazet return skb; 862472c2e07SEric Dumazet } 863472c2e07SEric Dumazet } 864f561d0f2SPavel Emelyanov /* The TCP header must be at least 32-bit aligned. */ 865f561d0f2SPavel Emelyanov size = ALIGN(size, 4); 866f561d0f2SPavel Emelyanov 8678e4d980aSEric Dumazet if (unlikely(tcp_under_memory_pressure(sk))) 8688e4d980aSEric Dumazet sk_mem_reclaim_partial(sk); 8698e4d980aSEric Dumazet 870f561d0f2SPavel Emelyanov skb = alloc_skb_fclone(size + sk->sk_prot->max_header, gfp); 8718e4d980aSEric Dumazet if (likely(skb)) { 872eb934478SEric Dumazet bool mem_scheduled; 8738e4d980aSEric Dumazet 874eb934478SEric Dumazet if (force_schedule) { 875eb934478SEric Dumazet mem_scheduled = true; 8768e4d980aSEric Dumazet sk_forced_mem_schedule(sk, skb->truesize); 8778e4d980aSEric Dumazet } else { 878eb934478SEric Dumazet mem_scheduled = sk_wmem_schedule(sk, skb->truesize); 8798e4d980aSEric Dumazet } 880eb934478SEric Dumazet if (likely(mem_scheduled)) { 881a21d4572SEric Dumazet skb_reserve(skb, sk->sk_prot->max_header); 882f561d0f2SPavel Emelyanov /* 883f561d0f2SPavel Emelyanov * Make sure that we have exactly size bytes 884f561d0f2SPavel Emelyanov * available to the caller, no more, no less. 885f561d0f2SPavel Emelyanov */ 88616fad69cSEric Dumazet skb->reserved_tailroom = skb->end - skb->tail - size; 887e2080072SEric Dumazet INIT_LIST_HEAD(&skb->tcp_tsorted_anchor); 888f561d0f2SPavel Emelyanov return skb; 889f561d0f2SPavel Emelyanov } 890f561d0f2SPavel Emelyanov __kfree_skb(skb); 891f561d0f2SPavel Emelyanov } else { 8925c52ba17SPavel Emelyanov sk->sk_prot->enter_memory_pressure(sk); 893f561d0f2SPavel Emelyanov sk_stream_moderate_sndbuf(sk); 894f561d0f2SPavel Emelyanov } 895f561d0f2SPavel Emelyanov return NULL; 896f561d0f2SPavel Emelyanov } 897f561d0f2SPavel Emelyanov 8980c54b85fSIlpo Järvinen static unsigned int tcp_xmit_size_goal(struct sock *sk, u32 mss_now, 8990c54b85fSIlpo Järvinen int large_allowed) 9000c54b85fSIlpo Järvinen { 9010c54b85fSIlpo Järvinen struct tcp_sock *tp = tcp_sk(sk); 9026c09fa09SEric Dumazet u32 new_size_goal, size_goal; 9030c54b85fSIlpo Järvinen 90474d4a8f8SEric Dumazet if (!large_allowed) 905605ad7f1SEric Dumazet return mss_now; 9060c54b85fSIlpo Järvinen 9076c09fa09SEric Dumazet /* Note : tcp_tso_autosize() will eventually split this later */ 9086c09fa09SEric Dumazet new_size_goal = sk->sk_gso_max_size - 1 - MAX_TCP_HEADER; 909605ad7f1SEric Dumazet new_size_goal = tcp_bound_to_half_wnd(tp, new_size_goal); 9102a3a041cSIlpo Järvinen 9112a3a041cSIlpo Järvinen /* We try hard to avoid divides here */ 912605ad7f1SEric Dumazet size_goal = tp->gso_segs * mss_now; 913605ad7f1SEric Dumazet if (unlikely(new_size_goal < size_goal || 914605ad7f1SEric Dumazet new_size_goal >= size_goal + mss_now)) { 915605ad7f1SEric Dumazet tp->gso_segs = min_t(u16, new_size_goal / mss_now, 9161485348dSBen Hutchings sk->sk_gso_max_segs); 917605ad7f1SEric Dumazet size_goal = tp->gso_segs * mss_now; 9180c54b85fSIlpo Järvinen } 9190c54b85fSIlpo Järvinen 920605ad7f1SEric Dumazet return max(size_goal, mss_now); 9210c54b85fSIlpo Järvinen } 9220c54b85fSIlpo Järvinen 9230c54b85fSIlpo Järvinen static int tcp_send_mss(struct sock *sk, int *size_goal, int flags) 9240c54b85fSIlpo Järvinen { 9250c54b85fSIlpo Järvinen int mss_now; 9260c54b85fSIlpo Järvinen 9270c54b85fSIlpo Järvinen mss_now = tcp_current_mss(sk); 9280c54b85fSIlpo Järvinen *size_goal = tcp_xmit_size_goal(sk, mss_now, !(flags & MSG_OOB)); 9290c54b85fSIlpo Järvinen 9300c54b85fSIlpo Järvinen return mss_now; 9310c54b85fSIlpo Järvinen } 9320c54b85fSIlpo Järvinen 933e3b5616aSDave Watson ssize_t do_tcp_sendpages(struct sock *sk, struct page *page, int offset, 93464022d0bSEric Dumazet size_t size, int flags) 9351da177e4SLinus Torvalds { 9361da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 937c1b4a7e6SDavid S. Miller int mss_now, size_goal; 9381da177e4SLinus Torvalds int err; 9391da177e4SLinus Torvalds ssize_t copied; 9401da177e4SLinus Torvalds long timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT); 9411da177e4SLinus Torvalds 942a10674bfSVasily Averin if (IS_ENABLED(CONFIG_DEBUG_VM) && 943a10674bfSVasily Averin WARN_ONCE(PageSlab(page), "page must not be a Slab one")) 944a10674bfSVasily Averin return -EINVAL; 945a10674bfSVasily Averin 9468336886fSJerry Chu /* Wait for a connection to finish. One exception is TCP Fast Open 9478336886fSJerry Chu * (passive side) where data is allowed to be sent before a connection 9488336886fSJerry Chu * is fully established. 9498336886fSJerry Chu */ 9508336886fSJerry Chu if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) && 9518336886fSJerry Chu !tcp_passive_fastopen(sk)) { 952686a5624SYuvaraja Mariappan err = sk_stream_wait_connect(sk, &timeo); 953686a5624SYuvaraja Mariappan if (err != 0) 9541da177e4SLinus Torvalds goto out_err; 9558336886fSJerry Chu } 9561da177e4SLinus Torvalds 9579cd3e072SEric Dumazet sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk); 9581da177e4SLinus Torvalds 9590c54b85fSIlpo Järvinen mss_now = tcp_send_mss(sk, &size_goal, flags); 9601da177e4SLinus Torvalds copied = 0; 9611da177e4SLinus Torvalds 9621da177e4SLinus Torvalds err = -EPIPE; 9631da177e4SLinus Torvalds if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN)) 9640d6a775eSIlpo Järvinen goto out_err; 9651da177e4SLinus Torvalds 96664022d0bSEric Dumazet while (size > 0) { 967fe067e8aSDavid S. Miller struct sk_buff *skb = tcp_write_queue_tail(sk); 96838ba0a65SEric Dumazet int copy, i; 96938ba0a65SEric Dumazet bool can_coalesce; 9701da177e4SLinus Torvalds 97175c119afSEric Dumazet if (!skb || (copy = size_goal - skb->len) <= 0 || 972c134ecb8SMartin KaFai Lau !tcp_skb_can_collapse_to(skb)) { 9731da177e4SLinus Torvalds new_segment: 9741da177e4SLinus Torvalds if (!sk_stream_memory_free(sk)) 9751da177e4SLinus Torvalds goto wait_for_sndbuf; 9761da177e4SLinus Torvalds 977eb934478SEric Dumazet skb = sk_stream_alloc_skb(sk, 0, sk->sk_allocation, 97875c119afSEric Dumazet tcp_rtx_and_write_queues_empty(sk)); 9791da177e4SLinus Torvalds if (!skb) 9801da177e4SLinus Torvalds goto wait_for_memory; 9811da177e4SLinus Torvalds 9829e412ba7SIlpo Järvinen skb_entail(sk, skb); 983c1b4a7e6SDavid S. Miller copy = size_goal; 9841da177e4SLinus Torvalds } 9851da177e4SLinus Torvalds 9861da177e4SLinus Torvalds if (copy > size) 9871da177e4SLinus Torvalds copy = size; 9881da177e4SLinus Torvalds 9891da177e4SLinus Torvalds i = skb_shinfo(skb)->nr_frags; 9901da177e4SLinus Torvalds can_coalesce = skb_can_coalesce(skb, i, page, offset); 9915f74f82eSHans Westgaard Ry if (!can_coalesce && i >= sysctl_max_skb_frags) { 9921da177e4SLinus Torvalds tcp_mark_push(tp, skb); 9931da177e4SLinus Torvalds goto new_segment; 9941da177e4SLinus Torvalds } 9953ab224beSHideo Aoki if (!sk_wmem_schedule(sk, copy)) 9961da177e4SLinus Torvalds goto wait_for_memory; 9971da177e4SLinus Torvalds 9981da177e4SLinus Torvalds if (can_coalesce) { 9999e903e08SEric Dumazet skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy); 10001da177e4SLinus Torvalds } else { 10011da177e4SLinus Torvalds get_page(page); 10021da177e4SLinus Torvalds skb_fill_page_desc(skb, i, page, offset, copy); 10031da177e4SLinus Torvalds } 1004312fc2b4SJohn Fastabend 1005312fc2b4SJohn Fastabend if (!(flags & MSG_NO_SHARED_FRAGS)) 1006c9af6db4SPravin B Shelar skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG; 1007cef401deSEric Dumazet 10081da177e4SLinus Torvalds skb->len += copy; 10091da177e4SLinus Torvalds skb->data_len += copy; 10101da177e4SLinus Torvalds skb->truesize += copy; 10111da177e4SLinus Torvalds sk->sk_wmem_queued += copy; 10123ab224beSHideo Aoki sk_mem_charge(sk, copy); 101384fa7933SPatrick McHardy skb->ip_summed = CHECKSUM_PARTIAL; 10141da177e4SLinus Torvalds tp->write_seq += copy; 10151da177e4SLinus Torvalds TCP_SKB_CB(skb)->end_seq += copy; 1016cd7d8498SEric Dumazet tcp_skb_pcount_set(skb, 0); 10171da177e4SLinus Torvalds 10181da177e4SLinus Torvalds if (!copied) 10194de075e0SEric Dumazet TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH; 10201da177e4SLinus Torvalds 10211da177e4SLinus Torvalds copied += copy; 102264022d0bSEric Dumazet offset += copy; 1023686a5624SYuvaraja Mariappan size -= copy; 1024ad02c4f5SSoheil Hassas Yeganeh if (!size) 10251da177e4SLinus Torvalds goto out; 10261da177e4SLinus Torvalds 102769d15067SHerbert Xu if (skb->len < size_goal || (flags & MSG_OOB)) 10281da177e4SLinus Torvalds continue; 10291da177e4SLinus Torvalds 10301da177e4SLinus Torvalds if (forced_push(tp)) { 10311da177e4SLinus Torvalds tcp_mark_push(tp, skb); 10329e412ba7SIlpo Järvinen __tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH); 1033fe067e8aSDavid S. Miller } else if (skb == tcp_send_head(sk)) 10341da177e4SLinus Torvalds tcp_push_one(sk, mss_now); 10351da177e4SLinus Torvalds continue; 10361da177e4SLinus Torvalds 10371da177e4SLinus Torvalds wait_for_sndbuf: 10381da177e4SLinus Torvalds set_bit(SOCK_NOSPACE, &sk->sk_socket->flags); 10391da177e4SLinus Torvalds wait_for_memory: 1040f54b3111SEric Dumazet tcp_push(sk, flags & ~MSG_MORE, mss_now, 1041f54b3111SEric Dumazet TCP_NAGLE_PUSH, size_goal); 10421da177e4SLinus Torvalds 1043686a5624SYuvaraja Mariappan err = sk_stream_wait_memory(sk, &timeo); 1044686a5624SYuvaraja Mariappan if (err != 0) 10451da177e4SLinus Torvalds goto do_error; 10461da177e4SLinus Torvalds 10470c54b85fSIlpo Järvinen mss_now = tcp_send_mss(sk, &size_goal, flags); 10481da177e4SLinus Torvalds } 10491da177e4SLinus Torvalds 10501da177e4SLinus Torvalds out: 1051ad02c4f5SSoheil Hassas Yeganeh if (copied) { 10524e8cc228SEric Dumazet tcp_tx_timestamp(sk, sk->sk_tsflags); 1053ad02c4f5SSoheil Hassas Yeganeh if (!(flags & MSG_SENDPAGE_NOTLAST)) 1054f54b3111SEric Dumazet tcp_push(sk, flags, mss_now, tp->nonagle, size_goal); 1055ad02c4f5SSoheil Hassas Yeganeh } 10561da177e4SLinus Torvalds return copied; 10571da177e4SLinus Torvalds 10581da177e4SLinus Torvalds do_error: 10591da177e4SLinus Torvalds if (copied) 10601da177e4SLinus Torvalds goto out; 10611da177e4SLinus Torvalds out_err: 1062ce5ec440SJason Baron /* make sure we wake any epoll edge trigger waiter */ 1063b0f71bd3SFrancis Yan if (unlikely(skb_queue_len(&sk->sk_write_queue) == 0 && 1064b0f71bd3SFrancis Yan err == -EAGAIN)) { 1065ce5ec440SJason Baron sk->sk_write_space(sk); 1066b0f71bd3SFrancis Yan tcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED); 1067b0f71bd3SFrancis Yan } 10681da177e4SLinus Torvalds return sk_stream_error(sk, flags, err); 10691da177e4SLinus Torvalds } 1070e3b5616aSDave Watson EXPORT_SYMBOL_GPL(do_tcp_sendpages); 10711da177e4SLinus Torvalds 1072306b13ebSTom Herbert int tcp_sendpage_locked(struct sock *sk, struct page *page, int offset, 10731da177e4SLinus Torvalds size_t size, int flags) 10741da177e4SLinus Torvalds { 1075dead7cdbSEric Dumazet if (!(sk->sk_route_caps & NETIF_F_SG)) 1076bd9dfc54SEric Dumazet return sock_no_sendpage_locked(sk, page, offset, size, flags); 10771da177e4SLinus Torvalds 1078d7722e85SSoheil Hassas Yeganeh tcp_rate_check_app_limited(sk); /* is sending application-limited? */ 1079d7722e85SSoheil Hassas Yeganeh 1080306b13ebSTom Herbert return do_tcp_sendpages(sk, page, offset, size, flags); 1081306b13ebSTom Herbert } 1082774c4673SDavid S. Miller EXPORT_SYMBOL_GPL(tcp_sendpage_locked); 1083306b13ebSTom Herbert 1084306b13ebSTom Herbert int tcp_sendpage(struct sock *sk, struct page *page, int offset, 1085306b13ebSTom Herbert size_t size, int flags) 1086306b13ebSTom Herbert { 1087306b13ebSTom Herbert int ret; 1088306b13ebSTom Herbert 1089306b13ebSTom Herbert lock_sock(sk); 1090306b13ebSTom Herbert ret = tcp_sendpage_locked(sk, page, offset, size, flags); 10911da177e4SLinus Torvalds release_sock(sk); 1092306b13ebSTom Herbert 1093306b13ebSTom Herbert return ret; 10941da177e4SLinus Torvalds } 10954bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_sendpage); 10961da177e4SLinus Torvalds 1097cf60af03SYuchung Cheng void tcp_free_fastopen_req(struct tcp_sock *tp) 1098cf60af03SYuchung Cheng { 109900db4124SIan Morris if (tp->fastopen_req) { 1100cf60af03SYuchung Cheng kfree(tp->fastopen_req); 1101cf60af03SYuchung Cheng tp->fastopen_req = NULL; 1102cf60af03SYuchung Cheng } 1103cf60af03SYuchung Cheng } 1104cf60af03SYuchung Cheng 1105f5ddcbbbSEric Dumazet static int tcp_sendmsg_fastopen(struct sock *sk, struct msghdr *msg, 1106f859a448SWillem de Bruijn int *copied, size_t size, 1107f859a448SWillem de Bruijn struct ubuf_info *uarg) 1108cf60af03SYuchung Cheng { 1109cf60af03SYuchung Cheng struct tcp_sock *tp = tcp_sk(sk); 111019f6d3f3SWei Wang struct inet_sock *inet = inet_sk(sk); 1111ba615f67SWei Wang struct sockaddr *uaddr = msg->msg_name; 1112cf60af03SYuchung Cheng int err, flags; 1113cf60af03SYuchung Cheng 1114e1cfcbe8SHaishuang Yan if (!(sock_net(sk)->ipv4.sysctl_tcp_fastopen & TFO_CLIENT_ENABLE) || 1115ba615f67SWei Wang (uaddr && msg->msg_namelen >= sizeof(uaddr->sa_family) && 1116ba615f67SWei Wang uaddr->sa_family == AF_UNSPEC)) 1117cf60af03SYuchung Cheng return -EOPNOTSUPP; 111800db4124SIan Morris if (tp->fastopen_req) 1119cf60af03SYuchung Cheng return -EALREADY; /* Another Fast Open is in progress */ 1120cf60af03SYuchung Cheng 1121cf60af03SYuchung Cheng tp->fastopen_req = kzalloc(sizeof(struct tcp_fastopen_request), 1122cf60af03SYuchung Cheng sk->sk_allocation); 112351456b29SIan Morris if (unlikely(!tp->fastopen_req)) 1124cf60af03SYuchung Cheng return -ENOBUFS; 1125cf60af03SYuchung Cheng tp->fastopen_req->data = msg; 1126f5ddcbbbSEric Dumazet tp->fastopen_req->size = size; 1127f859a448SWillem de Bruijn tp->fastopen_req->uarg = uarg; 1128cf60af03SYuchung Cheng 112919f6d3f3SWei Wang if (inet->defer_connect) { 113019f6d3f3SWei Wang err = tcp_connect(sk); 113119f6d3f3SWei Wang /* Same failure procedure as in tcp_v4/6_connect */ 113219f6d3f3SWei Wang if (err) { 113319f6d3f3SWei Wang tcp_set_state(sk, TCP_CLOSE); 113419f6d3f3SWei Wang inet->inet_dport = 0; 113519f6d3f3SWei Wang sk->sk_route_caps = 0; 113619f6d3f3SWei Wang } 113719f6d3f3SWei Wang } 1138cf60af03SYuchung Cheng flags = (msg->msg_flags & MSG_DONTWAIT) ? O_NONBLOCK : 0; 1139ba615f67SWei Wang err = __inet_stream_connect(sk->sk_socket, uaddr, 11403979ad7eSWilly Tarreau msg->msg_namelen, flags, 1); 11417db92362SWei Wang /* fastopen_req could already be freed in __inet_stream_connect 11427db92362SWei Wang * if the connection times out or gets rst 11437db92362SWei Wang */ 11447db92362SWei Wang if (tp->fastopen_req) { 1145f5ddcbbbSEric Dumazet *copied = tp->fastopen_req->copied; 1146cf60af03SYuchung Cheng tcp_free_fastopen_req(tp); 11477db92362SWei Wang inet->defer_connect = 0; 11487db92362SWei Wang } 1149cf60af03SYuchung Cheng return err; 1150cf60af03SYuchung Cheng } 1151cf60af03SYuchung Cheng 1152306b13ebSTom Herbert int tcp_sendmsg_locked(struct sock *sk, struct msghdr *msg, size_t size) 11531da177e4SLinus Torvalds { 11541da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 1155f214f915SWillem de Bruijn struct ubuf_info *uarg = NULL; 11561da177e4SLinus Torvalds struct sk_buff *skb; 1157c14ac945SSoheil Hassas Yeganeh struct sockcm_cookie sockc; 115857be5bdaSAl Viro int flags, err, copied = 0; 115957be5bdaSAl Viro int mss_now = 0, size_goal, copied_syn = 0; 1160d4011239SEric Dumazet bool process_backlog = false; 116174d4a8f8SEric Dumazet bool zc = false; 11621da177e4SLinus Torvalds long timeo; 11631da177e4SLinus Torvalds 11641da177e4SLinus Torvalds flags = msg->msg_flags; 1165f214f915SWillem de Bruijn 11665cf4a853SVincent Whitchurch if (flags & MSG_ZEROCOPY && size && sock_flag(sk, SOCK_ZEROCOPY)) { 116775c119afSEric Dumazet skb = tcp_write_queue_tail(sk); 1168f214f915SWillem de Bruijn uarg = sock_zerocopy_realloc(sk, size, skb_zcopy(skb)); 1169f214f915SWillem de Bruijn if (!uarg) { 1170f214f915SWillem de Bruijn err = -ENOBUFS; 1171f214f915SWillem de Bruijn goto out_err; 1172f214f915SWillem de Bruijn } 1173f214f915SWillem de Bruijn 1174dead7cdbSEric Dumazet zc = sk->sk_route_caps & NETIF_F_SG; 117502583adeSWillem de Bruijn if (!zc) 1176f214f915SWillem de Bruijn uarg->zerocopy = 0; 1177f214f915SWillem de Bruijn } 1178f214f915SWillem de Bruijn 117916ae6aa1SYuchung Cheng if (unlikely(flags & MSG_FASTOPEN || inet_sk(sk)->defer_connect) && 118016ae6aa1SYuchung Cheng !tp->repair) { 1181f859a448SWillem de Bruijn err = tcp_sendmsg_fastopen(sk, msg, &copied_syn, size, uarg); 1182cf60af03SYuchung Cheng if (err == -EINPROGRESS && copied_syn > 0) 1183cf60af03SYuchung Cheng goto out; 1184cf60af03SYuchung Cheng else if (err) 1185cf60af03SYuchung Cheng goto out_err; 1186cf60af03SYuchung Cheng } 1187cf60af03SYuchung Cheng 11881da177e4SLinus Torvalds timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT); 11891da177e4SLinus Torvalds 1190d7722e85SSoheil Hassas Yeganeh tcp_rate_check_app_limited(sk); /* is sending application-limited? */ 1191d7722e85SSoheil Hassas Yeganeh 11928336886fSJerry Chu /* Wait for a connection to finish. One exception is TCP Fast Open 11938336886fSJerry Chu * (passive side) where data is allowed to be sent before a connection 11948336886fSJerry Chu * is fully established. 11958336886fSJerry Chu */ 11968336886fSJerry Chu if (((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) && 11978336886fSJerry Chu !tcp_passive_fastopen(sk)) { 1198686a5624SYuvaraja Mariappan err = sk_stream_wait_connect(sk, &timeo); 1199686a5624SYuvaraja Mariappan if (err != 0) 1200cf60af03SYuchung Cheng goto do_error; 12018336886fSJerry Chu } 12021da177e4SLinus Torvalds 1203c0e88ff0SPavel Emelyanov if (unlikely(tp->repair)) { 1204c0e88ff0SPavel Emelyanov if (tp->repair_queue == TCP_RECV_QUEUE) { 1205c0e88ff0SPavel Emelyanov copied = tcp_send_rcvq(sk, msg, size); 12065924f17aSChristoph Paasch goto out_nopush; 1207c0e88ff0SPavel Emelyanov } 1208c0e88ff0SPavel Emelyanov 1209c0e88ff0SPavel Emelyanov err = -EINVAL; 1210c0e88ff0SPavel Emelyanov if (tp->repair_queue == TCP_NO_QUEUE) 1211c0e88ff0SPavel Emelyanov goto out_err; 1212c0e88ff0SPavel Emelyanov 1213c0e88ff0SPavel Emelyanov /* 'common' sending to sendq */ 1214c0e88ff0SPavel Emelyanov } 1215c0e88ff0SPavel Emelyanov 1216657a0667SWillem de Bruijn sockcm_init(&sockc, sk); 1217c14ac945SSoheil Hassas Yeganeh if (msg->msg_controllen) { 1218c14ac945SSoheil Hassas Yeganeh err = sock_cmsg_send(sk, msg, &sockc); 1219c14ac945SSoheil Hassas Yeganeh if (unlikely(err)) { 1220c14ac945SSoheil Hassas Yeganeh err = -EINVAL; 1221c14ac945SSoheil Hassas Yeganeh goto out_err; 1222c14ac945SSoheil Hassas Yeganeh } 1223c14ac945SSoheil Hassas Yeganeh } 1224c14ac945SSoheil Hassas Yeganeh 12251da177e4SLinus Torvalds /* This should be in poll */ 12269cd3e072SEric Dumazet sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk); 12271da177e4SLinus Torvalds 12281da177e4SLinus Torvalds /* Ok commence sending. */ 12291da177e4SLinus Torvalds copied = 0; 12301da177e4SLinus Torvalds 1231d41a69f1SEric Dumazet restart: 1232d41a69f1SEric Dumazet mss_now = tcp_send_mss(sk, &size_goal, flags); 1233d41a69f1SEric Dumazet 12341da177e4SLinus Torvalds err = -EPIPE; 12351da177e4SLinus Torvalds if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN)) 123679d8665bSEric Dumazet goto do_error; 12371da177e4SLinus Torvalds 123801e97e65SAl Viro while (msg_data_left(msg)) { 12396828b92bSHerbert Xu int copy = 0; 12401da177e4SLinus Torvalds 1241fe067e8aSDavid S. Miller skb = tcp_write_queue_tail(sk); 124265ec6097SEric Dumazet if (skb) 124365ec6097SEric Dumazet copy = size_goal - skb->len; 12441da177e4SLinus Torvalds 1245c134ecb8SMartin KaFai Lau if (copy <= 0 || !tcp_skb_can_collapse_to(skb)) { 12463613b3dbSEric Dumazet bool first_skb; 12473613b3dbSEric Dumazet 12481da177e4SLinus Torvalds new_segment: 12491da177e4SLinus Torvalds if (!sk_stream_memory_free(sk)) 12501da177e4SLinus Torvalds goto wait_for_sndbuf; 12511da177e4SLinus Torvalds 1252d4011239SEric Dumazet if (process_backlog && sk_flush_backlog(sk)) { 1253d4011239SEric Dumazet process_backlog = false; 1254d41a69f1SEric Dumazet goto restart; 1255d4011239SEric Dumazet } 125675c119afSEric Dumazet first_skb = tcp_rtx_and_write_queues_empty(sk); 1257472c2e07SEric Dumazet skb = sk_stream_alloc_skb(sk, 0, sk->sk_allocation, 12583613b3dbSEric Dumazet first_skb); 12591da177e4SLinus Torvalds if (!skb) 12601da177e4SLinus Torvalds goto wait_for_memory; 12611da177e4SLinus Torvalds 1262d4011239SEric Dumazet process_backlog = true; 126384fa7933SPatrick McHardy skb->ip_summed = CHECKSUM_PARTIAL; 12641da177e4SLinus Torvalds 12659e412ba7SIlpo Järvinen skb_entail(sk, skb); 1266c1b4a7e6SDavid S. Miller copy = size_goal; 12679d186cacSAndrey Vagin 12689d186cacSAndrey Vagin /* All packets are restored as if they have 1269d3edd06eSEric Dumazet * already been sent. skb_mstamp_ns isn't set to 12709d186cacSAndrey Vagin * avoid wrong rtt estimation. 12719d186cacSAndrey Vagin */ 12729d186cacSAndrey Vagin if (tp->repair) 12739d186cacSAndrey Vagin TCP_SKB_CB(skb)->sacked |= TCPCB_REPAIRED; 12741da177e4SLinus Torvalds } 12751da177e4SLinus Torvalds 12761da177e4SLinus Torvalds /* Try to append data to the end of skb. */ 127701e97e65SAl Viro if (copy > msg_data_left(msg)) 127801e97e65SAl Viro copy = msg_data_left(msg); 12791da177e4SLinus Torvalds 12801da177e4SLinus Torvalds /* Where to copy to? */ 128102583adeSWillem de Bruijn if (skb_availroom(skb) > 0 && !zc) { 12821da177e4SLinus Torvalds /* We have some space in skb head. Superb! */ 1283a21d4572SEric Dumazet copy = min_t(int, copy, skb_availroom(skb)); 128457be5bdaSAl Viro err = skb_add_data_nocache(sk, skb, &msg->msg_iter, copy); 1285c6e1a0d1STom Herbert if (err) 12861da177e4SLinus Torvalds goto do_fault; 128702583adeSWillem de Bruijn } else if (!zc) { 12885640f768SEric Dumazet bool merge = true; 12891da177e4SLinus Torvalds int i = skb_shinfo(skb)->nr_frags; 12905640f768SEric Dumazet struct page_frag *pfrag = sk_page_frag(sk); 1291761965eaSEric Dumazet 12925640f768SEric Dumazet if (!sk_page_frag_refill(sk, pfrag)) 12935640f768SEric Dumazet goto wait_for_memory; 1294761965eaSEric Dumazet 12955640f768SEric Dumazet if (!skb_can_coalesce(skb, i, pfrag->page, 12965640f768SEric Dumazet pfrag->offset)) { 129774d4a8f8SEric Dumazet if (i >= sysctl_max_skb_frags) { 12981da177e4SLinus Torvalds tcp_mark_push(tp, skb); 12991da177e4SLinus Torvalds goto new_segment; 13001da177e4SLinus Torvalds } 13015640f768SEric Dumazet merge = false; 13025640f768SEric Dumazet } 1303ef015786SHerbert Xu 13045640f768SEric Dumazet copy = min_t(int, copy, pfrag->size - pfrag->offset); 1305ef015786SHerbert Xu 13063ab224beSHideo Aoki if (!sk_wmem_schedule(sk, copy)) 1307ef015786SHerbert Xu goto wait_for_memory; 13081da177e4SLinus Torvalds 130957be5bdaSAl Viro err = skb_copy_to_page_nocache(sk, &msg->msg_iter, skb, 13105640f768SEric Dumazet pfrag->page, 13115640f768SEric Dumazet pfrag->offset, 13125640f768SEric Dumazet copy); 13135640f768SEric Dumazet if (err) 13141da177e4SLinus Torvalds goto do_error; 13151da177e4SLinus Torvalds 13161da177e4SLinus Torvalds /* Update the skb. */ 13171da177e4SLinus Torvalds if (merge) { 13189e903e08SEric Dumazet skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy); 13191da177e4SLinus Torvalds } else { 13205640f768SEric Dumazet skb_fill_page_desc(skb, i, pfrag->page, 13215640f768SEric Dumazet pfrag->offset, copy); 13224e33e346SEric Dumazet page_ref_inc(pfrag->page); 13231da177e4SLinus Torvalds } 13245640f768SEric Dumazet pfrag->offset += copy; 1325f214f915SWillem de Bruijn } else { 1326f214f915SWillem de Bruijn err = skb_zerocopy_iter_stream(sk, skb, msg, copy, uarg); 1327111856c7SWillem de Bruijn if (err == -EMSGSIZE || err == -EEXIST) { 1328111856c7SWillem de Bruijn tcp_mark_push(tp, skb); 1329f214f915SWillem de Bruijn goto new_segment; 1330111856c7SWillem de Bruijn } 1331f214f915SWillem de Bruijn if (err < 0) 1332f214f915SWillem de Bruijn goto do_error; 1333f214f915SWillem de Bruijn copy = err; 13341da177e4SLinus Torvalds } 13351da177e4SLinus Torvalds 13361da177e4SLinus Torvalds if (!copied) 13374de075e0SEric Dumazet TCP_SKB_CB(skb)->tcp_flags &= ~TCPHDR_PSH; 13381da177e4SLinus Torvalds 13391da177e4SLinus Torvalds tp->write_seq += copy; 13401da177e4SLinus Torvalds TCP_SKB_CB(skb)->end_seq += copy; 1341cd7d8498SEric Dumazet tcp_skb_pcount_set(skb, 0); 13421da177e4SLinus Torvalds 13431da177e4SLinus Torvalds copied += copy; 134401e97e65SAl Viro if (!msg_data_left(msg)) { 1345c134ecb8SMartin KaFai Lau if (unlikely(flags & MSG_EOR)) 1346c134ecb8SMartin KaFai Lau TCP_SKB_CB(skb)->eor = 1; 13471da177e4SLinus Torvalds goto out; 13484ed2d765SWillem de Bruijn } 13491da177e4SLinus Torvalds 135065ec6097SEric Dumazet if (skb->len < size_goal || (flags & MSG_OOB) || unlikely(tp->repair)) 13511da177e4SLinus Torvalds continue; 13521da177e4SLinus Torvalds 13531da177e4SLinus Torvalds if (forced_push(tp)) { 13541da177e4SLinus Torvalds tcp_mark_push(tp, skb); 13559e412ba7SIlpo Järvinen __tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH); 1356fe067e8aSDavid S. Miller } else if (skb == tcp_send_head(sk)) 13571da177e4SLinus Torvalds tcp_push_one(sk, mss_now); 13581da177e4SLinus Torvalds continue; 13591da177e4SLinus Torvalds 13601da177e4SLinus Torvalds wait_for_sndbuf: 13611da177e4SLinus Torvalds set_bit(SOCK_NOSPACE, &sk->sk_socket->flags); 13621da177e4SLinus Torvalds wait_for_memory: 1363ec342325SAndrew Vagin if (copied) 1364f54b3111SEric Dumazet tcp_push(sk, flags & ~MSG_MORE, mss_now, 1365f54b3111SEric Dumazet TCP_NAGLE_PUSH, size_goal); 13661da177e4SLinus Torvalds 1367686a5624SYuvaraja Mariappan err = sk_stream_wait_memory(sk, &timeo); 1368686a5624SYuvaraja Mariappan if (err != 0) 13691da177e4SLinus Torvalds goto do_error; 13701da177e4SLinus Torvalds 13710c54b85fSIlpo Järvinen mss_now = tcp_send_mss(sk, &size_goal, flags); 13721da177e4SLinus Torvalds } 13731da177e4SLinus Torvalds 13741da177e4SLinus Torvalds out: 1375ad02c4f5SSoheil Hassas Yeganeh if (copied) { 13764e8cc228SEric Dumazet tcp_tx_timestamp(sk, sockc.tsflags); 1377f54b3111SEric Dumazet tcp_push(sk, flags, mss_now, tp->nonagle, size_goal); 1378ad02c4f5SSoheil Hassas Yeganeh } 13795924f17aSChristoph Paasch out_nopush: 1380f214f915SWillem de Bruijn sock_zerocopy_put(uarg); 1381cf60af03SYuchung Cheng return copied + copied_syn; 13821da177e4SLinus Torvalds 13831da177e4SLinus Torvalds do_fault: 13841da177e4SLinus Torvalds if (!skb->len) { 1385fe067e8aSDavid S. Miller tcp_unlink_write_queue(skb, sk); 1386fe067e8aSDavid S. Miller /* It is the one place in all of TCP, except connection 1387fe067e8aSDavid S. Miller * reset, where we can be unlinking the send_head. 1388fe067e8aSDavid S. Miller */ 13896c7b4ee7SEric Dumazet if (tcp_write_queue_empty(sk)) 13906c7b4ee7SEric Dumazet tcp_chrono_stop(sk, TCP_CHRONO_BUSY); 13913ab224beSHideo Aoki sk_wmem_free_skb(sk, skb); 13921da177e4SLinus Torvalds } 13931da177e4SLinus Torvalds 13941da177e4SLinus Torvalds do_error: 1395cf60af03SYuchung Cheng if (copied + copied_syn) 13961da177e4SLinus Torvalds goto out; 13971da177e4SLinus Torvalds out_err: 139852900d22SWillem de Bruijn sock_zerocopy_put_abort(uarg, true); 13991da177e4SLinus Torvalds err = sk_stream_error(sk, flags, err); 1400ce5ec440SJason Baron /* make sure we wake any epoll edge trigger waiter */ 1401b0f71bd3SFrancis Yan if (unlikely(skb_queue_len(&sk->sk_write_queue) == 0 && 1402b0f71bd3SFrancis Yan err == -EAGAIN)) { 1403ce5ec440SJason Baron sk->sk_write_space(sk); 1404b0f71bd3SFrancis Yan tcp_chrono_stop(sk, TCP_CHRONO_SNDBUF_LIMITED); 1405b0f71bd3SFrancis Yan } 14061da177e4SLinus Torvalds return err; 14071da177e4SLinus Torvalds } 1408774c4673SDavid S. Miller EXPORT_SYMBOL_GPL(tcp_sendmsg_locked); 1409306b13ebSTom Herbert 1410306b13ebSTom Herbert int tcp_sendmsg(struct sock *sk, struct msghdr *msg, size_t size) 1411306b13ebSTom Herbert { 1412306b13ebSTom Herbert int ret; 1413306b13ebSTom Herbert 1414306b13ebSTom Herbert lock_sock(sk); 1415306b13ebSTom Herbert ret = tcp_sendmsg_locked(sk, msg, size); 1416306b13ebSTom Herbert release_sock(sk); 1417306b13ebSTom Herbert 1418306b13ebSTom Herbert return ret; 1419306b13ebSTom Herbert } 14204bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_sendmsg); 14211da177e4SLinus Torvalds 14221da177e4SLinus Torvalds /* 14231da177e4SLinus Torvalds * Handle reading urgent data. BSD has very simple semantics for 14241da177e4SLinus Torvalds * this, no blocking and very strange errors 8) 14251da177e4SLinus Torvalds */ 14261da177e4SLinus Torvalds 1427377f0a08SRami Rosen static int tcp_recv_urg(struct sock *sk, struct msghdr *msg, int len, int flags) 14281da177e4SLinus Torvalds { 14291da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 14301da177e4SLinus Torvalds 14311da177e4SLinus Torvalds /* No URG data to read. */ 14321da177e4SLinus Torvalds if (sock_flag(sk, SOCK_URGINLINE) || !tp->urg_data || 14331da177e4SLinus Torvalds tp->urg_data == TCP_URG_READ) 14341da177e4SLinus Torvalds return -EINVAL; /* Yes this is right ! */ 14351da177e4SLinus Torvalds 14361da177e4SLinus Torvalds if (sk->sk_state == TCP_CLOSE && !sock_flag(sk, SOCK_DONE)) 14371da177e4SLinus Torvalds return -ENOTCONN; 14381da177e4SLinus Torvalds 14391da177e4SLinus Torvalds if (tp->urg_data & TCP_URG_VALID) { 14401da177e4SLinus Torvalds int err = 0; 14411da177e4SLinus Torvalds char c = tp->urg_data; 14421da177e4SLinus Torvalds 14431da177e4SLinus Torvalds if (!(flags & MSG_PEEK)) 14441da177e4SLinus Torvalds tp->urg_data = TCP_URG_READ; 14451da177e4SLinus Torvalds 14461da177e4SLinus Torvalds /* Read urgent data. */ 14471da177e4SLinus Torvalds msg->msg_flags |= MSG_OOB; 14481da177e4SLinus Torvalds 14491da177e4SLinus Torvalds if (len > 0) { 14501da177e4SLinus Torvalds if (!(flags & MSG_TRUNC)) 14517eab8d9eSAl Viro err = memcpy_to_msg(msg, &c, 1); 14521da177e4SLinus Torvalds len = 1; 14531da177e4SLinus Torvalds } else 14541da177e4SLinus Torvalds msg->msg_flags |= MSG_TRUNC; 14551da177e4SLinus Torvalds 14561da177e4SLinus Torvalds return err ? -EFAULT : len; 14571da177e4SLinus Torvalds } 14581da177e4SLinus Torvalds 14591da177e4SLinus Torvalds if (sk->sk_state == TCP_CLOSE || (sk->sk_shutdown & RCV_SHUTDOWN)) 14601da177e4SLinus Torvalds return 0; 14611da177e4SLinus Torvalds 14621da177e4SLinus Torvalds /* Fixed the recv(..., MSG_OOB) behaviour. BSD docs and 14631da177e4SLinus Torvalds * the available implementations agree in this case: 14641da177e4SLinus Torvalds * this call should never block, independent of the 14651da177e4SLinus Torvalds * blocking state of the socket. 14661da177e4SLinus Torvalds * Mike <pall@rz.uni-karlsruhe.de> 14671da177e4SLinus Torvalds */ 14681da177e4SLinus Torvalds return -EAGAIN; 14691da177e4SLinus Torvalds } 14701da177e4SLinus Torvalds 1471c0e88ff0SPavel Emelyanov static int tcp_peek_sndq(struct sock *sk, struct msghdr *msg, int len) 1472c0e88ff0SPavel Emelyanov { 1473c0e88ff0SPavel Emelyanov struct sk_buff *skb; 1474c0e88ff0SPavel Emelyanov int copied = 0, err = 0; 1475c0e88ff0SPavel Emelyanov 1476c0e88ff0SPavel Emelyanov /* XXX -- need to support SO_PEEK_OFF */ 1477c0e88ff0SPavel Emelyanov 147875c119afSEric Dumazet skb_rbtree_walk(skb, &sk->tcp_rtx_queue) { 147975c119afSEric Dumazet err = skb_copy_datagram_msg(skb, 0, msg, skb->len); 148075c119afSEric Dumazet if (err) 148175c119afSEric Dumazet return err; 148275c119afSEric Dumazet copied += skb->len; 148375c119afSEric Dumazet } 148475c119afSEric Dumazet 1485c0e88ff0SPavel Emelyanov skb_queue_walk(&sk->sk_write_queue, skb) { 148651f3d02bSDavid S. Miller err = skb_copy_datagram_msg(skb, 0, msg, skb->len); 1487c0e88ff0SPavel Emelyanov if (err) 1488c0e88ff0SPavel Emelyanov break; 1489c0e88ff0SPavel Emelyanov 1490c0e88ff0SPavel Emelyanov copied += skb->len; 1491c0e88ff0SPavel Emelyanov } 1492c0e88ff0SPavel Emelyanov 1493c0e88ff0SPavel Emelyanov return err ?: copied; 1494c0e88ff0SPavel Emelyanov } 1495c0e88ff0SPavel Emelyanov 14961da177e4SLinus Torvalds /* Clean up the receive buffer for full frames taken by the user, 14971da177e4SLinus Torvalds * then send an ACK if necessary. COPIED is the number of bytes 14981da177e4SLinus Torvalds * tcp_recvmsg has given to the user so far, it speeds up the 14991da177e4SLinus Torvalds * calculation of whether or not we must ACK for the sake of 15001da177e4SLinus Torvalds * a window update. 15011da177e4SLinus Torvalds */ 15023f334078SDan Williams static void tcp_cleanup_rbuf(struct sock *sk, int copied) 15031da177e4SLinus Torvalds { 15041da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 1505a2a385d6SEric Dumazet bool time_to_ack = false; 15061da177e4SLinus Torvalds 15071da177e4SLinus Torvalds struct sk_buff *skb = skb_peek(&sk->sk_receive_queue); 15081da177e4SLinus Torvalds 1509d792c100SIlpo Järvinen WARN(skb && !before(tp->copied_seq, TCP_SKB_CB(skb)->end_seq), 15102af6fd8bSJoe Perches "cleanup rbuf bug: copied %X seq %X rcvnxt %X\n", 1511d792c100SIlpo Järvinen tp->copied_seq, TCP_SKB_CB(skb)->end_seq, tp->rcv_nxt); 15121da177e4SLinus Torvalds 1513463c84b9SArnaldo Carvalho de Melo if (inet_csk_ack_scheduled(sk)) { 1514463c84b9SArnaldo Carvalho de Melo const struct inet_connection_sock *icsk = inet_csk(sk); 15151da177e4SLinus Torvalds /* Delayed ACKs frequently hit locked sockets during bulk 15161da177e4SLinus Torvalds * receive. */ 1517463c84b9SArnaldo Carvalho de Melo if (icsk->icsk_ack.blocked || 15181da177e4SLinus Torvalds /* Once-per-two-segments ACK was not sent by tcp_input.c */ 1519463c84b9SArnaldo Carvalho de Melo tp->rcv_nxt - tp->rcv_wup > icsk->icsk_ack.rcv_mss || 15201da177e4SLinus Torvalds /* 15211da177e4SLinus Torvalds * If this read emptied read buffer, we send ACK, if 15221da177e4SLinus Torvalds * connection is not bidirectional, user drained 15231da177e4SLinus Torvalds * receive buffer and there was a small segment 15241da177e4SLinus Torvalds * in queue. 15251da177e4SLinus Torvalds */ 15261ef9696cSAlexey Kuznetsov (copied > 0 && 15271ef9696cSAlexey Kuznetsov ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED2) || 15281ef9696cSAlexey Kuznetsov ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED) && 152931954cd8SWei Wang !inet_csk_in_pingpong_mode(sk))) && 15301ef9696cSAlexey Kuznetsov !atomic_read(&sk->sk_rmem_alloc))) 1531a2a385d6SEric Dumazet time_to_ack = true; 15321da177e4SLinus Torvalds } 15331da177e4SLinus Torvalds 15341da177e4SLinus Torvalds /* We send an ACK if we can now advertise a non-zero window 15351da177e4SLinus Torvalds * which has been raised "significantly". 15361da177e4SLinus Torvalds * 15371da177e4SLinus Torvalds * Even if window raised up to infinity, do not send window open ACK 15381da177e4SLinus Torvalds * in states, where we will not receive more. It is useless. 15391da177e4SLinus Torvalds */ 15401da177e4SLinus Torvalds if (copied > 0 && !time_to_ack && !(sk->sk_shutdown & RCV_SHUTDOWN)) { 15411da177e4SLinus Torvalds __u32 rcv_window_now = tcp_receive_window(tp); 15421da177e4SLinus Torvalds 15431da177e4SLinus Torvalds /* Optimize, __tcp_select_window() is not cheap. */ 15441da177e4SLinus Torvalds if (2*rcv_window_now <= tp->window_clamp) { 15451da177e4SLinus Torvalds __u32 new_window = __tcp_select_window(sk); 15461da177e4SLinus Torvalds 15471da177e4SLinus Torvalds /* Send ACK now, if this read freed lots of space 15481da177e4SLinus Torvalds * in our buffer. Certainly, new_window is new window. 15491da177e4SLinus Torvalds * We can advertise it now, if it is not less than current one. 15501da177e4SLinus Torvalds * "Lots" means "at least twice" here. 15511da177e4SLinus Torvalds */ 15521da177e4SLinus Torvalds if (new_window && new_window >= 2 * rcv_window_now) 1553a2a385d6SEric Dumazet time_to_ack = true; 15541da177e4SLinus Torvalds } 15551da177e4SLinus Torvalds } 15561da177e4SLinus Torvalds if (time_to_ack) 15571da177e4SLinus Torvalds tcp_send_ack(sk); 15581da177e4SLinus Torvalds } 15591da177e4SLinus Torvalds 1560f26845b4SEric Dumazet static struct sk_buff *tcp_recv_skb(struct sock *sk, u32 seq, u32 *off) 15611da177e4SLinus Torvalds { 15621da177e4SLinus Torvalds struct sk_buff *skb; 15631da177e4SLinus Torvalds u32 offset; 15641da177e4SLinus Torvalds 1565f26845b4SEric Dumazet while ((skb = skb_peek(&sk->sk_receive_queue)) != NULL) { 15661da177e4SLinus Torvalds offset = seq - TCP_SKB_CB(skb)->seq; 15679d691539SEric Dumazet if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) { 15689d691539SEric Dumazet pr_err_once("%s: found a SYN, please report !\n", __func__); 15691da177e4SLinus Torvalds offset--; 15709d691539SEric Dumazet } 1571e11ecddfSEric Dumazet if (offset < skb->len || (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN)) { 15721da177e4SLinus Torvalds *off = offset; 15731da177e4SLinus Torvalds return skb; 15741da177e4SLinus Torvalds } 1575f26845b4SEric Dumazet /* This looks weird, but this can happen if TCP collapsing 1576f26845b4SEric Dumazet * splitted a fat GRO packet, while we released socket lock 1577f26845b4SEric Dumazet * in skb_splice_bits() 1578f26845b4SEric Dumazet */ 15797bced397SDan Williams sk_eat_skb(sk, skb); 15801da177e4SLinus Torvalds } 15811da177e4SLinus Torvalds return NULL; 15821da177e4SLinus Torvalds } 15831da177e4SLinus Torvalds 15841da177e4SLinus Torvalds /* 15851da177e4SLinus Torvalds * This routine provides an alternative to tcp_recvmsg() for routines 15861da177e4SLinus Torvalds * that would like to handle copying from skbuffs directly in 'sendfile' 15871da177e4SLinus Torvalds * fashion. 15881da177e4SLinus Torvalds * Note: 15891da177e4SLinus Torvalds * - It is assumed that the socket was locked by the caller. 15901da177e4SLinus Torvalds * - The routine does not block. 15911da177e4SLinus Torvalds * - At present, there is no support for reading OOB data 15921da177e4SLinus Torvalds * or for 'peeking' the socket using this routine 15931da177e4SLinus Torvalds * (although both would be easy to implement). 15941da177e4SLinus Torvalds */ 15951da177e4SLinus Torvalds int tcp_read_sock(struct sock *sk, read_descriptor_t *desc, 15961da177e4SLinus Torvalds sk_read_actor_t recv_actor) 15971da177e4SLinus Torvalds { 15981da177e4SLinus Torvalds struct sk_buff *skb; 15991da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 16001da177e4SLinus Torvalds u32 seq = tp->copied_seq; 16011da177e4SLinus Torvalds u32 offset; 16021da177e4SLinus Torvalds int copied = 0; 16031da177e4SLinus Torvalds 16041da177e4SLinus Torvalds if (sk->sk_state == TCP_LISTEN) 16051da177e4SLinus Torvalds return -ENOTCONN; 16061da177e4SLinus Torvalds while ((skb = tcp_recv_skb(sk, seq, &offset)) != NULL) { 16071da177e4SLinus Torvalds if (offset < skb->len) { 1608374e7b59SOctavian Purdila int used; 1609374e7b59SOctavian Purdila size_t len; 16101da177e4SLinus Torvalds 16111da177e4SLinus Torvalds len = skb->len - offset; 16121da177e4SLinus Torvalds /* Stop reading if we hit a patch of urgent data */ 16131da177e4SLinus Torvalds if (tp->urg_data) { 16141da177e4SLinus Torvalds u32 urg_offset = tp->urg_seq - seq; 16151da177e4SLinus Torvalds if (urg_offset < len) 16161da177e4SLinus Torvalds len = urg_offset; 16171da177e4SLinus Torvalds if (!len) 16181da177e4SLinus Torvalds break; 16191da177e4SLinus Torvalds } 16201da177e4SLinus Torvalds used = recv_actor(desc, skb, offset, len); 1621ff905b1eSEric Dumazet if (used <= 0) { 1622ddb61a57SJens Axboe if (!copied) 1623ddb61a57SJens Axboe copied = used; 1624ddb61a57SJens Axboe break; 1625ddb61a57SJens Axboe } else if (used <= len) { 16261da177e4SLinus Torvalds seq += used; 16271da177e4SLinus Torvalds copied += used; 16281da177e4SLinus Torvalds offset += used; 16291da177e4SLinus Torvalds } 163002275a2eSWilly Tarreau /* If recv_actor drops the lock (e.g. TCP splice 1631293ad604SOctavian Purdila * receive) the skb pointer might be invalid when 1632293ad604SOctavian Purdila * getting here: tcp_collapse might have deleted it 1633293ad604SOctavian Purdila * while aggregating skbs from the socket queue. 1634293ad604SOctavian Purdila */ 1635293ad604SOctavian Purdila skb = tcp_recv_skb(sk, seq - 1, &offset); 163602275a2eSWilly Tarreau if (!skb) 16371da177e4SLinus Torvalds break; 163802275a2eSWilly Tarreau /* TCP coalescing might have appended data to the skb. 163902275a2eSWilly Tarreau * Try to splice more frags 164002275a2eSWilly Tarreau */ 164102275a2eSWilly Tarreau if (offset + 1 != skb->len) 164202275a2eSWilly Tarreau continue; 16431da177e4SLinus Torvalds } 1644e11ecddfSEric Dumazet if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) { 16457bced397SDan Williams sk_eat_skb(sk, skb); 16461da177e4SLinus Torvalds ++seq; 16471da177e4SLinus Torvalds break; 16481da177e4SLinus Torvalds } 16497bced397SDan Williams sk_eat_skb(sk, skb); 16501da177e4SLinus Torvalds if (!desc->count) 16511da177e4SLinus Torvalds break; 1652baff42abSSteven J. Magnani tp->copied_seq = seq; 16531da177e4SLinus Torvalds } 16541da177e4SLinus Torvalds tp->copied_seq = seq; 16551da177e4SLinus Torvalds 16561da177e4SLinus Torvalds tcp_rcv_space_adjust(sk); 16571da177e4SLinus Torvalds 16581da177e4SLinus Torvalds /* Clean up data we have read: This will do ACK frames. */ 1659f26845b4SEric Dumazet if (copied > 0) { 1660f26845b4SEric Dumazet tcp_recv_skb(sk, seq, &offset); 16610e4b4992SChris Leech tcp_cleanup_rbuf(sk, copied); 1662f26845b4SEric Dumazet } 16631da177e4SLinus Torvalds return copied; 16641da177e4SLinus Torvalds } 16654bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_read_sock); 16661da177e4SLinus Torvalds 166732035585STom Herbert int tcp_peek_len(struct socket *sock) 166832035585STom Herbert { 166932035585STom Herbert return tcp_inq(sock->sk); 167032035585STom Herbert } 167132035585STom Herbert EXPORT_SYMBOL(tcp_peek_len); 167232035585STom Herbert 1673d1361840SEric Dumazet /* Make sure sk_rcvbuf is big enough to satisfy SO_RCVLOWAT hint */ 1674d1361840SEric Dumazet int tcp_set_rcvlowat(struct sock *sk, int val) 1675d1361840SEric Dumazet { 1676867f816bSSoheil Hassas Yeganeh int cap; 1677867f816bSSoheil Hassas Yeganeh 1678867f816bSSoheil Hassas Yeganeh if (sk->sk_userlocks & SOCK_RCVBUF_LOCK) 1679867f816bSSoheil Hassas Yeganeh cap = sk->sk_rcvbuf >> 1; 1680867f816bSSoheil Hassas Yeganeh else 1681867f816bSSoheil Hassas Yeganeh cap = sock_net(sk)->ipv4.sysctl_tcp_rmem[2] >> 1; 1682867f816bSSoheil Hassas Yeganeh val = min(val, cap); 1683d1361840SEric Dumazet sk->sk_rcvlowat = val ? : 1; 168403f45c88SEric Dumazet 168503f45c88SEric Dumazet /* Check if we need to signal EPOLLIN right now */ 168603f45c88SEric Dumazet tcp_data_ready(sk); 168703f45c88SEric Dumazet 1688d1361840SEric Dumazet if (sk->sk_userlocks & SOCK_RCVBUF_LOCK) 1689d1361840SEric Dumazet return 0; 1690d1361840SEric Dumazet 1691d1361840SEric Dumazet val <<= 1; 1692d1361840SEric Dumazet if (val > sk->sk_rcvbuf) { 1693d1361840SEric Dumazet sk->sk_rcvbuf = val; 1694d1361840SEric Dumazet tcp_sk(sk)->window_clamp = tcp_win_from_space(sk, val); 1695d1361840SEric Dumazet } 1696d1361840SEric Dumazet return 0; 1697d1361840SEric Dumazet } 1698d1361840SEric Dumazet EXPORT_SYMBOL(tcp_set_rcvlowat); 1699d1361840SEric Dumazet 170005255b82SEric Dumazet #ifdef CONFIG_MMU 170105255b82SEric Dumazet static const struct vm_operations_struct tcp_vm_ops = { 170205255b82SEric Dumazet }; 170305255b82SEric Dumazet 170493ab6cc6SEric Dumazet int tcp_mmap(struct file *file, struct socket *sock, 170593ab6cc6SEric Dumazet struct vm_area_struct *vma) 170693ab6cc6SEric Dumazet { 170705255b82SEric Dumazet if (vma->vm_flags & (VM_WRITE | VM_EXEC)) 170805255b82SEric Dumazet return -EPERM; 170905255b82SEric Dumazet vma->vm_flags &= ~(VM_MAYWRITE | VM_MAYEXEC); 171005255b82SEric Dumazet 171105255b82SEric Dumazet /* Instruct vm_insert_page() to not down_read(mmap_sem) */ 171205255b82SEric Dumazet vma->vm_flags |= VM_MIXEDMAP; 171305255b82SEric Dumazet 171405255b82SEric Dumazet vma->vm_ops = &tcp_vm_ops; 171505255b82SEric Dumazet return 0; 171605255b82SEric Dumazet } 171705255b82SEric Dumazet EXPORT_SYMBOL(tcp_mmap); 171805255b82SEric Dumazet 171905255b82SEric Dumazet static int tcp_zerocopy_receive(struct sock *sk, 172005255b82SEric Dumazet struct tcp_zerocopy_receive *zc) 172105255b82SEric Dumazet { 172205255b82SEric Dumazet unsigned long address = (unsigned long)zc->address; 172305255b82SEric Dumazet const skb_frag_t *frags = NULL; 172405255b82SEric Dumazet u32 length = 0, seq, offset; 172505255b82SEric Dumazet struct vm_area_struct *vma; 172605255b82SEric Dumazet struct sk_buff *skb = NULL; 172793ab6cc6SEric Dumazet struct tcp_sock *tp; 17288f2b0293SSoheil Hassas Yeganeh int inq; 172993ab6cc6SEric Dumazet int ret; 173093ab6cc6SEric Dumazet 173105255b82SEric Dumazet if (address & (PAGE_SIZE - 1) || address != zc->address) 173293ab6cc6SEric Dumazet return -EINVAL; 173393ab6cc6SEric Dumazet 173493ab6cc6SEric Dumazet if (sk->sk_state == TCP_LISTEN) 173505255b82SEric Dumazet return -ENOTCONN; 173693ab6cc6SEric Dumazet 173793ab6cc6SEric Dumazet sock_rps_record_flow(sk); 173893ab6cc6SEric Dumazet 173905255b82SEric Dumazet down_read(¤t->mm->mmap_sem); 174093ab6cc6SEric Dumazet 174193ab6cc6SEric Dumazet ret = -EINVAL; 174205255b82SEric Dumazet vma = find_vma(current->mm, address); 174305255b82SEric Dumazet if (!vma || vma->vm_start > address || vma->vm_ops != &tcp_vm_ops) 174493ab6cc6SEric Dumazet goto out; 174505255b82SEric Dumazet zc->length = min_t(unsigned long, zc->length, vma->vm_end - address); 174605255b82SEric Dumazet 174705255b82SEric Dumazet tp = tcp_sk(sk); 174805255b82SEric Dumazet seq = tp->copied_seq; 17498f2b0293SSoheil Hassas Yeganeh inq = tcp_inq(sk); 17508f2b0293SSoheil Hassas Yeganeh zc->length = min_t(u32, zc->length, inq); 175105255b82SEric Dumazet zc->length &= ~(PAGE_SIZE - 1); 17528f2b0293SSoheil Hassas Yeganeh if (zc->length) { 175305255b82SEric Dumazet zap_page_range(vma, address, zc->length); 175405255b82SEric Dumazet zc->recv_skip_hint = 0; 17558f2b0293SSoheil Hassas Yeganeh } else { 17568f2b0293SSoheil Hassas Yeganeh zc->recv_skip_hint = inq; 17578f2b0293SSoheil Hassas Yeganeh } 175805255b82SEric Dumazet ret = 0; 175905255b82SEric Dumazet while (length + PAGE_SIZE <= zc->length) { 176005255b82SEric Dumazet if (zc->recv_skip_hint < PAGE_SIZE) { 176105255b82SEric Dumazet if (skb) { 176205255b82SEric Dumazet skb = skb->next; 176305255b82SEric Dumazet offset = seq - TCP_SKB_CB(skb)->seq; 176405255b82SEric Dumazet } else { 176593ab6cc6SEric Dumazet skb = tcp_recv_skb(sk, seq, &offset); 176605255b82SEric Dumazet } 176705255b82SEric Dumazet 176805255b82SEric Dumazet zc->recv_skip_hint = skb->len - offset; 176993ab6cc6SEric Dumazet offset -= skb_headlen(skb); 177005255b82SEric Dumazet if ((int)offset < 0 || skb_has_frag_list(skb)) 177105255b82SEric Dumazet break; 177293ab6cc6SEric Dumazet frags = skb_shinfo(skb)->frags; 177393ab6cc6SEric Dumazet while (offset) { 177493ab6cc6SEric Dumazet if (frags->size > offset) 177593ab6cc6SEric Dumazet goto out; 177693ab6cc6SEric Dumazet offset -= frags->size; 177793ab6cc6SEric Dumazet frags++; 177893ab6cc6SEric Dumazet } 177905255b82SEric Dumazet } 1780789762ceSSoheil Hassas Yeganeh if (frags->size != PAGE_SIZE || frags->page_offset) { 1781789762ceSSoheil Hassas Yeganeh int remaining = zc->recv_skip_hint; 1782789762ceSSoheil Hassas Yeganeh 1783789762ceSSoheil Hassas Yeganeh while (remaining && (frags->size != PAGE_SIZE || 1784789762ceSSoheil Hassas Yeganeh frags->page_offset)) { 1785789762ceSSoheil Hassas Yeganeh remaining -= frags->size; 1786789762ceSSoheil Hassas Yeganeh frags++; 1787789762ceSSoheil Hassas Yeganeh } 1788789762ceSSoheil Hassas Yeganeh zc->recv_skip_hint -= remaining; 178905255b82SEric Dumazet break; 1790789762ceSSoheil Hassas Yeganeh } 179105255b82SEric Dumazet ret = vm_insert_page(vma, address + length, 179205255b82SEric Dumazet skb_frag_page(frags)); 179393ab6cc6SEric Dumazet if (ret) 179405255b82SEric Dumazet break; 179505255b82SEric Dumazet length += PAGE_SIZE; 179605255b82SEric Dumazet seq += PAGE_SIZE; 179705255b82SEric Dumazet zc->recv_skip_hint -= PAGE_SIZE; 179805255b82SEric Dumazet frags++; 179993ab6cc6SEric Dumazet } 180005255b82SEric Dumazet out: 180105255b82SEric Dumazet up_read(¤t->mm->mmap_sem); 180205255b82SEric Dumazet if (length) { 180393ab6cc6SEric Dumazet tp->copied_seq = seq; 180493ab6cc6SEric Dumazet tcp_rcv_space_adjust(sk); 180593ab6cc6SEric Dumazet 180693ab6cc6SEric Dumazet /* Clean up data we have read: This will do ACK frames. */ 180793ab6cc6SEric Dumazet tcp_recv_skb(sk, seq, &offset); 180805255b82SEric Dumazet tcp_cleanup_rbuf(sk, length); 180993ab6cc6SEric Dumazet ret = 0; 181005255b82SEric Dumazet if (length == zc->length) 181105255b82SEric Dumazet zc->recv_skip_hint = 0; 181205255b82SEric Dumazet } else { 181305255b82SEric Dumazet if (!zc->recv_skip_hint && sock_flag(sk, SOCK_DONE)) 181405255b82SEric Dumazet ret = -EIO; 181505255b82SEric Dumazet } 181605255b82SEric Dumazet zc->length = length; 181793ab6cc6SEric Dumazet return ret; 181893ab6cc6SEric Dumazet } 181905255b82SEric Dumazet #endif 182093ab6cc6SEric Dumazet 182198aaa913SMike Maloney static void tcp_update_recv_tstamps(struct sk_buff *skb, 18229718475eSDeepa Dinamani struct scm_timestamping_internal *tss) 182398aaa913SMike Maloney { 182498aaa913SMike Maloney if (skb->tstamp) 18259718475eSDeepa Dinamani tss->ts[0] = ktime_to_timespec64(skb->tstamp); 182698aaa913SMike Maloney else 18279718475eSDeepa Dinamani tss->ts[0] = (struct timespec64) {0}; 182898aaa913SMike Maloney 182998aaa913SMike Maloney if (skb_hwtstamps(skb)->hwtstamp) 18309718475eSDeepa Dinamani tss->ts[2] = ktime_to_timespec64(skb_hwtstamps(skb)->hwtstamp); 183198aaa913SMike Maloney else 18329718475eSDeepa Dinamani tss->ts[2] = (struct timespec64) {0}; 183398aaa913SMike Maloney } 183498aaa913SMike Maloney 183598aaa913SMike Maloney /* Similar to __sock_recv_timestamp, but does not require an skb */ 1836809a79e9SWei Yongjun static void tcp_recv_timestamp(struct msghdr *msg, const struct sock *sk, 18379718475eSDeepa Dinamani struct scm_timestamping_internal *tss) 183898aaa913SMike Maloney { 1839887feae3SDeepa Dinamani int new_tstamp = sock_flag(sk, SOCK_TSTAMP_NEW); 184098aaa913SMike Maloney bool has_timestamping = false; 184198aaa913SMike Maloney 184298aaa913SMike Maloney if (tss->ts[0].tv_sec || tss->ts[0].tv_nsec) { 184398aaa913SMike Maloney if (sock_flag(sk, SOCK_RCVTSTAMP)) { 184498aaa913SMike Maloney if (sock_flag(sk, SOCK_RCVTSTAMPNS)) { 1845887feae3SDeepa Dinamani if (new_tstamp) { 1846887feae3SDeepa Dinamani struct __kernel_timespec kts = {tss->ts[0].tv_sec, tss->ts[0].tv_nsec}; 1847887feae3SDeepa Dinamani 1848887feae3SDeepa Dinamani put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_NEW, 1849887feae3SDeepa Dinamani sizeof(kts), &kts); 1850887feae3SDeepa Dinamani } else { 18519718475eSDeepa Dinamani struct timespec ts_old = timespec64_to_timespec(tss->ts[0]); 18529718475eSDeepa Dinamani 18537f1bc6e9SDeepa Dinamani put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMPNS_OLD, 18549718475eSDeepa Dinamani sizeof(ts_old), &ts_old); 1855887feae3SDeepa Dinamani } 185698aaa913SMike Maloney } else { 1857887feae3SDeepa Dinamani if (new_tstamp) { 1858887feae3SDeepa Dinamani struct __kernel_sock_timeval stv; 1859887feae3SDeepa Dinamani 1860887feae3SDeepa Dinamani stv.tv_sec = tss->ts[0].tv_sec; 1861887feae3SDeepa Dinamani stv.tv_usec = tss->ts[0].tv_nsec / 1000; 1862887feae3SDeepa Dinamani put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_NEW, 1863887feae3SDeepa Dinamani sizeof(stv), &stv); 1864887feae3SDeepa Dinamani } else { 1865887feae3SDeepa Dinamani struct __kernel_old_timeval tv; 1866887feae3SDeepa Dinamani 186798aaa913SMike Maloney tv.tv_sec = tss->ts[0].tv_sec; 186898aaa913SMike Maloney tv.tv_usec = tss->ts[0].tv_nsec / 1000; 18697f1bc6e9SDeepa Dinamani put_cmsg(msg, SOL_SOCKET, SO_TIMESTAMP_OLD, 187098aaa913SMike Maloney sizeof(tv), &tv); 187198aaa913SMike Maloney } 187298aaa913SMike Maloney } 1873887feae3SDeepa Dinamani } 187498aaa913SMike Maloney 187598aaa913SMike Maloney if (sk->sk_tsflags & SOF_TIMESTAMPING_SOFTWARE) 187698aaa913SMike Maloney has_timestamping = true; 187798aaa913SMike Maloney else 18789718475eSDeepa Dinamani tss->ts[0] = (struct timespec64) {0}; 187998aaa913SMike Maloney } 188098aaa913SMike Maloney 188198aaa913SMike Maloney if (tss->ts[2].tv_sec || tss->ts[2].tv_nsec) { 188298aaa913SMike Maloney if (sk->sk_tsflags & SOF_TIMESTAMPING_RAW_HARDWARE) 188398aaa913SMike Maloney has_timestamping = true; 188498aaa913SMike Maloney else 18859718475eSDeepa Dinamani tss->ts[2] = (struct timespec64) {0}; 188698aaa913SMike Maloney } 188798aaa913SMike Maloney 188898aaa913SMike Maloney if (has_timestamping) { 18899718475eSDeepa Dinamani tss->ts[1] = (struct timespec64) {0}; 18909718475eSDeepa Dinamani if (sock_flag(sk, SOCK_TSTAMP_NEW)) 18919718475eSDeepa Dinamani put_cmsg_scm_timestamping64(msg, tss); 18929718475eSDeepa Dinamani else 18939718475eSDeepa Dinamani put_cmsg_scm_timestamping(msg, tss); 189498aaa913SMike Maloney } 189598aaa913SMike Maloney } 189698aaa913SMike Maloney 1897b75eba76SSoheil Hassas Yeganeh static int tcp_inq_hint(struct sock *sk) 1898b75eba76SSoheil Hassas Yeganeh { 1899b75eba76SSoheil Hassas Yeganeh const struct tcp_sock *tp = tcp_sk(sk); 1900b75eba76SSoheil Hassas Yeganeh u32 copied_seq = READ_ONCE(tp->copied_seq); 1901b75eba76SSoheil Hassas Yeganeh u32 rcv_nxt = READ_ONCE(tp->rcv_nxt); 1902b75eba76SSoheil Hassas Yeganeh int inq; 1903b75eba76SSoheil Hassas Yeganeh 1904b75eba76SSoheil Hassas Yeganeh inq = rcv_nxt - copied_seq; 1905b75eba76SSoheil Hassas Yeganeh if (unlikely(inq < 0 || copied_seq != READ_ONCE(tp->copied_seq))) { 1906b75eba76SSoheil Hassas Yeganeh lock_sock(sk); 1907b75eba76SSoheil Hassas Yeganeh inq = tp->rcv_nxt - tp->copied_seq; 1908b75eba76SSoheil Hassas Yeganeh release_sock(sk); 1909b75eba76SSoheil Hassas Yeganeh } 19106466e715SSoheil Hassas Yeganeh /* After receiving a FIN, tell the user-space to continue reading 19116466e715SSoheil Hassas Yeganeh * by returning a non-zero inq. 19126466e715SSoheil Hassas Yeganeh */ 19136466e715SSoheil Hassas Yeganeh if (inq == 0 && sock_flag(sk, SOCK_DONE)) 19146466e715SSoheil Hassas Yeganeh inq = 1; 1915b75eba76SSoheil Hassas Yeganeh return inq; 1916b75eba76SSoheil Hassas Yeganeh } 1917b75eba76SSoheil Hassas Yeganeh 19181da177e4SLinus Torvalds /* 19191da177e4SLinus Torvalds * This routine copies from a sock struct into the user buffer. 19201da177e4SLinus Torvalds * 19211da177e4SLinus Torvalds * Technical note: in 2.3 we work on _locked_ socket, so that 19221da177e4SLinus Torvalds * tricks with *seq access order and skb->users are not required. 19231da177e4SLinus Torvalds * Probably, code can be easily improved even more. 19241da177e4SLinus Torvalds */ 19251da177e4SLinus Torvalds 19261b784140SYing Xue int tcp_recvmsg(struct sock *sk, struct msghdr *msg, size_t len, int nonblock, 19271b784140SYing Xue int flags, int *addr_len) 19281da177e4SLinus Torvalds { 19291da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 19301da177e4SLinus Torvalds int copied = 0; 19311da177e4SLinus Torvalds u32 peek_seq; 19321da177e4SLinus Torvalds u32 *seq; 19331da177e4SLinus Torvalds unsigned long used; 1934b75eba76SSoheil Hassas Yeganeh int err, inq; 19351da177e4SLinus Torvalds int target; /* Read at least this many bytes */ 19361da177e4SLinus Torvalds long timeo; 1937dfbafc99SSabrina Dubroca struct sk_buff *skb, *last; 193877527313SIlpo Järvinen u32 urg_hole = 0; 19399718475eSDeepa Dinamani struct scm_timestamping_internal tss; 194098aaa913SMike Maloney bool has_tss = false; 1941b75eba76SSoheil Hassas Yeganeh bool has_cmsg; 19421da177e4SLinus Torvalds 19434ed2d765SWillem de Bruijn if (unlikely(flags & MSG_ERRQUEUE)) 1944f4713a3dSWillem de Bruijn return inet_recv_error(sk, msg, len, addr_len); 19454ed2d765SWillem de Bruijn 1946cbf55001SEliezer Tamir if (sk_can_busy_loop(sk) && skb_queue_empty(&sk->sk_receive_queue) && 1947cbf55001SEliezer Tamir (sk->sk_state == TCP_ESTABLISHED)) 1948cbf55001SEliezer Tamir sk_busy_loop(sk, nonblock); 1949d30e383bSEliezer Tamir 19501da177e4SLinus Torvalds lock_sock(sk); 19511da177e4SLinus Torvalds 19521da177e4SLinus Torvalds err = -ENOTCONN; 19531da177e4SLinus Torvalds if (sk->sk_state == TCP_LISTEN) 19541da177e4SLinus Torvalds goto out; 19551da177e4SLinus Torvalds 1956b75eba76SSoheil Hassas Yeganeh has_cmsg = tp->recvmsg_inq; 19571da177e4SLinus Torvalds timeo = sock_rcvtimeo(sk, nonblock); 19581da177e4SLinus Torvalds 19591da177e4SLinus Torvalds /* Urgent data needs to be handled specially. */ 19601da177e4SLinus Torvalds if (flags & MSG_OOB) 19611da177e4SLinus Torvalds goto recv_urg; 19621da177e4SLinus Torvalds 1963c0e88ff0SPavel Emelyanov if (unlikely(tp->repair)) { 1964c0e88ff0SPavel Emelyanov err = -EPERM; 1965c0e88ff0SPavel Emelyanov if (!(flags & MSG_PEEK)) 1966c0e88ff0SPavel Emelyanov goto out; 1967c0e88ff0SPavel Emelyanov 1968c0e88ff0SPavel Emelyanov if (tp->repair_queue == TCP_SEND_QUEUE) 1969c0e88ff0SPavel Emelyanov goto recv_sndq; 1970c0e88ff0SPavel Emelyanov 1971c0e88ff0SPavel Emelyanov err = -EINVAL; 1972c0e88ff0SPavel Emelyanov if (tp->repair_queue == TCP_NO_QUEUE) 1973c0e88ff0SPavel Emelyanov goto out; 1974c0e88ff0SPavel Emelyanov 1975c0e88ff0SPavel Emelyanov /* 'common' recv queue MSG_PEEK-ing */ 1976c0e88ff0SPavel Emelyanov } 1977c0e88ff0SPavel Emelyanov 19781da177e4SLinus Torvalds seq = &tp->copied_seq; 19791da177e4SLinus Torvalds if (flags & MSG_PEEK) { 19801da177e4SLinus Torvalds peek_seq = tp->copied_seq; 19811da177e4SLinus Torvalds seq = &peek_seq; 19821da177e4SLinus Torvalds } 19831da177e4SLinus Torvalds 19841da177e4SLinus Torvalds target = sock_rcvlowat(sk, flags & MSG_WAITALL, len); 19851da177e4SLinus Torvalds 19861da177e4SLinus Torvalds do { 19871da177e4SLinus Torvalds u32 offset; 19881da177e4SLinus Torvalds 19891da177e4SLinus Torvalds /* Are we at urgent data? Stop if we have read anything or have SIGURG pending. */ 19901da177e4SLinus Torvalds if (tp->urg_data && tp->urg_seq == *seq) { 19911da177e4SLinus Torvalds if (copied) 19921da177e4SLinus Torvalds break; 19931da177e4SLinus Torvalds if (signal_pending(current)) { 19941da177e4SLinus Torvalds copied = timeo ? sock_intr_errno(timeo) : -EAGAIN; 19951da177e4SLinus Torvalds break; 19961da177e4SLinus Torvalds } 19971da177e4SLinus Torvalds } 19981da177e4SLinus Torvalds 19991da177e4SLinus Torvalds /* Next get a buffer. */ 20001da177e4SLinus Torvalds 2001dfbafc99SSabrina Dubroca last = skb_peek_tail(&sk->sk_receive_queue); 200291521944SDavid S. Miller skb_queue_walk(&sk->sk_receive_queue, skb) { 2003dfbafc99SSabrina Dubroca last = skb; 20041da177e4SLinus Torvalds /* Now that we have two receive queues this 20051da177e4SLinus Torvalds * shouldn't happen. 20061da177e4SLinus Torvalds */ 2007d792c100SIlpo Järvinen if (WARN(before(*seq, TCP_SKB_CB(skb)->seq), 2008e56b8ce3SRandy Dunlap "TCP recvmsg seq # bug: copied %X, seq %X, rcvnxt %X, fl %X\n", 20092af6fd8bSJoe Perches *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt, 2010d792c100SIlpo Järvinen flags)) 20111da177e4SLinus Torvalds break; 2012d792c100SIlpo Järvinen 20131da177e4SLinus Torvalds offset = *seq - TCP_SKB_CB(skb)->seq; 20149d691539SEric Dumazet if (unlikely(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_SYN)) { 20159d691539SEric Dumazet pr_err_once("%s: found a SYN, please report !\n", __func__); 20161da177e4SLinus Torvalds offset--; 20179d691539SEric Dumazet } 20181da177e4SLinus Torvalds if (offset < skb->len) 20191da177e4SLinus Torvalds goto found_ok_skb; 2020e11ecddfSEric Dumazet if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) 20211da177e4SLinus Torvalds goto found_fin_ok; 20222af6fd8bSJoe Perches WARN(!(flags & MSG_PEEK), 2023e56b8ce3SRandy Dunlap "TCP recvmsg seq # bug 2: copied %X, seq %X, rcvnxt %X, fl %X\n", 20242af6fd8bSJoe Perches *seq, TCP_SKB_CB(skb)->seq, tp->rcv_nxt, flags); 202591521944SDavid S. Miller } 20261da177e4SLinus Torvalds 20271da177e4SLinus Torvalds /* Well, if we have backlog, try to process it now yet. */ 20281da177e4SLinus Torvalds 20291da177e4SLinus Torvalds if (copied >= target && !sk->sk_backlog.tail) 20301da177e4SLinus Torvalds break; 20311da177e4SLinus Torvalds 20321da177e4SLinus Torvalds if (copied) { 20331da177e4SLinus Torvalds if (sk->sk_err || 20341da177e4SLinus Torvalds sk->sk_state == TCP_CLOSE || 20351da177e4SLinus Torvalds (sk->sk_shutdown & RCV_SHUTDOWN) || 20361da177e4SLinus Torvalds !timeo || 2037518a09efSDavid S. Miller signal_pending(current)) 20381da177e4SLinus Torvalds break; 20391da177e4SLinus Torvalds } else { 20401da177e4SLinus Torvalds if (sock_flag(sk, SOCK_DONE)) 20411da177e4SLinus Torvalds break; 20421da177e4SLinus Torvalds 20431da177e4SLinus Torvalds if (sk->sk_err) { 20441da177e4SLinus Torvalds copied = sock_error(sk); 20451da177e4SLinus Torvalds break; 20461da177e4SLinus Torvalds } 20471da177e4SLinus Torvalds 20481da177e4SLinus Torvalds if (sk->sk_shutdown & RCV_SHUTDOWN) 20491da177e4SLinus Torvalds break; 20501da177e4SLinus Torvalds 20511da177e4SLinus Torvalds if (sk->sk_state == TCP_CLOSE) { 20521da177e4SLinus Torvalds /* This occurs when user tries to read 20531da177e4SLinus Torvalds * from never connected socket. 20541da177e4SLinus Torvalds */ 20551da177e4SLinus Torvalds copied = -ENOTCONN; 20561da177e4SLinus Torvalds break; 20571da177e4SLinus Torvalds } 20581da177e4SLinus Torvalds 20591da177e4SLinus Torvalds if (!timeo) { 20601da177e4SLinus Torvalds copied = -EAGAIN; 20611da177e4SLinus Torvalds break; 20621da177e4SLinus Torvalds } 20631da177e4SLinus Torvalds 20641da177e4SLinus Torvalds if (signal_pending(current)) { 20651da177e4SLinus Torvalds copied = sock_intr_errno(timeo); 20661da177e4SLinus Torvalds break; 20671da177e4SLinus Torvalds } 20681da177e4SLinus Torvalds } 20691da177e4SLinus Torvalds 20700e4b4992SChris Leech tcp_cleanup_rbuf(sk, copied); 20711da177e4SLinus Torvalds 20721da177e4SLinus Torvalds if (copied >= target) { 20731da177e4SLinus Torvalds /* Do not sleep, just process backlog. */ 20741da177e4SLinus Torvalds release_sock(sk); 20751da177e4SLinus Torvalds lock_sock(sk); 2076dfbafc99SSabrina Dubroca } else { 2077dfbafc99SSabrina Dubroca sk_wait_data(sk, &timeo, last); 2078dfbafc99SSabrina Dubroca } 20791da177e4SLinus Torvalds 208077527313SIlpo Järvinen if ((flags & MSG_PEEK) && 208177527313SIlpo Järvinen (peek_seq - copied - urg_hole != tp->copied_seq)) { 2082e87cc472SJoe Perches net_dbg_ratelimited("TCP(%s:%d): Application bug, race in MSG_PEEK\n", 2083e87cc472SJoe Perches current->comm, 2084e87cc472SJoe Perches task_pid_nr(current)); 20851da177e4SLinus Torvalds peek_seq = tp->copied_seq; 20861da177e4SLinus Torvalds } 20871da177e4SLinus Torvalds continue; 20881da177e4SLinus Torvalds 20891da177e4SLinus Torvalds found_ok_skb: 20901da177e4SLinus Torvalds /* Ok so how much can we use? */ 20911da177e4SLinus Torvalds used = skb->len - offset; 20921da177e4SLinus Torvalds if (len < used) 20931da177e4SLinus Torvalds used = len; 20941da177e4SLinus Torvalds 20951da177e4SLinus Torvalds /* Do we have urgent data here? */ 20961da177e4SLinus Torvalds if (tp->urg_data) { 20971da177e4SLinus Torvalds u32 urg_offset = tp->urg_seq - *seq; 20981da177e4SLinus Torvalds if (urg_offset < used) { 20991da177e4SLinus Torvalds if (!urg_offset) { 21001da177e4SLinus Torvalds if (!sock_flag(sk, SOCK_URGINLINE)) { 21011da177e4SLinus Torvalds ++*seq; 210277527313SIlpo Järvinen urg_hole++; 21031da177e4SLinus Torvalds offset++; 21041da177e4SLinus Torvalds used--; 21051da177e4SLinus Torvalds if (!used) 21061da177e4SLinus Torvalds goto skip_copy; 21071da177e4SLinus Torvalds } 21081da177e4SLinus Torvalds } else 21091da177e4SLinus Torvalds used = urg_offset; 21101da177e4SLinus Torvalds } 21111da177e4SLinus Torvalds } 21121da177e4SLinus Torvalds 21131da177e4SLinus Torvalds if (!(flags & MSG_TRUNC)) { 211451f3d02bSDavid S. Miller err = skb_copy_datagram_msg(skb, offset, msg, used); 21151da177e4SLinus Torvalds if (err) { 21161da177e4SLinus Torvalds /* Exception. Bailout! */ 21171da177e4SLinus Torvalds if (!copied) 21181da177e4SLinus Torvalds copied = -EFAULT; 21191da177e4SLinus Torvalds break; 21201da177e4SLinus Torvalds } 21211da177e4SLinus Torvalds } 21221da177e4SLinus Torvalds 21231da177e4SLinus Torvalds *seq += used; 21241da177e4SLinus Torvalds copied += used; 21251da177e4SLinus Torvalds len -= used; 21261da177e4SLinus Torvalds 21271da177e4SLinus Torvalds tcp_rcv_space_adjust(sk); 21281da177e4SLinus Torvalds 21291da177e4SLinus Torvalds skip_copy: 213031770e34SFlorian Westphal if (tp->urg_data && after(tp->copied_seq, tp->urg_seq)) { 21311da177e4SLinus Torvalds tp->urg_data = 0; 213231770e34SFlorian Westphal tcp_fast_path_check(sk); 213331770e34SFlorian Westphal } 21341da177e4SLinus Torvalds if (used + offset < skb->len) 21351da177e4SLinus Torvalds continue; 21361da177e4SLinus Torvalds 213798aaa913SMike Maloney if (TCP_SKB_CB(skb)->has_rxtstamp) { 213898aaa913SMike Maloney tcp_update_recv_tstamps(skb, &tss); 213998aaa913SMike Maloney has_tss = true; 2140b75eba76SSoheil Hassas Yeganeh has_cmsg = true; 214198aaa913SMike Maloney } 2142e11ecddfSEric Dumazet if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) 21431da177e4SLinus Torvalds goto found_fin_ok; 21447bced397SDan Williams if (!(flags & MSG_PEEK)) 21457bced397SDan Williams sk_eat_skb(sk, skb); 21461da177e4SLinus Torvalds continue; 21471da177e4SLinus Torvalds 21481da177e4SLinus Torvalds found_fin_ok: 21491da177e4SLinus Torvalds /* Process the FIN. */ 21501da177e4SLinus Torvalds ++*seq; 21517bced397SDan Williams if (!(flags & MSG_PEEK)) 21527bced397SDan Williams sk_eat_skb(sk, skb); 21531da177e4SLinus Torvalds break; 21541da177e4SLinus Torvalds } while (len > 0); 21551da177e4SLinus Torvalds 21561da177e4SLinus Torvalds /* According to UNIX98, msg_name/msg_namelen are ignored 21571da177e4SLinus Torvalds * on connected socket. I was just happy when found this 8) --ANK 21581da177e4SLinus Torvalds */ 21591da177e4SLinus Torvalds 21601da177e4SLinus Torvalds /* Clean up data we have read: This will do ACK frames. */ 21610e4b4992SChris Leech tcp_cleanup_rbuf(sk, copied); 21621da177e4SLinus Torvalds 21631da177e4SLinus Torvalds release_sock(sk); 2164b75eba76SSoheil Hassas Yeganeh 2165b75eba76SSoheil Hassas Yeganeh if (has_cmsg) { 2166b75eba76SSoheil Hassas Yeganeh if (has_tss) 2167b75eba76SSoheil Hassas Yeganeh tcp_recv_timestamp(msg, sk, &tss); 2168b75eba76SSoheil Hassas Yeganeh if (tp->recvmsg_inq) { 2169b75eba76SSoheil Hassas Yeganeh inq = tcp_inq_hint(sk); 2170b75eba76SSoheil Hassas Yeganeh put_cmsg(msg, SOL_TCP, TCP_CM_INQ, sizeof(inq), &inq); 2171b75eba76SSoheil Hassas Yeganeh } 2172b75eba76SSoheil Hassas Yeganeh } 2173b75eba76SSoheil Hassas Yeganeh 21741da177e4SLinus Torvalds return copied; 21751da177e4SLinus Torvalds 21761da177e4SLinus Torvalds out: 21771da177e4SLinus Torvalds release_sock(sk); 21781da177e4SLinus Torvalds return err; 21791da177e4SLinus Torvalds 21801da177e4SLinus Torvalds recv_urg: 2181377f0a08SRami Rosen err = tcp_recv_urg(sk, msg, len, flags); 21821da177e4SLinus Torvalds goto out; 2183c0e88ff0SPavel Emelyanov 2184c0e88ff0SPavel Emelyanov recv_sndq: 2185c0e88ff0SPavel Emelyanov err = tcp_peek_sndq(sk, msg, len); 2186c0e88ff0SPavel Emelyanov goto out; 21871da177e4SLinus Torvalds } 21884bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_recvmsg); 21891da177e4SLinus Torvalds 2190490d5046SIlpo Järvinen void tcp_set_state(struct sock *sk, int state) 2191490d5046SIlpo Järvinen { 2192490d5046SIlpo Järvinen int oldstate = sk->sk_state; 2193490d5046SIlpo Järvinen 2194d4487491SLawrence Brakmo /* We defined a new enum for TCP states that are exported in BPF 2195d4487491SLawrence Brakmo * so as not force the internal TCP states to be frozen. The 2196d4487491SLawrence Brakmo * following checks will detect if an internal state value ever 2197d4487491SLawrence Brakmo * differs from the BPF value. If this ever happens, then we will 2198d4487491SLawrence Brakmo * need to remap the internal value to the BPF value before calling 2199d4487491SLawrence Brakmo * tcp_call_bpf_2arg. 2200d4487491SLawrence Brakmo */ 2201d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_ESTABLISHED != (int)TCP_ESTABLISHED); 2202d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_SYN_SENT != (int)TCP_SYN_SENT); 2203d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_SYN_RECV != (int)TCP_SYN_RECV); 2204d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT1 != (int)TCP_FIN_WAIT1); 2205d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_FIN_WAIT2 != (int)TCP_FIN_WAIT2); 2206d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_TIME_WAIT != (int)TCP_TIME_WAIT); 2207d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_CLOSE != (int)TCP_CLOSE); 2208d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_CLOSE_WAIT != (int)TCP_CLOSE_WAIT); 2209d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_LAST_ACK != (int)TCP_LAST_ACK); 2210d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_LISTEN != (int)TCP_LISTEN); 2211d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_CLOSING != (int)TCP_CLOSING); 2212d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_NEW_SYN_RECV != (int)TCP_NEW_SYN_RECV); 2213d4487491SLawrence Brakmo BUILD_BUG_ON((int)BPF_TCP_MAX_STATES != (int)TCP_MAX_STATES); 2214d4487491SLawrence Brakmo 2215d4487491SLawrence Brakmo if (BPF_SOCK_OPS_TEST_FLAG(tcp_sk(sk), BPF_SOCK_OPS_STATE_CB_FLAG)) 2216d4487491SLawrence Brakmo tcp_call_bpf_2arg(sk, BPF_SOCK_OPS_STATE_CB, oldstate, state); 2217e8fce239SSong Liu 2218490d5046SIlpo Järvinen switch (state) { 2219490d5046SIlpo Järvinen case TCP_ESTABLISHED: 2220490d5046SIlpo Järvinen if (oldstate != TCP_ESTABLISHED) 222181cc8a75SPavel Emelyanov TCP_INC_STATS(sock_net(sk), TCP_MIB_CURRESTAB); 2222490d5046SIlpo Järvinen break; 2223490d5046SIlpo Järvinen 2224490d5046SIlpo Järvinen case TCP_CLOSE: 2225490d5046SIlpo Järvinen if (oldstate == TCP_CLOSE_WAIT || oldstate == TCP_ESTABLISHED) 222681cc8a75SPavel Emelyanov TCP_INC_STATS(sock_net(sk), TCP_MIB_ESTABRESETS); 2227490d5046SIlpo Järvinen 2228490d5046SIlpo Järvinen sk->sk_prot->unhash(sk); 2229490d5046SIlpo Järvinen if (inet_csk(sk)->icsk_bind_hash && 2230490d5046SIlpo Järvinen !(sk->sk_userlocks & SOCK_BINDPORT_LOCK)) 2231ab1e0a13SArnaldo Carvalho de Melo inet_put_port(sk); 2232490d5046SIlpo Järvinen /* fall through */ 2233490d5046SIlpo Järvinen default: 2234490d5046SIlpo Järvinen if (oldstate == TCP_ESTABLISHED) 223574688e48SPavel Emelyanov TCP_DEC_STATS(sock_net(sk), TCP_MIB_CURRESTAB); 2236490d5046SIlpo Järvinen } 2237490d5046SIlpo Järvinen 2238490d5046SIlpo Järvinen /* Change state AFTER socket is unhashed to avoid closed 2239490d5046SIlpo Järvinen * socket sitting in hash tables. 2240490d5046SIlpo Järvinen */ 2241563e0bb0SYafang Shao inet_sk_state_store(sk, state); 2242490d5046SIlpo Järvinen } 2243490d5046SIlpo Järvinen EXPORT_SYMBOL_GPL(tcp_set_state); 2244490d5046SIlpo Järvinen 22451da177e4SLinus Torvalds /* 22461da177e4SLinus Torvalds * State processing on a close. This implements the state shift for 22471da177e4SLinus Torvalds * sending our FIN frame. Note that we only send a FIN for some 22481da177e4SLinus Torvalds * states. A shutdown() may have already sent the FIN, or we may be 22491da177e4SLinus Torvalds * closed. 22501da177e4SLinus Torvalds */ 22511da177e4SLinus Torvalds 22529b5b5cffSArjan van de Ven static const unsigned char new_state[16] = { 22531da177e4SLinus Torvalds /* current state: new state: action: */ 22540980c1e3SEric Dumazet [0 /* (Invalid) */] = TCP_CLOSE, 22550980c1e3SEric Dumazet [TCP_ESTABLISHED] = TCP_FIN_WAIT1 | TCP_ACTION_FIN, 22560980c1e3SEric Dumazet [TCP_SYN_SENT] = TCP_CLOSE, 22570980c1e3SEric Dumazet [TCP_SYN_RECV] = TCP_FIN_WAIT1 | TCP_ACTION_FIN, 22580980c1e3SEric Dumazet [TCP_FIN_WAIT1] = TCP_FIN_WAIT1, 22590980c1e3SEric Dumazet [TCP_FIN_WAIT2] = TCP_FIN_WAIT2, 22600980c1e3SEric Dumazet [TCP_TIME_WAIT] = TCP_CLOSE, 22610980c1e3SEric Dumazet [TCP_CLOSE] = TCP_CLOSE, 22620980c1e3SEric Dumazet [TCP_CLOSE_WAIT] = TCP_LAST_ACK | TCP_ACTION_FIN, 22630980c1e3SEric Dumazet [TCP_LAST_ACK] = TCP_LAST_ACK, 22640980c1e3SEric Dumazet [TCP_LISTEN] = TCP_CLOSE, 22650980c1e3SEric Dumazet [TCP_CLOSING] = TCP_CLOSING, 22660980c1e3SEric Dumazet [TCP_NEW_SYN_RECV] = TCP_CLOSE, /* should not happen ! */ 22671da177e4SLinus Torvalds }; 22681da177e4SLinus Torvalds 22691da177e4SLinus Torvalds static int tcp_close_state(struct sock *sk) 22701da177e4SLinus Torvalds { 22711da177e4SLinus Torvalds int next = (int)new_state[sk->sk_state]; 22721da177e4SLinus Torvalds int ns = next & TCP_STATE_MASK; 22731da177e4SLinus Torvalds 22741da177e4SLinus Torvalds tcp_set_state(sk, ns); 22751da177e4SLinus Torvalds 22761da177e4SLinus Torvalds return next & TCP_ACTION_FIN; 22771da177e4SLinus Torvalds } 22781da177e4SLinus Torvalds 22791da177e4SLinus Torvalds /* 22801da177e4SLinus Torvalds * Shutdown the sending side of a connection. Much like close except 22811f29b058SSatoru SATOH * that we don't receive shut down or sock_set_flag(sk, SOCK_DEAD). 22821da177e4SLinus Torvalds */ 22831da177e4SLinus Torvalds 22841da177e4SLinus Torvalds void tcp_shutdown(struct sock *sk, int how) 22851da177e4SLinus Torvalds { 22861da177e4SLinus Torvalds /* We need to grab some memory, and put together a FIN, 22871da177e4SLinus Torvalds * and then put it into the queue to be sent. 22881da177e4SLinus Torvalds * Tim MacKenzie(tym@dibbler.cs.monash.edu.au) 4 Dec '92. 22891da177e4SLinus Torvalds */ 22901da177e4SLinus Torvalds if (!(how & SEND_SHUTDOWN)) 22911da177e4SLinus Torvalds return; 22921da177e4SLinus Torvalds 22931da177e4SLinus Torvalds /* If we've already sent a FIN, or it's a closed state, skip this. */ 22941da177e4SLinus Torvalds if ((1 << sk->sk_state) & 22951da177e4SLinus Torvalds (TCPF_ESTABLISHED | TCPF_SYN_SENT | 22961da177e4SLinus Torvalds TCPF_SYN_RECV | TCPF_CLOSE_WAIT)) { 22971da177e4SLinus Torvalds /* Clear out any half completed packets. FIN if needed. */ 22981da177e4SLinus Torvalds if (tcp_close_state(sk)) 22991da177e4SLinus Torvalds tcp_send_fin(sk); 23001da177e4SLinus Torvalds } 23011da177e4SLinus Torvalds } 23024bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_shutdown); 23031da177e4SLinus Torvalds 2304efcdbf24SArun Sharma bool tcp_check_oom(struct sock *sk, int shift) 2305efcdbf24SArun Sharma { 2306efcdbf24SArun Sharma bool too_many_orphans, out_of_socket_memory; 2307efcdbf24SArun Sharma 2308efcdbf24SArun Sharma too_many_orphans = tcp_too_many_orphans(sk, shift); 2309efcdbf24SArun Sharma out_of_socket_memory = tcp_out_of_memory(sk); 2310efcdbf24SArun Sharma 2311e87cc472SJoe Perches if (too_many_orphans) 2312e87cc472SJoe Perches net_info_ratelimited("too many orphaned sockets\n"); 2313e87cc472SJoe Perches if (out_of_socket_memory) 2314e87cc472SJoe Perches net_info_ratelimited("out of memory -- consider tuning tcp_mem\n"); 2315efcdbf24SArun Sharma return too_many_orphans || out_of_socket_memory; 2316efcdbf24SArun Sharma } 2317efcdbf24SArun Sharma 23181da177e4SLinus Torvalds void tcp_close(struct sock *sk, long timeout) 23191da177e4SLinus Torvalds { 23201da177e4SLinus Torvalds struct sk_buff *skb; 23211da177e4SLinus Torvalds int data_was_unread = 0; 232275c2d907SHerbert Xu int state; 23231da177e4SLinus Torvalds 23241da177e4SLinus Torvalds lock_sock(sk); 23251da177e4SLinus Torvalds sk->sk_shutdown = SHUTDOWN_MASK; 23261da177e4SLinus Torvalds 23271da177e4SLinus Torvalds if (sk->sk_state == TCP_LISTEN) { 23281da177e4SLinus Torvalds tcp_set_state(sk, TCP_CLOSE); 23291da177e4SLinus Torvalds 23301da177e4SLinus Torvalds /* Special case. */ 23310a5578cfSArnaldo Carvalho de Melo inet_csk_listen_stop(sk); 23321da177e4SLinus Torvalds 23331da177e4SLinus Torvalds goto adjudge_to_death; 23341da177e4SLinus Torvalds } 23351da177e4SLinus Torvalds 23361da177e4SLinus Torvalds /* We need to flush the recv. buffs. We do this only on the 23371da177e4SLinus Torvalds * descriptor close, not protocol-sourced closes, because the 23381da177e4SLinus Torvalds * reader process may not have drained the data yet! 23391da177e4SLinus Torvalds */ 23401da177e4SLinus Torvalds while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) { 2341e11ecddfSEric Dumazet u32 len = TCP_SKB_CB(skb)->end_seq - TCP_SKB_CB(skb)->seq; 2342e11ecddfSEric Dumazet 2343e11ecddfSEric Dumazet if (TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN) 2344e11ecddfSEric Dumazet len--; 23451da177e4SLinus Torvalds data_was_unread += len; 23461da177e4SLinus Torvalds __kfree_skb(skb); 23471da177e4SLinus Torvalds } 23481da177e4SLinus Torvalds 23493ab224beSHideo Aoki sk_mem_reclaim(sk); 23501da177e4SLinus Torvalds 2351565b7b2dSKonstantin Khorenko /* If socket has been already reset (e.g. in tcp_reset()) - kill it. */ 2352565b7b2dSKonstantin Khorenko if (sk->sk_state == TCP_CLOSE) 2353565b7b2dSKonstantin Khorenko goto adjudge_to_death; 2354565b7b2dSKonstantin Khorenko 235565bb723cSGerrit Renker /* As outlined in RFC 2525, section 2.17, we send a RST here because 235665bb723cSGerrit Renker * data was lost. To witness the awful effects of the old behavior of 235765bb723cSGerrit Renker * always doing a FIN, run an older 2.1.x kernel or 2.0.x, start a bulk 235865bb723cSGerrit Renker * GET in an FTP client, suspend the process, wait for the client to 235965bb723cSGerrit Renker * advertise a zero window, then kill -9 the FTP client, wheee... 236065bb723cSGerrit Renker * Note: timeout is always zero in such a case. 23611da177e4SLinus Torvalds */ 2362ee995283SPavel Emelyanov if (unlikely(tcp_sk(sk)->repair)) { 2363ee995283SPavel Emelyanov sk->sk_prot->disconnect(sk, 0); 2364ee995283SPavel Emelyanov } else if (data_was_unread) { 23651da177e4SLinus Torvalds /* Unread data was tossed, zap the connection. */ 23666aef70a8SEric Dumazet NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONCLOSE); 23671da177e4SLinus Torvalds tcp_set_state(sk, TCP_CLOSE); 2368aa133076SWu Fengguang tcp_send_active_reset(sk, sk->sk_allocation); 23691da177e4SLinus Torvalds } else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) { 23701da177e4SLinus Torvalds /* Check zero linger _after_ checking for unread data. */ 23711da177e4SLinus Torvalds sk->sk_prot->disconnect(sk, 0); 23726aef70a8SEric Dumazet NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPABORTONDATA); 23731da177e4SLinus Torvalds } else if (tcp_close_state(sk)) { 23741da177e4SLinus Torvalds /* We FIN if the application ate all the data before 23751da177e4SLinus Torvalds * zapping the connection. 23761da177e4SLinus Torvalds */ 23771da177e4SLinus Torvalds 23781da177e4SLinus Torvalds /* RED-PEN. Formally speaking, we have broken TCP state 23791da177e4SLinus Torvalds * machine. State transitions: 23801da177e4SLinus Torvalds * 23811da177e4SLinus Torvalds * TCP_ESTABLISHED -> TCP_FIN_WAIT1 23821da177e4SLinus Torvalds * TCP_SYN_RECV -> TCP_FIN_WAIT1 (forget it, it's impossible) 23831da177e4SLinus Torvalds * TCP_CLOSE_WAIT -> TCP_LAST_ACK 23841da177e4SLinus Torvalds * 23851da177e4SLinus Torvalds * are legal only when FIN has been sent (i.e. in window), 23861da177e4SLinus Torvalds * rather than queued out of window. Purists blame. 23871da177e4SLinus Torvalds * 23881da177e4SLinus Torvalds * F.e. "RFC state" is ESTABLISHED, 23891da177e4SLinus Torvalds * if Linux state is FIN-WAIT-1, but FIN is still not sent. 23901da177e4SLinus Torvalds * 23911da177e4SLinus Torvalds * The visible declinations are that sometimes 23921da177e4SLinus Torvalds * we enter time-wait state, when it is not required really 23931da177e4SLinus Torvalds * (harmless), do not send active resets, when they are 23941da177e4SLinus Torvalds * required by specs (TCP_ESTABLISHED, TCP_CLOSE_WAIT, when 23951da177e4SLinus Torvalds * they look as CLOSING or LAST_ACK for Linux) 23961da177e4SLinus Torvalds * Probably, I missed some more holelets. 23971da177e4SLinus Torvalds * --ANK 23988336886fSJerry Chu * XXX (TFO) - To start off we don't support SYN+ACK+FIN 23998336886fSJerry Chu * in a single packet! (May consider it later but will 24008336886fSJerry Chu * probably need API support or TCP_CORK SYN-ACK until 24018336886fSJerry Chu * data is written and socket is closed.) 24021da177e4SLinus Torvalds */ 24031da177e4SLinus Torvalds tcp_send_fin(sk); 24041da177e4SLinus Torvalds } 24051da177e4SLinus Torvalds 24061da177e4SLinus Torvalds sk_stream_wait_close(sk, timeout); 24071da177e4SLinus Torvalds 24081da177e4SLinus Torvalds adjudge_to_death: 240975c2d907SHerbert Xu state = sk->sk_state; 241075c2d907SHerbert Xu sock_hold(sk); 241175c2d907SHerbert Xu sock_orphan(sk); 241275c2d907SHerbert Xu 24131da177e4SLinus Torvalds local_bh_disable(); 24141da177e4SLinus Torvalds bh_lock_sock(sk); 24158873c064SEric Dumazet /* remove backlog if any, without releasing ownership. */ 24168873c064SEric Dumazet __release_sock(sk); 24171da177e4SLinus Torvalds 2418eb4dea58SHerbert Xu percpu_counter_inc(sk->sk_prot->orphan_count); 2419eb4dea58SHerbert Xu 242075c2d907SHerbert Xu /* Have we already been destroyed by a softirq or backlog? */ 242175c2d907SHerbert Xu if (state != TCP_CLOSE && sk->sk_state == TCP_CLOSE) 242275c2d907SHerbert Xu goto out; 24231da177e4SLinus Torvalds 24241da177e4SLinus Torvalds /* This is a (useful) BSD violating of the RFC. There is a 24251da177e4SLinus Torvalds * problem with TCP as specified in that the other end could 24261da177e4SLinus Torvalds * keep a socket open forever with no application left this end. 2427b10bd54cSJesper Juhl * We use a 1 minute timeout (about the same as BSD) then kill 24281da177e4SLinus Torvalds * our end. If they send after that then tough - BUT: long enough 24291da177e4SLinus Torvalds * that we won't make the old 4*rto = almost no time - whoops 24301da177e4SLinus Torvalds * reset mistake. 24311da177e4SLinus Torvalds * 24321da177e4SLinus Torvalds * Nope, it was not mistake. It is really desired behaviour 24331da177e4SLinus Torvalds * f.e. on http servers, when such sockets are useless, but 24341da177e4SLinus Torvalds * consume significant resources. Let's do it with special 24351da177e4SLinus Torvalds * linger2 option. --ANK 24361da177e4SLinus Torvalds */ 24371da177e4SLinus Torvalds 24381da177e4SLinus Torvalds if (sk->sk_state == TCP_FIN_WAIT2) { 24391da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 24401da177e4SLinus Torvalds if (tp->linger2 < 0) { 24411da177e4SLinus Torvalds tcp_set_state(sk, TCP_CLOSE); 24421da177e4SLinus Torvalds tcp_send_active_reset(sk, GFP_ATOMIC); 244302a1d6e7SEric Dumazet __NET_INC_STATS(sock_net(sk), 2444de0744afSPavel Emelyanov LINUX_MIB_TCPABORTONLINGER); 24451da177e4SLinus Torvalds } else { 2446463c84b9SArnaldo Carvalho de Melo const int tmo = tcp_fin_time(sk); 24471da177e4SLinus Torvalds 24481da177e4SLinus Torvalds if (tmo > TCP_TIMEWAIT_LEN) { 244952499afeSDavid S. Miller inet_csk_reset_keepalive_timer(sk, 245052499afeSDavid S. Miller tmo - TCP_TIMEWAIT_LEN); 24511da177e4SLinus Torvalds } else { 24521da177e4SLinus Torvalds tcp_time_wait(sk, TCP_FIN_WAIT2, tmo); 24531da177e4SLinus Torvalds goto out; 24541da177e4SLinus Torvalds } 24551da177e4SLinus Torvalds } 24561da177e4SLinus Torvalds } 24571da177e4SLinus Torvalds if (sk->sk_state != TCP_CLOSE) { 24583ab224beSHideo Aoki sk_mem_reclaim(sk); 2459efcdbf24SArun Sharma if (tcp_check_oom(sk, 0)) { 24601da177e4SLinus Torvalds tcp_set_state(sk, TCP_CLOSE); 24611da177e4SLinus Torvalds tcp_send_active_reset(sk, GFP_ATOMIC); 246202a1d6e7SEric Dumazet __NET_INC_STATS(sock_net(sk), 2463de0744afSPavel Emelyanov LINUX_MIB_TCPABORTONMEMORY); 24644ee806d5SDan Streetman } else if (!check_net(sock_net(sk))) { 24654ee806d5SDan Streetman /* Not possible to send reset; just close */ 24664ee806d5SDan Streetman tcp_set_state(sk, TCP_CLOSE); 24671da177e4SLinus Torvalds } 24681da177e4SLinus Torvalds } 24691da177e4SLinus Torvalds 24708336886fSJerry Chu if (sk->sk_state == TCP_CLOSE) { 24718336886fSJerry Chu struct request_sock *req = tcp_sk(sk)->fastopen_rsk; 24728336886fSJerry Chu /* We could get here with a non-NULL req if the socket is 24738336886fSJerry Chu * aborted (e.g., closed with unread data) before 3WHS 24748336886fSJerry Chu * finishes. 24758336886fSJerry Chu */ 247600db4124SIan Morris if (req) 24778336886fSJerry Chu reqsk_fastopen_remove(sk, req, false); 24780a5578cfSArnaldo Carvalho de Melo inet_csk_destroy_sock(sk); 24798336886fSJerry Chu } 24801da177e4SLinus Torvalds /* Otherwise, socket is reprieved until protocol close. */ 24811da177e4SLinus Torvalds 24821da177e4SLinus Torvalds out: 24831da177e4SLinus Torvalds bh_unlock_sock(sk); 24841da177e4SLinus Torvalds local_bh_enable(); 24858873c064SEric Dumazet release_sock(sk); 24861da177e4SLinus Torvalds sock_put(sk); 24871da177e4SLinus Torvalds } 24884bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_close); 24891da177e4SLinus Torvalds 24901da177e4SLinus Torvalds /* These states need RST on ABORT according to RFC793 */ 24911da177e4SLinus Torvalds 2492a2a385d6SEric Dumazet static inline bool tcp_need_reset(int state) 24931da177e4SLinus Torvalds { 24941da177e4SLinus Torvalds return (1 << state) & 24951da177e4SLinus Torvalds (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT | TCPF_FIN_WAIT1 | 24961da177e4SLinus Torvalds TCPF_FIN_WAIT2 | TCPF_SYN_RECV); 24971da177e4SLinus Torvalds } 24981da177e4SLinus Torvalds 249975c119afSEric Dumazet static void tcp_rtx_queue_purge(struct sock *sk) 250075c119afSEric Dumazet { 250175c119afSEric Dumazet struct rb_node *p = rb_first(&sk->tcp_rtx_queue); 250275c119afSEric Dumazet 250375c119afSEric Dumazet while (p) { 250475c119afSEric Dumazet struct sk_buff *skb = rb_to_skb(p); 250575c119afSEric Dumazet 250675c119afSEric Dumazet p = rb_next(p); 250775c119afSEric Dumazet /* Since we are deleting whole queue, no need to 250875c119afSEric Dumazet * list_del(&skb->tcp_tsorted_anchor) 250975c119afSEric Dumazet */ 251075c119afSEric Dumazet tcp_rtx_queue_unlink(skb, sk); 251175c119afSEric Dumazet sk_wmem_free_skb(sk, skb); 251275c119afSEric Dumazet } 251375c119afSEric Dumazet } 251475c119afSEric Dumazet 2515ac3f09baSEric Dumazet void tcp_write_queue_purge(struct sock *sk) 2516ac3f09baSEric Dumazet { 2517ac3f09baSEric Dumazet struct sk_buff *skb; 2518ac3f09baSEric Dumazet 2519ac3f09baSEric Dumazet tcp_chrono_stop(sk, TCP_CHRONO_BUSY); 2520ac3f09baSEric Dumazet while ((skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) { 2521ac3f09baSEric Dumazet tcp_skb_tsorted_anchor_cleanup(skb); 2522ac3f09baSEric Dumazet sk_wmem_free_skb(sk, skb); 2523ac3f09baSEric Dumazet } 252475c119afSEric Dumazet tcp_rtx_queue_purge(sk); 2525472c2e07SEric Dumazet skb = sk->sk_tx_skb_cache; 2526472c2e07SEric Dumazet if (skb) { 2527472c2e07SEric Dumazet __kfree_skb(skb); 2528472c2e07SEric Dumazet sk->sk_tx_skb_cache = NULL; 2529472c2e07SEric Dumazet } 2530ac3f09baSEric Dumazet INIT_LIST_HEAD(&tcp_sk(sk)->tsorted_sent_queue); 2531ac3f09baSEric Dumazet sk_mem_reclaim(sk); 2532ac3f09baSEric Dumazet tcp_clear_all_retrans_hints(tcp_sk(sk)); 2533bffd168cSSoheil Hassas Yeganeh tcp_sk(sk)->packets_out = 0; 253404c03114SEric Dumazet inet_csk(sk)->icsk_backoff = 0; 2535ac3f09baSEric Dumazet } 2536ac3f09baSEric Dumazet 25371da177e4SLinus Torvalds int tcp_disconnect(struct sock *sk, int flags) 25381da177e4SLinus Torvalds { 25391da177e4SLinus Torvalds struct inet_sock *inet = inet_sk(sk); 2540463c84b9SArnaldo Carvalho de Melo struct inet_connection_sock *icsk = inet_csk(sk); 25411da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 25421da177e4SLinus Torvalds int old_state = sk->sk_state; 25431da177e4SLinus Torvalds 25441da177e4SLinus Torvalds if (old_state != TCP_CLOSE) 25451da177e4SLinus Torvalds tcp_set_state(sk, TCP_CLOSE); 25461da177e4SLinus Torvalds 25471da177e4SLinus Torvalds /* ABORT function of RFC793 */ 25481da177e4SLinus Torvalds if (old_state == TCP_LISTEN) { 25490a5578cfSArnaldo Carvalho de Melo inet_csk_listen_stop(sk); 2550ee995283SPavel Emelyanov } else if (unlikely(tp->repair)) { 2551ee995283SPavel Emelyanov sk->sk_err = ECONNABORTED; 25521da177e4SLinus Torvalds } else if (tcp_need_reset(old_state) || 25531da177e4SLinus Torvalds (tp->snd_nxt != tp->write_seq && 25541da177e4SLinus Torvalds (1 << old_state) & (TCPF_CLOSING | TCPF_LAST_ACK))) { 2555caa20d9aSStephen Hemminger /* The last check adjusts for discrepancy of Linux wrt. RFC 25561da177e4SLinus Torvalds * states 25571da177e4SLinus Torvalds */ 25581da177e4SLinus Torvalds tcp_send_active_reset(sk, gfp_any()); 25591da177e4SLinus Torvalds sk->sk_err = ECONNRESET; 25601da177e4SLinus Torvalds } else if (old_state == TCP_SYN_SENT) 25611da177e4SLinus Torvalds sk->sk_err = ECONNRESET; 25621da177e4SLinus Torvalds 25631da177e4SLinus Torvalds tcp_clear_xmit_timers(sk); 25641da177e4SLinus Torvalds __skb_queue_purge(&sk->sk_receive_queue); 25658b27dae5SEric Dumazet if (sk->sk_rx_skb_cache) { 25668b27dae5SEric Dumazet __kfree_skb(sk->sk_rx_skb_cache); 25678b27dae5SEric Dumazet sk->sk_rx_skb_cache = NULL; 25688b27dae5SEric Dumazet } 25696508b678SEric Dumazet tp->copied_seq = tp->rcv_nxt; 25706508b678SEric Dumazet tp->urg_data = 0; 2571fe067e8aSDavid S. Miller tcp_write_queue_purge(sk); 2572cf1ef3f0SWei Wang tcp_fastopen_active_disable_ofo_check(sk); 25739f5afeaeSYaogong Wang skb_rbtree_purge(&tp->out_of_order_queue); 25741da177e4SLinus Torvalds 2575c720c7e8SEric Dumazet inet->inet_dport = 0; 25761da177e4SLinus Torvalds 25771da177e4SLinus Torvalds if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK)) 25781da177e4SLinus Torvalds inet_reset_saddr(sk); 25791da177e4SLinus Torvalds 25801da177e4SLinus Torvalds sk->sk_shutdown = 0; 25811da177e4SLinus Torvalds sock_reset_flag(sk, SOCK_DONE); 2582740b0f18SEric Dumazet tp->srtt_us = 0; 2583b9e2e689SEric Dumazet tp->mdev_us = jiffies_to_usecs(TCP_TIMEOUT_INIT); 25843f6c65d6SWei Wang tp->rcv_rtt_last_tsecr = 0; 2585686a5624SYuvaraja Mariappan tp->write_seq += tp->max_window + 2; 2586686a5624SYuvaraja Mariappan if (tp->write_seq == 0) 25871da177e4SLinus Torvalds tp->write_seq = 1; 2588463c84b9SArnaldo Carvalho de Melo icsk->icsk_backoff = 0; 25891da177e4SLinus Torvalds tp->snd_cwnd = 2; 25906687e988SArnaldo Carvalho de Melo icsk->icsk_probes_out = 0; 25916a408147SEric Dumazet icsk->icsk_rto = TCP_TIMEOUT_INIT; 25920b6a05c1SIlpo Järvinen tp->snd_ssthresh = TCP_INFINITE_SSTHRESH; 25933a9a57f6SEric Dumazet tp->snd_cwnd = TCP_INIT_CWND; 25941da177e4SLinus Torvalds tp->snd_cwnd_cnt = 0; 25951fdf475aSEric Dumazet tp->window_clamp = 0; 2596e21db6f6SYuchung Cheng tp->delivered_ce = 0; 25976687e988SArnaldo Carvalho de Melo tcp_set_ca_state(sk, TCP_CA_Open); 2598d4761754SYousuk Seung tp->is_sack_reneg = 0; 25991da177e4SLinus Torvalds tcp_clear_retrans(tp); 2600463c84b9SArnaldo Carvalho de Melo inet_csk_delack_init(sk); 2601499350a5SWei Wang /* Initialize rcv_mss to TCP_MIN_MSS to avoid division by 0 2602499350a5SWei Wang * issue in __tcp_select_window() 2603499350a5SWei Wang */ 2604499350a5SWei Wang icsk->icsk_ack.rcv_mss = TCP_MIN_MSS; 2605b40b4f79SSrinivas Aji memset(&tp->rx_opt, 0, sizeof(tp->rx_opt)); 26061da177e4SLinus Torvalds __sk_dst_reset(sk); 2607d747a7a5SWANG Cong dst_release(sk->sk_rx_dst); 2608d747a7a5SWANG Cong sk->sk_rx_dst = NULL; 260917c3060bSEric Dumazet tcp_saved_syn_free(tp); 26105d9f4262SEric Dumazet tp->compressed_ack = 0; 2611ba113c3aSWei Wang tp->bytes_sent = 0; 2612fb31c9b9SWei Wang tp->bytes_retrans = 0; 26137788174eSYuchung Cheng tp->duplicate_sack[0].start_seq = 0; 26147788174eSYuchung Cheng tp->duplicate_sack[0].end_seq = 0; 26157e10b655SWei Wang tp->dsack_dups = 0; 26167ec65372SWei Wang tp->reord_seen = 0; 26175c701549SEric Dumazet tp->retrans_out = 0; 26185c701549SEric Dumazet tp->sacked_out = 0; 26195c701549SEric Dumazet tp->tlp_high_seq = 0; 26205c701549SEric Dumazet tp->last_oow_ack_time = 0; 26216cda8b74SEric Dumazet /* There's a bubble in the pipe until at least the first ACK. */ 26226cda8b74SEric Dumazet tp->app_limited = ~0U; 2623792c4354SEric Dumazet tp->rack.mstamp = 0; 2624792c4354SEric Dumazet tp->rack.advanced = 0; 2625792c4354SEric Dumazet tp->rack.reo_wnd_steps = 1; 2626792c4354SEric Dumazet tp->rack.last_delivered = 0; 2627792c4354SEric Dumazet tp->rack.reo_wnd_persist = 0; 2628792c4354SEric Dumazet tp->rack.dsack_seen = 0; 26296bcdc40dSEric Dumazet tp->syn_data_acked = 0; 26306bcdc40dSEric Dumazet tp->rx_opt.saw_tstamp = 0; 26316bcdc40dSEric Dumazet tp->rx_opt.dsack = 0; 26326bcdc40dSEric Dumazet tp->rx_opt.num_sacks = 0; 26336cda8b74SEric Dumazet 26341da177e4SLinus Torvalds 26357db92362SWei Wang /* Clean up fastopen related fields */ 26367db92362SWei Wang tcp_free_fastopen_req(tp); 26377db92362SWei Wang inet->defer_connect = 0; 26387db92362SWei Wang 2639c720c7e8SEric Dumazet WARN_ON(inet->inet_num && !icsk->icsk_bind_hash); 26401da177e4SLinus Torvalds 26419b42d55aSLi RongQing if (sk->sk_frag.page) { 26429b42d55aSLi RongQing put_page(sk->sk_frag.page); 26439b42d55aSLi RongQing sk->sk_frag.page = NULL; 26449b42d55aSLi RongQing sk->sk_frag.offset = 0; 26459b42d55aSLi RongQing } 26469b42d55aSLi RongQing 26471da177e4SLinus Torvalds sk->sk_error_report(sk); 2648a01512b1SYueHaibing return 0; 26491da177e4SLinus Torvalds } 26504bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_disconnect); 26511da177e4SLinus Torvalds 2652a2a385d6SEric Dumazet static inline bool tcp_can_repair_sock(const struct sock *sk) 2653ee995283SPavel Emelyanov { 265452e804c6SEric W. Biederman return ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN) && 2655319b0534SAndrey Vagin (sk->sk_state != TCP_LISTEN); 2656ee995283SPavel Emelyanov } 2657ee995283SPavel Emelyanov 2658b1ed4c4fSAndrey Vagin static int tcp_repair_set_window(struct tcp_sock *tp, char __user *optbuf, int len) 2659b1ed4c4fSAndrey Vagin { 2660b1ed4c4fSAndrey Vagin struct tcp_repair_window opt; 2661b1ed4c4fSAndrey Vagin 2662b1ed4c4fSAndrey Vagin if (!tp->repair) 2663b1ed4c4fSAndrey Vagin return -EPERM; 2664b1ed4c4fSAndrey Vagin 2665b1ed4c4fSAndrey Vagin if (len != sizeof(opt)) 2666b1ed4c4fSAndrey Vagin return -EINVAL; 2667b1ed4c4fSAndrey Vagin 2668b1ed4c4fSAndrey Vagin if (copy_from_user(&opt, optbuf, sizeof(opt))) 2669b1ed4c4fSAndrey Vagin return -EFAULT; 2670b1ed4c4fSAndrey Vagin 2671b1ed4c4fSAndrey Vagin if (opt.max_window < opt.snd_wnd) 2672b1ed4c4fSAndrey Vagin return -EINVAL; 2673b1ed4c4fSAndrey Vagin 2674b1ed4c4fSAndrey Vagin if (after(opt.snd_wl1, tp->rcv_nxt + opt.rcv_wnd)) 2675b1ed4c4fSAndrey Vagin return -EINVAL; 2676b1ed4c4fSAndrey Vagin 2677b1ed4c4fSAndrey Vagin if (after(opt.rcv_wup, tp->rcv_nxt)) 2678b1ed4c4fSAndrey Vagin return -EINVAL; 2679b1ed4c4fSAndrey Vagin 2680b1ed4c4fSAndrey Vagin tp->snd_wl1 = opt.snd_wl1; 2681b1ed4c4fSAndrey Vagin tp->snd_wnd = opt.snd_wnd; 2682b1ed4c4fSAndrey Vagin tp->max_window = opt.max_window; 2683b1ed4c4fSAndrey Vagin 2684b1ed4c4fSAndrey Vagin tp->rcv_wnd = opt.rcv_wnd; 2685b1ed4c4fSAndrey Vagin tp->rcv_wup = opt.rcv_wup; 2686b1ed4c4fSAndrey Vagin 2687b1ed4c4fSAndrey Vagin return 0; 2688b1ed4c4fSAndrey Vagin } 2689b1ed4c4fSAndrey Vagin 269015e56515SDouglas Caetano dos Santos static int tcp_repair_options_est(struct sock *sk, 2691de248a75SPavel Emelyanov struct tcp_repair_opt __user *optbuf, unsigned int len) 2692b139ba4eSPavel Emelyanov { 269315e56515SDouglas Caetano dos Santos struct tcp_sock *tp = tcp_sk(sk); 2694de248a75SPavel Emelyanov struct tcp_repair_opt opt; 2695b139ba4eSPavel Emelyanov 2696de248a75SPavel Emelyanov while (len >= sizeof(opt)) { 2697de248a75SPavel Emelyanov if (copy_from_user(&opt, optbuf, sizeof(opt))) 2698b139ba4eSPavel Emelyanov return -EFAULT; 2699b139ba4eSPavel Emelyanov 2700b139ba4eSPavel Emelyanov optbuf++; 2701de248a75SPavel Emelyanov len -= sizeof(opt); 2702b139ba4eSPavel Emelyanov 2703de248a75SPavel Emelyanov switch (opt.opt_code) { 2704de248a75SPavel Emelyanov case TCPOPT_MSS: 2705de248a75SPavel Emelyanov tp->rx_opt.mss_clamp = opt.opt_val; 270615e56515SDouglas Caetano dos Santos tcp_mtup_init(sk); 2707b139ba4eSPavel Emelyanov break; 2708de248a75SPavel Emelyanov case TCPOPT_WINDOW: 2709bc26ccd8SAndrey Vagin { 2710bc26ccd8SAndrey Vagin u16 snd_wscale = opt.opt_val & 0xFFFF; 2711bc26ccd8SAndrey Vagin u16 rcv_wscale = opt.opt_val >> 16; 2712bc26ccd8SAndrey Vagin 2713589c49cbSGao Feng if (snd_wscale > TCP_MAX_WSCALE || rcv_wscale > TCP_MAX_WSCALE) 2714b139ba4eSPavel Emelyanov return -EFBIG; 2715b139ba4eSPavel Emelyanov 2716bc26ccd8SAndrey Vagin tp->rx_opt.snd_wscale = snd_wscale; 2717bc26ccd8SAndrey Vagin tp->rx_opt.rcv_wscale = rcv_wscale; 2718bc26ccd8SAndrey Vagin tp->rx_opt.wscale_ok = 1; 2719bc26ccd8SAndrey Vagin } 2720b139ba4eSPavel Emelyanov break; 2721b139ba4eSPavel Emelyanov case TCPOPT_SACK_PERM: 2722de248a75SPavel Emelyanov if (opt.opt_val != 0) 2723de248a75SPavel Emelyanov return -EINVAL; 2724de248a75SPavel Emelyanov 2725b139ba4eSPavel Emelyanov tp->rx_opt.sack_ok |= TCP_SACK_SEEN; 2726b139ba4eSPavel Emelyanov break; 2727b139ba4eSPavel Emelyanov case TCPOPT_TIMESTAMP: 2728de248a75SPavel Emelyanov if (opt.opt_val != 0) 2729de248a75SPavel Emelyanov return -EINVAL; 2730de248a75SPavel Emelyanov 2731b139ba4eSPavel Emelyanov tp->rx_opt.tstamp_ok = 1; 2732b139ba4eSPavel Emelyanov break; 2733b139ba4eSPavel Emelyanov } 2734b139ba4eSPavel Emelyanov } 2735b139ba4eSPavel Emelyanov 2736b139ba4eSPavel Emelyanov return 0; 2737b139ba4eSPavel Emelyanov } 2738b139ba4eSPavel Emelyanov 2739*a842fe14SEric Dumazet DEFINE_STATIC_KEY_FALSE(tcp_tx_delay_enabled); 2740*a842fe14SEric Dumazet EXPORT_SYMBOL(tcp_tx_delay_enabled); 2741*a842fe14SEric Dumazet 2742*a842fe14SEric Dumazet static void tcp_enable_tx_delay(void) 2743*a842fe14SEric Dumazet { 2744*a842fe14SEric Dumazet if (!static_branch_unlikely(&tcp_tx_delay_enabled)) { 2745*a842fe14SEric Dumazet static int __tcp_tx_delay_enabled = 0; 2746*a842fe14SEric Dumazet 2747*a842fe14SEric Dumazet if (cmpxchg(&__tcp_tx_delay_enabled, 0, 1) == 0) { 2748*a842fe14SEric Dumazet static_branch_enable(&tcp_tx_delay_enabled); 2749*a842fe14SEric Dumazet pr_info("TCP_TX_DELAY enabled\n"); 2750*a842fe14SEric Dumazet } 2751*a842fe14SEric Dumazet } 2752*a842fe14SEric Dumazet } 2753*a842fe14SEric Dumazet 27541da177e4SLinus Torvalds /* 27551da177e4SLinus Torvalds * Socket option code for TCP. 27561da177e4SLinus Torvalds */ 27573fdadf7dSDmitry Mishin static int do_tcp_setsockopt(struct sock *sk, int level, 2758b7058842SDavid S. Miller int optname, char __user *optval, unsigned int optlen) 27591da177e4SLinus Torvalds { 27601da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 2761463c84b9SArnaldo Carvalho de Melo struct inet_connection_sock *icsk = inet_csk(sk); 27621e579caaSNikolay Borisov struct net *net = sock_net(sk); 27631da177e4SLinus Torvalds int val; 27641da177e4SLinus Torvalds int err = 0; 27651da177e4SLinus Torvalds 2766e56fb50fSWilliam Allen Simpson /* These are data/string values, all the others are ints */ 2767e56fb50fSWilliam Allen Simpson switch (optname) { 2768e56fb50fSWilliam Allen Simpson case TCP_CONGESTION: { 27695f8ef48dSStephen Hemminger char name[TCP_CA_NAME_MAX]; 27705f8ef48dSStephen Hemminger 27715f8ef48dSStephen Hemminger if (optlen < 1) 27725f8ef48dSStephen Hemminger return -EINVAL; 27735f8ef48dSStephen Hemminger 27745f8ef48dSStephen Hemminger val = strncpy_from_user(name, optval, 27754fdb78d3SAndrew Morton min_t(long, TCP_CA_NAME_MAX-1, optlen)); 27765f8ef48dSStephen Hemminger if (val < 0) 27775f8ef48dSStephen Hemminger return -EFAULT; 27785f8ef48dSStephen Hemminger name[val] = 0; 27795f8ef48dSStephen Hemminger 27805f8ef48dSStephen Hemminger lock_sock(sk); 2781ebfa00c5SSabrina Dubroca err = tcp_set_congestion_control(sk, name, true, true); 27825f8ef48dSStephen Hemminger release_sock(sk); 27835f8ef48dSStephen Hemminger return err; 27845f8ef48dSStephen Hemminger } 2785734942ccSDave Watson case TCP_ULP: { 2786734942ccSDave Watson char name[TCP_ULP_NAME_MAX]; 2787734942ccSDave Watson 2788734942ccSDave Watson if (optlen < 1) 2789734942ccSDave Watson return -EINVAL; 2790734942ccSDave Watson 2791734942ccSDave Watson val = strncpy_from_user(name, optval, 2792734942ccSDave Watson min_t(long, TCP_ULP_NAME_MAX - 1, 2793734942ccSDave Watson optlen)); 2794734942ccSDave Watson if (val < 0) 2795734942ccSDave Watson return -EFAULT; 2796734942ccSDave Watson name[val] = 0; 2797734942ccSDave Watson 2798734942ccSDave Watson lock_sock(sk); 2799734942ccSDave Watson err = tcp_set_ulp(sk, name); 2800734942ccSDave Watson release_sock(sk); 2801734942ccSDave Watson return err; 2802734942ccSDave Watson } 28031fba70e5SYuchung Cheng case TCP_FASTOPEN_KEY: { 28040f1ce023SJason Baron __u8 key[TCP_FASTOPEN_KEY_BUF_LENGTH]; 28050f1ce023SJason Baron __u8 *backup_key = NULL; 28061fba70e5SYuchung Cheng 28070f1ce023SJason Baron /* Allow a backup key as well to facilitate key rotation 28080f1ce023SJason Baron * First key is the active one. 28090f1ce023SJason Baron */ 28100f1ce023SJason Baron if (optlen != TCP_FASTOPEN_KEY_LENGTH && 28110f1ce023SJason Baron optlen != TCP_FASTOPEN_KEY_BUF_LENGTH) 28121fba70e5SYuchung Cheng return -EINVAL; 28131fba70e5SYuchung Cheng 28141fba70e5SYuchung Cheng if (copy_from_user(key, optval, optlen)) 28151fba70e5SYuchung Cheng return -EFAULT; 28161fba70e5SYuchung Cheng 28170f1ce023SJason Baron if (optlen == TCP_FASTOPEN_KEY_BUF_LENGTH) 28180f1ce023SJason Baron backup_key = key + TCP_FASTOPEN_KEY_LENGTH; 28190f1ce023SJason Baron 28200f1ce023SJason Baron return tcp_fastopen_reset_cipher(net, sk, key, backup_key, 28210f1ce023SJason Baron TCP_FASTOPEN_KEY_LENGTH); 28221fba70e5SYuchung Cheng } 2823e56fb50fSWilliam Allen Simpson default: 2824e56fb50fSWilliam Allen Simpson /* fallthru */ 2825e56fb50fSWilliam Allen Simpson break; 2826ccbd6a5aSJoe Perches } 28275f8ef48dSStephen Hemminger 28281da177e4SLinus Torvalds if (optlen < sizeof(int)) 28291da177e4SLinus Torvalds return -EINVAL; 28301da177e4SLinus Torvalds 28311da177e4SLinus Torvalds if (get_user(val, (int __user *)optval)) 28321da177e4SLinus Torvalds return -EFAULT; 28331da177e4SLinus Torvalds 28341da177e4SLinus Torvalds lock_sock(sk); 28351da177e4SLinus Torvalds 28361da177e4SLinus Torvalds switch (optname) { 28371da177e4SLinus Torvalds case TCP_MAXSEG: 28381da177e4SLinus Torvalds /* Values greater than interface MTU won't take effect. However 28391da177e4SLinus Torvalds * at the point when this call is done we typically don't yet 2840a777f715SRohit Chavan * know which interface is going to be used 2841a777f715SRohit Chavan */ 2842cfc62d87SGao Feng if (val && (val < TCP_MIN_MSS || val > MAX_TCP_WINDOW)) { 28431da177e4SLinus Torvalds err = -EINVAL; 28441da177e4SLinus Torvalds break; 28451da177e4SLinus Torvalds } 28461da177e4SLinus Torvalds tp->rx_opt.user_mss = val; 28471da177e4SLinus Torvalds break; 28481da177e4SLinus Torvalds 28491da177e4SLinus Torvalds case TCP_NODELAY: 28501da177e4SLinus Torvalds if (val) { 28511da177e4SLinus Torvalds /* TCP_NODELAY is weaker than TCP_CORK, so that 28521da177e4SLinus Torvalds * this option on corked socket is remembered, but 28531da177e4SLinus Torvalds * it is not activated until cork is cleared. 28541da177e4SLinus Torvalds * 28551da177e4SLinus Torvalds * However, when TCP_NODELAY is set we make 28561da177e4SLinus Torvalds * an explicit push, which overrides even TCP_CORK 28571da177e4SLinus Torvalds * for currently queued segments. 28581da177e4SLinus Torvalds */ 28591da177e4SLinus Torvalds tp->nonagle |= TCP_NAGLE_OFF|TCP_NAGLE_PUSH; 28609e412ba7SIlpo Järvinen tcp_push_pending_frames(sk); 28611da177e4SLinus Torvalds } else { 28621da177e4SLinus Torvalds tp->nonagle &= ~TCP_NAGLE_OFF; 28631da177e4SLinus Torvalds } 28641da177e4SLinus Torvalds break; 28651da177e4SLinus Torvalds 286636e31b0aSAndreas Petlund case TCP_THIN_LINEAR_TIMEOUTS: 286736e31b0aSAndreas Petlund if (val < 0 || val > 1) 286836e31b0aSAndreas Petlund err = -EINVAL; 286936e31b0aSAndreas Petlund else 287036e31b0aSAndreas Petlund tp->thin_lto = val; 287136e31b0aSAndreas Petlund break; 287236e31b0aSAndreas Petlund 28737e380175SAndreas Petlund case TCP_THIN_DUPACK: 28747e380175SAndreas Petlund if (val < 0 || val > 1) 28757e380175SAndreas Petlund err = -EINVAL; 28767e380175SAndreas Petlund break; 28777e380175SAndreas Petlund 2878ee995283SPavel Emelyanov case TCP_REPAIR: 2879ee995283SPavel Emelyanov if (!tcp_can_repair_sock(sk)) 2880ee995283SPavel Emelyanov err = -EPERM; 288131048d7aSStefan Baranoff else if (val == TCP_REPAIR_ON) { 2882ee995283SPavel Emelyanov tp->repair = 1; 2883ee995283SPavel Emelyanov sk->sk_reuse = SK_FORCE_REUSE; 2884ee995283SPavel Emelyanov tp->repair_queue = TCP_NO_QUEUE; 288531048d7aSStefan Baranoff } else if (val == TCP_REPAIR_OFF) { 2886ee995283SPavel Emelyanov tp->repair = 0; 2887ee995283SPavel Emelyanov sk->sk_reuse = SK_NO_REUSE; 2888ee995283SPavel Emelyanov tcp_send_window_probe(sk); 288931048d7aSStefan Baranoff } else if (val == TCP_REPAIR_OFF_NO_WP) { 289031048d7aSStefan Baranoff tp->repair = 0; 289131048d7aSStefan Baranoff sk->sk_reuse = SK_NO_REUSE; 2892ee995283SPavel Emelyanov } else 2893ee995283SPavel Emelyanov err = -EINVAL; 2894ee995283SPavel Emelyanov 2895ee995283SPavel Emelyanov break; 2896ee995283SPavel Emelyanov 2897ee995283SPavel Emelyanov case TCP_REPAIR_QUEUE: 2898ee995283SPavel Emelyanov if (!tp->repair) 2899ee995283SPavel Emelyanov err = -EPERM; 2900bf2acc94SEric Dumazet else if ((unsigned int)val < TCP_QUEUES_NR) 2901ee995283SPavel Emelyanov tp->repair_queue = val; 2902ee995283SPavel Emelyanov else 2903ee995283SPavel Emelyanov err = -EINVAL; 2904ee995283SPavel Emelyanov break; 2905ee995283SPavel Emelyanov 2906ee995283SPavel Emelyanov case TCP_QUEUE_SEQ: 2907ee995283SPavel Emelyanov if (sk->sk_state != TCP_CLOSE) 2908ee995283SPavel Emelyanov err = -EPERM; 2909ee995283SPavel Emelyanov else if (tp->repair_queue == TCP_SEND_QUEUE) 2910ee995283SPavel Emelyanov tp->write_seq = val; 2911ee995283SPavel Emelyanov else if (tp->repair_queue == TCP_RECV_QUEUE) 2912ee995283SPavel Emelyanov tp->rcv_nxt = val; 2913ee995283SPavel Emelyanov else 2914ee995283SPavel Emelyanov err = -EINVAL; 2915ee995283SPavel Emelyanov break; 2916ee995283SPavel Emelyanov 2917b139ba4eSPavel Emelyanov case TCP_REPAIR_OPTIONS: 2918b139ba4eSPavel Emelyanov if (!tp->repair) 2919b139ba4eSPavel Emelyanov err = -EINVAL; 2920b139ba4eSPavel Emelyanov else if (sk->sk_state == TCP_ESTABLISHED) 292115e56515SDouglas Caetano dos Santos err = tcp_repair_options_est(sk, 2922de248a75SPavel Emelyanov (struct tcp_repair_opt __user *)optval, 2923de248a75SPavel Emelyanov optlen); 2924b139ba4eSPavel Emelyanov else 2925b139ba4eSPavel Emelyanov err = -EPERM; 2926b139ba4eSPavel Emelyanov break; 2927b139ba4eSPavel Emelyanov 29281da177e4SLinus Torvalds case TCP_CORK: 29291da177e4SLinus Torvalds /* When set indicates to always queue non-full frames. 29301da177e4SLinus Torvalds * Later the user clears this option and we transmit 29311da177e4SLinus Torvalds * any pending partial frames in the queue. This is 29321da177e4SLinus Torvalds * meant to be used alongside sendfile() to get properly 29331da177e4SLinus Torvalds * filled frames when the user (for example) must write 29341da177e4SLinus Torvalds * out headers with a write() call first and then use 29351da177e4SLinus Torvalds * sendfile to send out the data parts. 29361da177e4SLinus Torvalds * 29371da177e4SLinus Torvalds * TCP_CORK can be set together with TCP_NODELAY and it is 29381da177e4SLinus Torvalds * stronger than TCP_NODELAY. 29391da177e4SLinus Torvalds */ 29401da177e4SLinus Torvalds if (val) { 29411da177e4SLinus Torvalds tp->nonagle |= TCP_NAGLE_CORK; 29421da177e4SLinus Torvalds } else { 29431da177e4SLinus Torvalds tp->nonagle &= ~TCP_NAGLE_CORK; 29441da177e4SLinus Torvalds if (tp->nonagle&TCP_NAGLE_OFF) 29451da177e4SLinus Torvalds tp->nonagle |= TCP_NAGLE_PUSH; 29469e412ba7SIlpo Järvinen tcp_push_pending_frames(sk); 29471da177e4SLinus Torvalds } 29481da177e4SLinus Torvalds break; 29491da177e4SLinus Torvalds 29501da177e4SLinus Torvalds case TCP_KEEPIDLE: 29511da177e4SLinus Torvalds if (val < 1 || val > MAX_TCP_KEEPIDLE) 29521da177e4SLinus Torvalds err = -EINVAL; 29531da177e4SLinus Torvalds else { 29541da177e4SLinus Torvalds tp->keepalive_time = val * HZ; 29551da177e4SLinus Torvalds if (sock_flag(sk, SOCK_KEEPOPEN) && 29561da177e4SLinus Torvalds !((1 << sk->sk_state) & 29571da177e4SLinus Torvalds (TCPF_CLOSE | TCPF_LISTEN))) { 29586c37e5deSFlavio Leitner u32 elapsed = keepalive_time_elapsed(tp); 29591da177e4SLinus Torvalds if (tp->keepalive_time > elapsed) 29601da177e4SLinus Torvalds elapsed = tp->keepalive_time - elapsed; 29611da177e4SLinus Torvalds else 29621da177e4SLinus Torvalds elapsed = 0; 2963463c84b9SArnaldo Carvalho de Melo inet_csk_reset_keepalive_timer(sk, elapsed); 29641da177e4SLinus Torvalds } 29651da177e4SLinus Torvalds } 29661da177e4SLinus Torvalds break; 29671da177e4SLinus Torvalds case TCP_KEEPINTVL: 29681da177e4SLinus Torvalds if (val < 1 || val > MAX_TCP_KEEPINTVL) 29691da177e4SLinus Torvalds err = -EINVAL; 29701da177e4SLinus Torvalds else 29711da177e4SLinus Torvalds tp->keepalive_intvl = val * HZ; 29721da177e4SLinus Torvalds break; 29731da177e4SLinus Torvalds case TCP_KEEPCNT: 29741da177e4SLinus Torvalds if (val < 1 || val > MAX_TCP_KEEPCNT) 29751da177e4SLinus Torvalds err = -EINVAL; 29761da177e4SLinus Torvalds else 29771da177e4SLinus Torvalds tp->keepalive_probes = val; 29781da177e4SLinus Torvalds break; 29791da177e4SLinus Torvalds case TCP_SYNCNT: 29801da177e4SLinus Torvalds if (val < 1 || val > MAX_TCP_SYNCNT) 29811da177e4SLinus Torvalds err = -EINVAL; 29821da177e4SLinus Torvalds else 2983463c84b9SArnaldo Carvalho de Melo icsk->icsk_syn_retries = val; 29841da177e4SLinus Torvalds break; 29851da177e4SLinus Torvalds 2986cd8ae852SEric Dumazet case TCP_SAVE_SYN: 2987cd8ae852SEric Dumazet if (val < 0 || val > 1) 2988cd8ae852SEric Dumazet err = -EINVAL; 2989cd8ae852SEric Dumazet else 2990cd8ae852SEric Dumazet tp->save_syn = val; 2991cd8ae852SEric Dumazet break; 2992cd8ae852SEric Dumazet 29931da177e4SLinus Torvalds case TCP_LINGER2: 29941da177e4SLinus Torvalds if (val < 0) 29951da177e4SLinus Torvalds tp->linger2 = -1; 29961e579caaSNikolay Borisov else if (val > net->ipv4.sysctl_tcp_fin_timeout / HZ) 29971da177e4SLinus Torvalds tp->linger2 = 0; 29981da177e4SLinus Torvalds else 29991da177e4SLinus Torvalds tp->linger2 = val * HZ; 30001da177e4SLinus Torvalds break; 30011da177e4SLinus Torvalds 30021da177e4SLinus Torvalds case TCP_DEFER_ACCEPT: 3003b103cf34SJulian Anastasov /* Translate value in seconds to number of retransmits */ 3004b103cf34SJulian Anastasov icsk->icsk_accept_queue.rskq_defer_accept = 3005b103cf34SJulian Anastasov secs_to_retrans(val, TCP_TIMEOUT_INIT / HZ, 3006b103cf34SJulian Anastasov TCP_RTO_MAX / HZ); 30071da177e4SLinus Torvalds break; 30081da177e4SLinus Torvalds 30091da177e4SLinus Torvalds case TCP_WINDOW_CLAMP: 30101da177e4SLinus Torvalds if (!val) { 30111da177e4SLinus Torvalds if (sk->sk_state != TCP_CLOSE) { 30121da177e4SLinus Torvalds err = -EINVAL; 30131da177e4SLinus Torvalds break; 30141da177e4SLinus Torvalds } 30151da177e4SLinus Torvalds tp->window_clamp = 0; 30161da177e4SLinus Torvalds } else 30171da177e4SLinus Torvalds tp->window_clamp = val < SOCK_MIN_RCVBUF / 2 ? 30181da177e4SLinus Torvalds SOCK_MIN_RCVBUF / 2 : val; 30191da177e4SLinus Torvalds break; 30201da177e4SLinus Torvalds 30211da177e4SLinus Torvalds case TCP_QUICKACK: 30221da177e4SLinus Torvalds if (!val) { 302331954cd8SWei Wang inet_csk_enter_pingpong_mode(sk); 30241da177e4SLinus Torvalds } else { 302531954cd8SWei Wang inet_csk_exit_pingpong_mode(sk); 30261da177e4SLinus Torvalds if ((1 << sk->sk_state) & 30271da177e4SLinus Torvalds (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT) && 3028463c84b9SArnaldo Carvalho de Melo inet_csk_ack_scheduled(sk)) { 3029463c84b9SArnaldo Carvalho de Melo icsk->icsk_ack.pending |= ICSK_ACK_PUSHED; 30300e4b4992SChris Leech tcp_cleanup_rbuf(sk, 1); 30311da177e4SLinus Torvalds if (!(val & 1)) 303231954cd8SWei Wang inet_csk_enter_pingpong_mode(sk); 30331da177e4SLinus Torvalds } 30341da177e4SLinus Torvalds } 30351da177e4SLinus Torvalds break; 30361da177e4SLinus Torvalds 3037cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG 3038cfb6eeb4SYOSHIFUJI Hideaki case TCP_MD5SIG: 30398917a777SIvan Delalande case TCP_MD5SIG_EXT: 304072123032SEric Dumazet if ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)) 30418917a777SIvan Delalande err = tp->af_specific->md5_parse(sk, optname, optval, optlen); 304272123032SEric Dumazet else 304372123032SEric Dumazet err = -EINVAL; 3044cfb6eeb4SYOSHIFUJI Hideaki break; 3045cfb6eeb4SYOSHIFUJI Hideaki #endif 3046dca43c75SJerry Chu case TCP_USER_TIMEOUT: 3047b248230cSYuchung Cheng /* Cap the max time in ms TCP will retry or probe the window 3048dca43c75SJerry Chu * before giving up and aborting (ETIMEDOUT) a connection. 3049dca43c75SJerry Chu */ 305042493570SHangbin Liu if (val < 0) 305142493570SHangbin Liu err = -EINVAL; 305242493570SHangbin Liu else 30539bcc66e1SJon Maxwell icsk->icsk_user_timeout = val; 3054dca43c75SJerry Chu break; 30558336886fSJerry Chu 30568336886fSJerry Chu case TCP_FASTOPEN: 30578336886fSJerry Chu if (val >= 0 && ((1 << sk->sk_state) & (TCPF_CLOSE | 3058dfea2aa6SChristoph Paasch TCPF_LISTEN))) { 305943713848SHaishuang Yan tcp_fastopen_init_key_once(net); 3060dfea2aa6SChristoph Paasch 30610536fcc0SEric Dumazet fastopen_queue_tune(sk, val); 3062dfea2aa6SChristoph Paasch } else { 30638336886fSJerry Chu err = -EINVAL; 3064dfea2aa6SChristoph Paasch } 30658336886fSJerry Chu break; 306619f6d3f3SWei Wang case TCP_FASTOPEN_CONNECT: 306719f6d3f3SWei Wang if (val > 1 || val < 0) { 306819f6d3f3SWei Wang err = -EINVAL; 3069e1cfcbe8SHaishuang Yan } else if (net->ipv4.sysctl_tcp_fastopen & TFO_CLIENT_ENABLE) { 307019f6d3f3SWei Wang if (sk->sk_state == TCP_CLOSE) 307119f6d3f3SWei Wang tp->fastopen_connect = val; 307219f6d3f3SWei Wang else 307319f6d3f3SWei Wang err = -EINVAL; 307419f6d3f3SWei Wang } else { 307519f6d3f3SWei Wang err = -EOPNOTSUPP; 307619f6d3f3SWei Wang } 307719f6d3f3SWei Wang break; 307871c02379SChristoph Paasch case TCP_FASTOPEN_NO_COOKIE: 307971c02379SChristoph Paasch if (val > 1 || val < 0) 308071c02379SChristoph Paasch err = -EINVAL; 308171c02379SChristoph Paasch else if (!((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN))) 308271c02379SChristoph Paasch err = -EINVAL; 308371c02379SChristoph Paasch else 308471c02379SChristoph Paasch tp->fastopen_no_cookie = val; 308571c02379SChristoph Paasch break; 308693be6ce0SAndrey Vagin case TCP_TIMESTAMP: 308793be6ce0SAndrey Vagin if (!tp->repair) 308893be6ce0SAndrey Vagin err = -EPERM; 308993be6ce0SAndrey Vagin else 30909a568de4SEric Dumazet tp->tsoffset = val - tcp_time_stamp_raw(); 309193be6ce0SAndrey Vagin break; 3092b1ed4c4fSAndrey Vagin case TCP_REPAIR_WINDOW: 3093b1ed4c4fSAndrey Vagin err = tcp_repair_set_window(tp, optval, optlen); 3094b1ed4c4fSAndrey Vagin break; 3095c9bee3b7SEric Dumazet case TCP_NOTSENT_LOWAT: 3096c9bee3b7SEric Dumazet tp->notsent_lowat = val; 3097c9bee3b7SEric Dumazet sk->sk_write_space(sk); 3098c9bee3b7SEric Dumazet break; 3099b75eba76SSoheil Hassas Yeganeh case TCP_INQ: 3100b75eba76SSoheil Hassas Yeganeh if (val > 1 || val < 0) 3101b75eba76SSoheil Hassas Yeganeh err = -EINVAL; 3102b75eba76SSoheil Hassas Yeganeh else 3103b75eba76SSoheil Hassas Yeganeh tp->recvmsg_inq = val; 3104b75eba76SSoheil Hassas Yeganeh break; 3105*a842fe14SEric Dumazet case TCP_TX_DELAY: 3106*a842fe14SEric Dumazet if (val) 3107*a842fe14SEric Dumazet tcp_enable_tx_delay(); 3108*a842fe14SEric Dumazet tp->tcp_tx_delay = val; 3109*a842fe14SEric Dumazet break; 31101da177e4SLinus Torvalds default: 31111da177e4SLinus Torvalds err = -ENOPROTOOPT; 31121da177e4SLinus Torvalds break; 31133ff50b79SStephen Hemminger } 31143ff50b79SStephen Hemminger 31151da177e4SLinus Torvalds release_sock(sk); 31161da177e4SLinus Torvalds return err; 31171da177e4SLinus Torvalds } 31181da177e4SLinus Torvalds 31193fdadf7dSDmitry Mishin int tcp_setsockopt(struct sock *sk, int level, int optname, char __user *optval, 3120b7058842SDavid S. Miller unsigned int optlen) 31213fdadf7dSDmitry Mishin { 3122cf533ea5SEric Dumazet const struct inet_connection_sock *icsk = inet_csk(sk); 31233fdadf7dSDmitry Mishin 31243fdadf7dSDmitry Mishin if (level != SOL_TCP) 31253fdadf7dSDmitry Mishin return icsk->icsk_af_ops->setsockopt(sk, level, optname, 31263fdadf7dSDmitry Mishin optval, optlen); 31273fdadf7dSDmitry Mishin return do_tcp_setsockopt(sk, level, optname, optval, optlen); 31283fdadf7dSDmitry Mishin } 31294bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_setsockopt); 31303fdadf7dSDmitry Mishin 31313fdadf7dSDmitry Mishin #ifdef CONFIG_COMPAT 3132543d9cfeSArnaldo Carvalho de Melo int compat_tcp_setsockopt(struct sock *sk, int level, int optname, 3133b7058842SDavid S. Miller char __user *optval, unsigned int optlen) 31343fdadf7dSDmitry Mishin { 3135dec73ff0SArnaldo Carvalho de Melo if (level != SOL_TCP) 3136dec73ff0SArnaldo Carvalho de Melo return inet_csk_compat_setsockopt(sk, level, optname, 3137dec73ff0SArnaldo Carvalho de Melo optval, optlen); 31383fdadf7dSDmitry Mishin return do_tcp_setsockopt(sk, level, optname, optval, optlen); 31393fdadf7dSDmitry Mishin } 3140543d9cfeSArnaldo Carvalho de Melo EXPORT_SYMBOL(compat_tcp_setsockopt); 31413fdadf7dSDmitry Mishin #endif 31423fdadf7dSDmitry Mishin 3143efd90174SFrancis Yan static void tcp_get_info_chrono_stats(const struct tcp_sock *tp, 3144efd90174SFrancis Yan struct tcp_info *info) 3145efd90174SFrancis Yan { 3146efd90174SFrancis Yan u64 stats[__TCP_CHRONO_MAX], total = 0; 3147efd90174SFrancis Yan enum tcp_chrono i; 3148efd90174SFrancis Yan 3149efd90174SFrancis Yan for (i = TCP_CHRONO_BUSY; i < __TCP_CHRONO_MAX; ++i) { 3150efd90174SFrancis Yan stats[i] = tp->chrono_stat[i - 1]; 3151efd90174SFrancis Yan if (i == tp->chrono_type) 3152628174ccSEric Dumazet stats[i] += tcp_jiffies32 - tp->chrono_start; 3153efd90174SFrancis Yan stats[i] *= USEC_PER_SEC / HZ; 3154efd90174SFrancis Yan total += stats[i]; 3155efd90174SFrancis Yan } 3156efd90174SFrancis Yan 3157efd90174SFrancis Yan info->tcpi_busy_time = total; 3158efd90174SFrancis Yan info->tcpi_rwnd_limited = stats[TCP_CHRONO_RWND_LIMITED]; 3159efd90174SFrancis Yan info->tcpi_sndbuf_limited = stats[TCP_CHRONO_SNDBUF_LIMITED]; 3160efd90174SFrancis Yan } 3161efd90174SFrancis Yan 31621da177e4SLinus Torvalds /* Return information about state of tcp endpoint in API format. */ 31630df48c26SEric Dumazet void tcp_get_info(struct sock *sk, struct tcp_info *info) 31641da177e4SLinus Torvalds { 316535ac838aSCraig Gallek const struct tcp_sock *tp = tcp_sk(sk); /* iff sk_type == SOCK_STREAM */ 3166463c84b9SArnaldo Carvalho de Melo const struct inet_connection_sock *icsk = inet_csk(sk); 316776a9ebe8SEric Dumazet unsigned long rate; 31680263598cSWei Wang u32 now; 3169ff5d7497SEric Dumazet u64 rate64; 317067db3e4bSEric Dumazet bool slow; 31711da177e4SLinus Torvalds 31721da177e4SLinus Torvalds memset(info, 0, sizeof(*info)); 317335ac838aSCraig Gallek if (sk->sk_type != SOCK_STREAM) 317435ac838aSCraig Gallek return; 31751da177e4SLinus Torvalds 3176986ffdfdSYafang Shao info->tcpi_state = inet_sk_state_load(sk); 317700fd38d9SEric Dumazet 3178ccbf3bfaSEric Dumazet /* Report meaningful fields for all TCP states, including listeners */ 3179ccbf3bfaSEric Dumazet rate = READ_ONCE(sk->sk_pacing_rate); 318076a9ebe8SEric Dumazet rate64 = (rate != ~0UL) ? rate : ~0ULL; 3181f522a5fcSEric Dumazet info->tcpi_pacing_rate = rate64; 3182ccbf3bfaSEric Dumazet 3183ccbf3bfaSEric Dumazet rate = READ_ONCE(sk->sk_max_pacing_rate); 318476a9ebe8SEric Dumazet rate64 = (rate != ~0UL) ? rate : ~0ULL; 3185f522a5fcSEric Dumazet info->tcpi_max_pacing_rate = rate64; 3186ccbf3bfaSEric Dumazet 3187ccbf3bfaSEric Dumazet info->tcpi_reordering = tp->reordering; 3188ccbf3bfaSEric Dumazet info->tcpi_snd_cwnd = tp->snd_cwnd; 3189ccbf3bfaSEric Dumazet 3190ccbf3bfaSEric Dumazet if (info->tcpi_state == TCP_LISTEN) { 3191ccbf3bfaSEric Dumazet /* listeners aliased fields : 3192ccbf3bfaSEric Dumazet * tcpi_unacked -> Number of children ready for accept() 3193ccbf3bfaSEric Dumazet * tcpi_sacked -> max backlog 3194ccbf3bfaSEric Dumazet */ 3195ccbf3bfaSEric Dumazet info->tcpi_unacked = sk->sk_ack_backlog; 3196ccbf3bfaSEric Dumazet info->tcpi_sacked = sk->sk_max_ack_backlog; 3197ccbf3bfaSEric Dumazet return; 3198ccbf3bfaSEric Dumazet } 3199b369e7fdSEric Dumazet 3200b369e7fdSEric Dumazet slow = lock_sock_fast(sk); 3201b369e7fdSEric Dumazet 32026687e988SArnaldo Carvalho de Melo info->tcpi_ca_state = icsk->icsk_ca_state; 3203463c84b9SArnaldo Carvalho de Melo info->tcpi_retransmits = icsk->icsk_retransmits; 32046687e988SArnaldo Carvalho de Melo info->tcpi_probes = icsk->icsk_probes_out; 3205463c84b9SArnaldo Carvalho de Melo info->tcpi_backoff = icsk->icsk_backoff; 32061da177e4SLinus Torvalds 32071da177e4SLinus Torvalds if (tp->rx_opt.tstamp_ok) 32081da177e4SLinus Torvalds info->tcpi_options |= TCPI_OPT_TIMESTAMPS; 3209e60402d0SIlpo Järvinen if (tcp_is_sack(tp)) 32101da177e4SLinus Torvalds info->tcpi_options |= TCPI_OPT_SACK; 32111da177e4SLinus Torvalds if (tp->rx_opt.wscale_ok) { 32121da177e4SLinus Torvalds info->tcpi_options |= TCPI_OPT_WSCALE; 32131da177e4SLinus Torvalds info->tcpi_snd_wscale = tp->rx_opt.snd_wscale; 32141da177e4SLinus Torvalds info->tcpi_rcv_wscale = tp->rx_opt.rcv_wscale; 32151da177e4SLinus Torvalds } 32161da177e4SLinus Torvalds 32171da177e4SLinus Torvalds if (tp->ecn_flags & TCP_ECN_OK) 32181da177e4SLinus Torvalds info->tcpi_options |= TCPI_OPT_ECN; 3219b5c5693bSEric Dumazet if (tp->ecn_flags & TCP_ECN_SEEN) 3220b5c5693bSEric Dumazet info->tcpi_options |= TCPI_OPT_ECN_SEEN; 32216f73601eSYuchung Cheng if (tp->syn_data_acked) 32226f73601eSYuchung Cheng info->tcpi_options |= TCPI_OPT_SYN_DATA; 32231da177e4SLinus Torvalds 3224463c84b9SArnaldo Carvalho de Melo info->tcpi_rto = jiffies_to_usecs(icsk->icsk_rto); 3225463c84b9SArnaldo Carvalho de Melo info->tcpi_ato = jiffies_to_usecs(icsk->icsk_ack.ato); 3226c1b4a7e6SDavid S. Miller info->tcpi_snd_mss = tp->mss_cache; 3227463c84b9SArnaldo Carvalho de Melo info->tcpi_rcv_mss = icsk->icsk_ack.rcv_mss; 32281da177e4SLinus Torvalds 32291da177e4SLinus Torvalds info->tcpi_unacked = tp->packets_out; 32301da177e4SLinus Torvalds info->tcpi_sacked = tp->sacked_out; 3231ccbf3bfaSEric Dumazet 32321da177e4SLinus Torvalds info->tcpi_lost = tp->lost_out; 32331da177e4SLinus Torvalds info->tcpi_retrans = tp->retrans_out; 32341da177e4SLinus Torvalds 3235d635fbe2SEric Dumazet now = tcp_jiffies32; 32361da177e4SLinus Torvalds info->tcpi_last_data_sent = jiffies_to_msecs(now - tp->lsndtime); 3237463c84b9SArnaldo Carvalho de Melo info->tcpi_last_data_recv = jiffies_to_msecs(now - icsk->icsk_ack.lrcvtime); 32381da177e4SLinus Torvalds info->tcpi_last_ack_recv = jiffies_to_msecs(now - tp->rcv_tstamp); 32391da177e4SLinus Torvalds 3240d83d8461SArnaldo Carvalho de Melo info->tcpi_pmtu = icsk->icsk_pmtu_cookie; 32411da177e4SLinus Torvalds info->tcpi_rcv_ssthresh = tp->rcv_ssthresh; 3242740b0f18SEric Dumazet info->tcpi_rtt = tp->srtt_us >> 3; 3243740b0f18SEric Dumazet info->tcpi_rttvar = tp->mdev_us >> 2; 32441da177e4SLinus Torvalds info->tcpi_snd_ssthresh = tp->snd_ssthresh; 32451da177e4SLinus Torvalds info->tcpi_advmss = tp->advmss; 32461da177e4SLinus Torvalds 3247645f4c6fSEric Dumazet info->tcpi_rcv_rtt = tp->rcv_rtt_est.rtt_us >> 3; 32481da177e4SLinus Torvalds info->tcpi_rcv_space = tp->rcvq_space.space; 32491da177e4SLinus Torvalds 32501da177e4SLinus Torvalds info->tcpi_total_retrans = tp->total_retrans; 3251977cb0ecSEric Dumazet 3252f522a5fcSEric Dumazet info->tcpi_bytes_acked = tp->bytes_acked; 3253f522a5fcSEric Dumazet info->tcpi_bytes_received = tp->bytes_received; 325467db3e4bSEric Dumazet info->tcpi_notsent_bytes = max_t(int, 0, tp->write_seq - tp->snd_nxt); 3255efd90174SFrancis Yan tcp_get_info_chrono_stats(tp, info); 325667db3e4bSEric Dumazet 32572efd055cSMarcelo Ricardo Leitner info->tcpi_segs_out = tp->segs_out; 32582efd055cSMarcelo Ricardo Leitner info->tcpi_segs_in = tp->segs_in; 3259cd9b2660SEric Dumazet 3260cd9b2660SEric Dumazet info->tcpi_min_rtt = tcp_min_rtt(tp); 3261a44d6eacSMartin KaFai Lau info->tcpi_data_segs_in = tp->data_segs_in; 3262a44d6eacSMartin KaFai Lau info->tcpi_data_segs_out = tp->data_segs_out; 3263eb8329e0SYuchung Cheng 3264eb8329e0SYuchung Cheng info->tcpi_delivery_rate_app_limited = tp->rate_app_limited ? 1 : 0; 32650263598cSWei Wang rate64 = tcp_compute_delivery_rate(tp); 32660263598cSWei Wang if (rate64) 3267f522a5fcSEric Dumazet info->tcpi_delivery_rate = rate64; 3268feb5f2ecSYuchung Cheng info->tcpi_delivered = tp->delivered; 3269feb5f2ecSYuchung Cheng info->tcpi_delivered_ce = tp->delivered_ce; 3270ba113c3aSWei Wang info->tcpi_bytes_sent = tp->bytes_sent; 3271fb31c9b9SWei Wang info->tcpi_bytes_retrans = tp->bytes_retrans; 32727e10b655SWei Wang info->tcpi_dsack_dups = tp->dsack_dups; 32737ec65372SWei Wang info->tcpi_reord_seen = tp->reord_seen; 3274b369e7fdSEric Dumazet unlock_sock_fast(sk, slow); 32751da177e4SLinus Torvalds } 32761da177e4SLinus Torvalds EXPORT_SYMBOL_GPL(tcp_get_info); 32771da177e4SLinus Torvalds 3278984988aaSWei Wang static size_t tcp_opt_stats_get_size(void) 3279984988aaSWei Wang { 3280984988aaSWei Wang return 3281984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BUSY */ 3282984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_RWND_LIMITED */ 3283984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_SNDBUF_LIMITED */ 3284984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DATA_SEGS_OUT */ 3285984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_TOTAL_RETRANS */ 3286984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_PACING_RATE */ 3287984988aaSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_DELIVERY_RATE */ 3288984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_CWND */ 3289984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_REORDERING */ 3290984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_MIN_RTT */ 3291984988aaSWei Wang nla_total_size(sizeof(u8)) + /* TCP_NLA_RECUR_RETRANS */ 3292984988aaSWei Wang nla_total_size(sizeof(u8)) + /* TCP_NLA_DELIVERY_RATE_APP_LMT */ 3293984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_SNDQ_SIZE */ 3294984988aaSWei Wang nla_total_size(sizeof(u8)) + /* TCP_NLA_CA_STATE */ 3295984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_SND_SSTHRESH */ 3296984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED */ 3297984988aaSWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_DELIVERED_CE */ 3298ba113c3aSWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_SENT */ 3299fb31c9b9SWei Wang nla_total_size_64bit(sizeof(u64)) + /* TCP_NLA_BYTES_RETRANS */ 33007e10b655SWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_DSACK_DUPS */ 33017ec65372SWei Wang nla_total_size(sizeof(u32)) + /* TCP_NLA_REORD_SEEN */ 3302e8bd8fcaSYousuk Seung nla_total_size(sizeof(u32)) + /* TCP_NLA_SRTT */ 3303984988aaSWei Wang 0; 3304984988aaSWei Wang } 3305984988aaSWei Wang 33061c885808SFrancis Yan struct sk_buff *tcp_get_timestamping_opt_stats(const struct sock *sk) 33071c885808SFrancis Yan { 33081c885808SFrancis Yan const struct tcp_sock *tp = tcp_sk(sk); 33091c885808SFrancis Yan struct sk_buff *stats; 33101c885808SFrancis Yan struct tcp_info info; 331176a9ebe8SEric Dumazet unsigned long rate; 3312bb7c19f9SWei Wang u64 rate64; 33131c885808SFrancis Yan 3314984988aaSWei Wang stats = alloc_skb(tcp_opt_stats_get_size(), GFP_ATOMIC); 33151c885808SFrancis Yan if (!stats) 33161c885808SFrancis Yan return NULL; 33171c885808SFrancis Yan 33181c885808SFrancis Yan tcp_get_info_chrono_stats(tp, &info); 33191c885808SFrancis Yan nla_put_u64_64bit(stats, TCP_NLA_BUSY, 33201c885808SFrancis Yan info.tcpi_busy_time, TCP_NLA_PAD); 33211c885808SFrancis Yan nla_put_u64_64bit(stats, TCP_NLA_RWND_LIMITED, 33221c885808SFrancis Yan info.tcpi_rwnd_limited, TCP_NLA_PAD); 33231c885808SFrancis Yan nla_put_u64_64bit(stats, TCP_NLA_SNDBUF_LIMITED, 33241c885808SFrancis Yan info.tcpi_sndbuf_limited, TCP_NLA_PAD); 33257e98102fSYuchung Cheng nla_put_u64_64bit(stats, TCP_NLA_DATA_SEGS_OUT, 33267e98102fSYuchung Cheng tp->data_segs_out, TCP_NLA_PAD); 33277e98102fSYuchung Cheng nla_put_u64_64bit(stats, TCP_NLA_TOTAL_RETRANS, 33287e98102fSYuchung Cheng tp->total_retrans, TCP_NLA_PAD); 3329bb7c19f9SWei Wang 3330bb7c19f9SWei Wang rate = READ_ONCE(sk->sk_pacing_rate); 333176a9ebe8SEric Dumazet rate64 = (rate != ~0UL) ? rate : ~0ULL; 3332bb7c19f9SWei Wang nla_put_u64_64bit(stats, TCP_NLA_PACING_RATE, rate64, TCP_NLA_PAD); 3333bb7c19f9SWei Wang 3334bb7c19f9SWei Wang rate64 = tcp_compute_delivery_rate(tp); 3335bb7c19f9SWei Wang nla_put_u64_64bit(stats, TCP_NLA_DELIVERY_RATE, rate64, TCP_NLA_PAD); 3336bb7c19f9SWei Wang 3337bb7c19f9SWei Wang nla_put_u32(stats, TCP_NLA_SND_CWND, tp->snd_cwnd); 3338bb7c19f9SWei Wang nla_put_u32(stats, TCP_NLA_REORDERING, tp->reordering); 3339bb7c19f9SWei Wang nla_put_u32(stats, TCP_NLA_MIN_RTT, tcp_min_rtt(tp)); 3340bb7c19f9SWei Wang 3341bb7c19f9SWei Wang nla_put_u8(stats, TCP_NLA_RECUR_RETRANS, inet_csk(sk)->icsk_retransmits); 3342bb7c19f9SWei Wang nla_put_u8(stats, TCP_NLA_DELIVERY_RATE_APP_LMT, !!tp->rate_app_limited); 33437156d194SYousuk Seung nla_put_u32(stats, TCP_NLA_SND_SSTHRESH, tp->snd_ssthresh); 3344feb5f2ecSYuchung Cheng nla_put_u32(stats, TCP_NLA_DELIVERED, tp->delivered); 3345feb5f2ecSYuchung Cheng nla_put_u32(stats, TCP_NLA_DELIVERED_CE, tp->delivered_ce); 334687ecc95dSPriyaranjan Jha 334787ecc95dSPriyaranjan Jha nla_put_u32(stats, TCP_NLA_SNDQ_SIZE, tp->write_seq - tp->snd_una); 3348be631892SPriyaranjan Jha nla_put_u8(stats, TCP_NLA_CA_STATE, inet_csk(sk)->icsk_ca_state); 3349feb5f2ecSYuchung Cheng 3350ba113c3aSWei Wang nla_put_u64_64bit(stats, TCP_NLA_BYTES_SENT, tp->bytes_sent, 3351ba113c3aSWei Wang TCP_NLA_PAD); 3352fb31c9b9SWei Wang nla_put_u64_64bit(stats, TCP_NLA_BYTES_RETRANS, tp->bytes_retrans, 3353fb31c9b9SWei Wang TCP_NLA_PAD); 33547e10b655SWei Wang nla_put_u32(stats, TCP_NLA_DSACK_DUPS, tp->dsack_dups); 33557ec65372SWei Wang nla_put_u32(stats, TCP_NLA_REORD_SEEN, tp->reord_seen); 3356e8bd8fcaSYousuk Seung nla_put_u32(stats, TCP_NLA_SRTT, tp->srtt_us >> 3); 3357ba113c3aSWei Wang 33581c885808SFrancis Yan return stats; 33591c885808SFrancis Yan } 33601c885808SFrancis Yan 33613fdadf7dSDmitry Mishin static int do_tcp_getsockopt(struct sock *sk, int level, 33623fdadf7dSDmitry Mishin int optname, char __user *optval, int __user *optlen) 33631da177e4SLinus Torvalds { 3364295f7324SArnaldo Carvalho de Melo struct inet_connection_sock *icsk = inet_csk(sk); 33651da177e4SLinus Torvalds struct tcp_sock *tp = tcp_sk(sk); 33666fa25166SNikolay Borisov struct net *net = sock_net(sk); 33671da177e4SLinus Torvalds int val, len; 33681da177e4SLinus Torvalds 33691da177e4SLinus Torvalds if (get_user(len, optlen)) 33701da177e4SLinus Torvalds return -EFAULT; 33711da177e4SLinus Torvalds 33721da177e4SLinus Torvalds len = min_t(unsigned int, len, sizeof(int)); 33731da177e4SLinus Torvalds 33741da177e4SLinus Torvalds if (len < 0) 33751da177e4SLinus Torvalds return -EINVAL; 33761da177e4SLinus Torvalds 33771da177e4SLinus Torvalds switch (optname) { 33781da177e4SLinus Torvalds case TCP_MAXSEG: 3379c1b4a7e6SDavid S. Miller val = tp->mss_cache; 33801da177e4SLinus Torvalds if (!val && ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN))) 33811da177e4SLinus Torvalds val = tp->rx_opt.user_mss; 33825e6a3ce6SPavel Emelyanov if (tp->repair) 33835e6a3ce6SPavel Emelyanov val = tp->rx_opt.mss_clamp; 33841da177e4SLinus Torvalds break; 33851da177e4SLinus Torvalds case TCP_NODELAY: 33861da177e4SLinus Torvalds val = !!(tp->nonagle&TCP_NAGLE_OFF); 33871da177e4SLinus Torvalds break; 33881da177e4SLinus Torvalds case TCP_CORK: 33891da177e4SLinus Torvalds val = !!(tp->nonagle&TCP_NAGLE_CORK); 33901da177e4SLinus Torvalds break; 33911da177e4SLinus Torvalds case TCP_KEEPIDLE: 3392df19a626SEric Dumazet val = keepalive_time_when(tp) / HZ; 33931da177e4SLinus Torvalds break; 33941da177e4SLinus Torvalds case TCP_KEEPINTVL: 3395df19a626SEric Dumazet val = keepalive_intvl_when(tp) / HZ; 33961da177e4SLinus Torvalds break; 33971da177e4SLinus Torvalds case TCP_KEEPCNT: 3398df19a626SEric Dumazet val = keepalive_probes(tp); 33991da177e4SLinus Torvalds break; 34001da177e4SLinus Torvalds case TCP_SYNCNT: 34016fa25166SNikolay Borisov val = icsk->icsk_syn_retries ? : net->ipv4.sysctl_tcp_syn_retries; 34021da177e4SLinus Torvalds break; 34031da177e4SLinus Torvalds case TCP_LINGER2: 34041da177e4SLinus Torvalds val = tp->linger2; 34051da177e4SLinus Torvalds if (val >= 0) 34061e579caaSNikolay Borisov val = (val ? : net->ipv4.sysctl_tcp_fin_timeout) / HZ; 34071da177e4SLinus Torvalds break; 34081da177e4SLinus Torvalds case TCP_DEFER_ACCEPT: 3409b103cf34SJulian Anastasov val = retrans_to_secs(icsk->icsk_accept_queue.rskq_defer_accept, 3410b103cf34SJulian Anastasov TCP_TIMEOUT_INIT / HZ, TCP_RTO_MAX / HZ); 34111da177e4SLinus Torvalds break; 34121da177e4SLinus Torvalds case TCP_WINDOW_CLAMP: 34131da177e4SLinus Torvalds val = tp->window_clamp; 34141da177e4SLinus Torvalds break; 34151da177e4SLinus Torvalds case TCP_INFO: { 34161da177e4SLinus Torvalds struct tcp_info info; 34171da177e4SLinus Torvalds 34181da177e4SLinus Torvalds if (get_user(len, optlen)) 34191da177e4SLinus Torvalds return -EFAULT; 34201da177e4SLinus Torvalds 34211da177e4SLinus Torvalds tcp_get_info(sk, &info); 34221da177e4SLinus Torvalds 34231da177e4SLinus Torvalds len = min_t(unsigned int, len, sizeof(info)); 34241da177e4SLinus Torvalds if (put_user(len, optlen)) 34251da177e4SLinus Torvalds return -EFAULT; 34261da177e4SLinus Torvalds if (copy_to_user(optval, &info, len)) 34271da177e4SLinus Torvalds return -EFAULT; 34281da177e4SLinus Torvalds return 0; 34291da177e4SLinus Torvalds } 34306e9250f5SEric Dumazet case TCP_CC_INFO: { 34316e9250f5SEric Dumazet const struct tcp_congestion_ops *ca_ops; 34326e9250f5SEric Dumazet union tcp_cc_info info; 34336e9250f5SEric Dumazet size_t sz = 0; 34346e9250f5SEric Dumazet int attr; 34356e9250f5SEric Dumazet 34366e9250f5SEric Dumazet if (get_user(len, optlen)) 34376e9250f5SEric Dumazet return -EFAULT; 34386e9250f5SEric Dumazet 34396e9250f5SEric Dumazet ca_ops = icsk->icsk_ca_ops; 34406e9250f5SEric Dumazet if (ca_ops && ca_ops->get_info) 34416e9250f5SEric Dumazet sz = ca_ops->get_info(sk, ~0U, &attr, &info); 34426e9250f5SEric Dumazet 34436e9250f5SEric Dumazet len = min_t(unsigned int, len, sz); 34446e9250f5SEric Dumazet if (put_user(len, optlen)) 34456e9250f5SEric Dumazet return -EFAULT; 34466e9250f5SEric Dumazet if (copy_to_user(optval, &info, len)) 34476e9250f5SEric Dumazet return -EFAULT; 34486e9250f5SEric Dumazet return 0; 34496e9250f5SEric Dumazet } 34501da177e4SLinus Torvalds case TCP_QUICKACK: 345131954cd8SWei Wang val = !inet_csk_in_pingpong_mode(sk); 34521da177e4SLinus Torvalds break; 34535f8ef48dSStephen Hemminger 34545f8ef48dSStephen Hemminger case TCP_CONGESTION: 34555f8ef48dSStephen Hemminger if (get_user(len, optlen)) 34565f8ef48dSStephen Hemminger return -EFAULT; 34575f8ef48dSStephen Hemminger len = min_t(unsigned int, len, TCP_CA_NAME_MAX); 34585f8ef48dSStephen Hemminger if (put_user(len, optlen)) 34595f8ef48dSStephen Hemminger return -EFAULT; 34606687e988SArnaldo Carvalho de Melo if (copy_to_user(optval, icsk->icsk_ca_ops->name, len)) 34615f8ef48dSStephen Hemminger return -EFAULT; 34625f8ef48dSStephen Hemminger return 0; 3463e56fb50fSWilliam Allen Simpson 3464734942ccSDave Watson case TCP_ULP: 3465734942ccSDave Watson if (get_user(len, optlen)) 3466734942ccSDave Watson return -EFAULT; 3467734942ccSDave Watson len = min_t(unsigned int, len, TCP_ULP_NAME_MAX); 3468d97af30fSDave Watson if (!icsk->icsk_ulp_ops) { 3469d97af30fSDave Watson if (put_user(0, optlen)) 3470d97af30fSDave Watson return -EFAULT; 3471d97af30fSDave Watson return 0; 3472d97af30fSDave Watson } 3473734942ccSDave Watson if (put_user(len, optlen)) 3474734942ccSDave Watson return -EFAULT; 3475734942ccSDave Watson if (copy_to_user(optval, icsk->icsk_ulp_ops->name, len)) 3476734942ccSDave Watson return -EFAULT; 3477734942ccSDave Watson return 0; 3478734942ccSDave Watson 34791fba70e5SYuchung Cheng case TCP_FASTOPEN_KEY: { 34800f1ce023SJason Baron __u8 key[TCP_FASTOPEN_KEY_BUF_LENGTH]; 34811fba70e5SYuchung Cheng struct tcp_fastopen_context *ctx; 34820f1ce023SJason Baron unsigned int key_len = 0; 34831fba70e5SYuchung Cheng 34841fba70e5SYuchung Cheng if (get_user(len, optlen)) 34851fba70e5SYuchung Cheng return -EFAULT; 34861fba70e5SYuchung Cheng 34871fba70e5SYuchung Cheng rcu_read_lock(); 34881fba70e5SYuchung Cheng ctx = rcu_dereference(icsk->icsk_accept_queue.fastopenq.ctx); 34890f1ce023SJason Baron if (ctx) { 34900f1ce023SJason Baron key_len = tcp_fastopen_context_len(ctx) * 34910f1ce023SJason Baron TCP_FASTOPEN_KEY_LENGTH; 34920f1ce023SJason Baron memcpy(&key[0], &ctx->key[0], key_len); 34930f1ce023SJason Baron } 34941fba70e5SYuchung Cheng rcu_read_unlock(); 34951fba70e5SYuchung Cheng 34960f1ce023SJason Baron len = min_t(unsigned int, len, key_len); 34971fba70e5SYuchung Cheng if (put_user(len, optlen)) 34981fba70e5SYuchung Cheng return -EFAULT; 34991fba70e5SYuchung Cheng if (copy_to_user(optval, key, len)) 35001fba70e5SYuchung Cheng return -EFAULT; 35011fba70e5SYuchung Cheng return 0; 35021fba70e5SYuchung Cheng } 35033c0fef0bSJosh Hunt case TCP_THIN_LINEAR_TIMEOUTS: 35043c0fef0bSJosh Hunt val = tp->thin_lto; 35053c0fef0bSJosh Hunt break; 35064a7f6009SYuchung Cheng 35073c0fef0bSJosh Hunt case TCP_THIN_DUPACK: 35084a7f6009SYuchung Cheng val = 0; 35093c0fef0bSJosh Hunt break; 3510dca43c75SJerry Chu 3511ee995283SPavel Emelyanov case TCP_REPAIR: 3512ee995283SPavel Emelyanov val = tp->repair; 3513ee995283SPavel Emelyanov break; 3514ee995283SPavel Emelyanov 3515ee995283SPavel Emelyanov case TCP_REPAIR_QUEUE: 3516ee995283SPavel Emelyanov if (tp->repair) 3517ee995283SPavel Emelyanov val = tp->repair_queue; 3518ee995283SPavel Emelyanov else 3519ee995283SPavel Emelyanov return -EINVAL; 3520ee995283SPavel Emelyanov break; 3521ee995283SPavel Emelyanov 3522b1ed4c4fSAndrey Vagin case TCP_REPAIR_WINDOW: { 3523b1ed4c4fSAndrey Vagin struct tcp_repair_window opt; 3524b1ed4c4fSAndrey Vagin 3525b1ed4c4fSAndrey Vagin if (get_user(len, optlen)) 3526b1ed4c4fSAndrey Vagin return -EFAULT; 3527b1ed4c4fSAndrey Vagin 3528b1ed4c4fSAndrey Vagin if (len != sizeof(opt)) 3529b1ed4c4fSAndrey Vagin return -EINVAL; 3530b1ed4c4fSAndrey Vagin 3531b1ed4c4fSAndrey Vagin if (!tp->repair) 3532b1ed4c4fSAndrey Vagin return -EPERM; 3533b1ed4c4fSAndrey Vagin 3534b1ed4c4fSAndrey Vagin opt.snd_wl1 = tp->snd_wl1; 3535b1ed4c4fSAndrey Vagin opt.snd_wnd = tp->snd_wnd; 3536b1ed4c4fSAndrey Vagin opt.max_window = tp->max_window; 3537b1ed4c4fSAndrey Vagin opt.rcv_wnd = tp->rcv_wnd; 3538b1ed4c4fSAndrey Vagin opt.rcv_wup = tp->rcv_wup; 3539b1ed4c4fSAndrey Vagin 3540b1ed4c4fSAndrey Vagin if (copy_to_user(optval, &opt, len)) 3541b1ed4c4fSAndrey Vagin return -EFAULT; 3542b1ed4c4fSAndrey Vagin return 0; 3543b1ed4c4fSAndrey Vagin } 3544ee995283SPavel Emelyanov case TCP_QUEUE_SEQ: 3545ee995283SPavel Emelyanov if (tp->repair_queue == TCP_SEND_QUEUE) 3546ee995283SPavel Emelyanov val = tp->write_seq; 3547ee995283SPavel Emelyanov else if (tp->repair_queue == TCP_RECV_QUEUE) 3548ee995283SPavel Emelyanov val = tp->rcv_nxt; 3549ee995283SPavel Emelyanov else 3550ee995283SPavel Emelyanov return -EINVAL; 3551ee995283SPavel Emelyanov break; 3552ee995283SPavel Emelyanov 3553dca43c75SJerry Chu case TCP_USER_TIMEOUT: 35549bcc66e1SJon Maxwell val = icsk->icsk_user_timeout; 3555dca43c75SJerry Chu break; 35561536e285SKenjiro Nakayama 35571536e285SKenjiro Nakayama case TCP_FASTOPEN: 35580536fcc0SEric Dumazet val = icsk->icsk_accept_queue.fastopenq.max_qlen; 35591536e285SKenjiro Nakayama break; 35601536e285SKenjiro Nakayama 356119f6d3f3SWei Wang case TCP_FASTOPEN_CONNECT: 356219f6d3f3SWei Wang val = tp->fastopen_connect; 356319f6d3f3SWei Wang break; 356419f6d3f3SWei Wang 356571c02379SChristoph Paasch case TCP_FASTOPEN_NO_COOKIE: 356671c02379SChristoph Paasch val = tp->fastopen_no_cookie; 356771c02379SChristoph Paasch break; 356871c02379SChristoph Paasch 3569*a842fe14SEric Dumazet case TCP_TX_DELAY: 3570*a842fe14SEric Dumazet val = tp->tcp_tx_delay; 3571*a842fe14SEric Dumazet break; 3572*a842fe14SEric Dumazet 357393be6ce0SAndrey Vagin case TCP_TIMESTAMP: 35749a568de4SEric Dumazet val = tcp_time_stamp_raw() + tp->tsoffset; 357593be6ce0SAndrey Vagin break; 3576c9bee3b7SEric Dumazet case TCP_NOTSENT_LOWAT: 3577c9bee3b7SEric Dumazet val = tp->notsent_lowat; 3578c9bee3b7SEric Dumazet break; 3579b75eba76SSoheil Hassas Yeganeh case TCP_INQ: 3580b75eba76SSoheil Hassas Yeganeh val = tp->recvmsg_inq; 3581b75eba76SSoheil Hassas Yeganeh break; 3582cd8ae852SEric Dumazet case TCP_SAVE_SYN: 3583cd8ae852SEric Dumazet val = tp->save_syn; 3584cd8ae852SEric Dumazet break; 3585cd8ae852SEric Dumazet case TCP_SAVED_SYN: { 3586cd8ae852SEric Dumazet if (get_user(len, optlen)) 3587cd8ae852SEric Dumazet return -EFAULT; 3588cd8ae852SEric Dumazet 3589cd8ae852SEric Dumazet lock_sock(sk); 3590cd8ae852SEric Dumazet if (tp->saved_syn) { 3591aea0929eSEric B Munson if (len < tp->saved_syn[0]) { 3592aea0929eSEric B Munson if (put_user(tp->saved_syn[0], optlen)) { 3593aea0929eSEric B Munson release_sock(sk); 3594aea0929eSEric B Munson return -EFAULT; 3595aea0929eSEric B Munson } 3596aea0929eSEric B Munson release_sock(sk); 3597aea0929eSEric B Munson return -EINVAL; 3598aea0929eSEric B Munson } 3599aea0929eSEric B Munson len = tp->saved_syn[0]; 3600cd8ae852SEric Dumazet if (put_user(len, optlen)) { 3601cd8ae852SEric Dumazet release_sock(sk); 3602cd8ae852SEric Dumazet return -EFAULT; 3603cd8ae852SEric Dumazet } 3604cd8ae852SEric Dumazet if (copy_to_user(optval, tp->saved_syn + 1, len)) { 3605cd8ae852SEric Dumazet release_sock(sk); 3606cd8ae852SEric Dumazet return -EFAULT; 3607cd8ae852SEric Dumazet } 3608cd8ae852SEric Dumazet tcp_saved_syn_free(tp); 3609cd8ae852SEric Dumazet release_sock(sk); 3610cd8ae852SEric Dumazet } else { 3611cd8ae852SEric Dumazet release_sock(sk); 3612cd8ae852SEric Dumazet len = 0; 3613cd8ae852SEric Dumazet if (put_user(len, optlen)) 3614cd8ae852SEric Dumazet return -EFAULT; 3615cd8ae852SEric Dumazet } 3616cd8ae852SEric Dumazet return 0; 3617cd8ae852SEric Dumazet } 361805255b82SEric Dumazet #ifdef CONFIG_MMU 361905255b82SEric Dumazet case TCP_ZEROCOPY_RECEIVE: { 362005255b82SEric Dumazet struct tcp_zerocopy_receive zc; 362105255b82SEric Dumazet int err; 362205255b82SEric Dumazet 362305255b82SEric Dumazet if (get_user(len, optlen)) 362405255b82SEric Dumazet return -EFAULT; 362505255b82SEric Dumazet if (len != sizeof(zc)) 362605255b82SEric Dumazet return -EINVAL; 362705255b82SEric Dumazet if (copy_from_user(&zc, optval, len)) 362805255b82SEric Dumazet return -EFAULT; 362905255b82SEric Dumazet lock_sock(sk); 363005255b82SEric Dumazet err = tcp_zerocopy_receive(sk, &zc); 363105255b82SEric Dumazet release_sock(sk); 363205255b82SEric Dumazet if (!err && copy_to_user(optval, &zc, len)) 363305255b82SEric Dumazet err = -EFAULT; 363405255b82SEric Dumazet return err; 363505255b82SEric Dumazet } 363605255b82SEric Dumazet #endif 36371da177e4SLinus Torvalds default: 36381da177e4SLinus Torvalds return -ENOPROTOOPT; 36393ff50b79SStephen Hemminger } 36401da177e4SLinus Torvalds 36411da177e4SLinus Torvalds if (put_user(len, optlen)) 36421da177e4SLinus Torvalds return -EFAULT; 36431da177e4SLinus Torvalds if (copy_to_user(optval, &val, len)) 36441da177e4SLinus Torvalds return -EFAULT; 36451da177e4SLinus Torvalds return 0; 36461da177e4SLinus Torvalds } 36471da177e4SLinus Torvalds 36483fdadf7dSDmitry Mishin int tcp_getsockopt(struct sock *sk, int level, int optname, char __user *optval, 36493fdadf7dSDmitry Mishin int __user *optlen) 36503fdadf7dSDmitry Mishin { 36513fdadf7dSDmitry Mishin struct inet_connection_sock *icsk = inet_csk(sk); 36523fdadf7dSDmitry Mishin 36533fdadf7dSDmitry Mishin if (level != SOL_TCP) 36543fdadf7dSDmitry Mishin return icsk->icsk_af_ops->getsockopt(sk, level, optname, 36553fdadf7dSDmitry Mishin optval, optlen); 36563fdadf7dSDmitry Mishin return do_tcp_getsockopt(sk, level, optname, optval, optlen); 36573fdadf7dSDmitry Mishin } 36584bc2f18bSEric Dumazet EXPORT_SYMBOL(tcp_getsockopt); 36593fdadf7dSDmitry Mishin 36603fdadf7dSDmitry Mishin #ifdef CONFIG_COMPAT 3661543d9cfeSArnaldo Carvalho de Melo int compat_tcp_getsockopt(struct sock *sk, int level, int optname, 3662543d9cfeSArnaldo Carvalho de Melo char __user *optval, int __user *optlen) 36633fdadf7dSDmitry Mishin { 3664dec73ff0SArnaldo Carvalho de Melo if (level != SOL_TCP) 3665dec73ff0SArnaldo Carvalho de Melo return inet_csk_compat_getsockopt(sk, level, optname, 3666dec73ff0SArnaldo Carvalho de Melo optval, optlen); 36673fdadf7dSDmitry Mishin return do_tcp_getsockopt(sk, level, optname, optval, optlen); 36683fdadf7dSDmitry Mishin } 3669543d9cfeSArnaldo Carvalho de Melo EXPORT_SYMBOL(compat_tcp_getsockopt); 36703fdadf7dSDmitry Mishin #endif 36711da177e4SLinus Torvalds 3672cfb6eeb4SYOSHIFUJI Hideaki #ifdef CONFIG_TCP_MD5SIG 3673349ce993SEric Dumazet static DEFINE_PER_CPU(struct tcp_md5sig_pool, tcp_md5sig_pool); 367471cea17eSEric Dumazet static DEFINE_MUTEX(tcp_md5sig_mutex); 3675349ce993SEric Dumazet static bool tcp_md5sig_pool_populated = false; 3676cfb6eeb4SYOSHIFUJI Hideaki 367771cea17eSEric Dumazet static void __tcp_alloc_md5sig_pool(void) 3678cfb6eeb4SYOSHIFUJI Hideaki { 3679cf80e0e4SHerbert Xu struct crypto_ahash *hash; 3680cfb6eeb4SYOSHIFUJI Hideaki int cpu; 3681cfb6eeb4SYOSHIFUJI Hideaki 3682cf80e0e4SHerbert Xu hash = crypto_alloc_ahash("md5", 0, CRYPTO_ALG_ASYNC); 36831eea84b7SInsu Yun if (IS_ERR(hash)) 3684349ce993SEric Dumazet return; 3685cf80e0e4SHerbert Xu 3686cf80e0e4SHerbert Xu for_each_possible_cpu(cpu) { 368719689e38SEric Dumazet void *scratch = per_cpu(tcp_md5sig_pool, cpu).scratch; 3688cf80e0e4SHerbert Xu struct ahash_request *req; 3689cf80e0e4SHerbert Xu 369019689e38SEric Dumazet if (!scratch) { 369119689e38SEric Dumazet scratch = kmalloc_node(sizeof(union tcp_md5sum_block) + 369219689e38SEric Dumazet sizeof(struct tcphdr), 369319689e38SEric Dumazet GFP_KERNEL, 369419689e38SEric Dumazet cpu_to_node(cpu)); 369519689e38SEric Dumazet if (!scratch) 369619689e38SEric Dumazet return; 369719689e38SEric Dumazet per_cpu(tcp_md5sig_pool, cpu).scratch = scratch; 369819689e38SEric Dumazet } 3699cf80e0e4SHerbert Xu if (per_cpu(tcp_md5sig_pool, cpu).md5_req) 3700cf80e0e4SHerbert Xu continue; 3701cf80e0e4SHerbert Xu 3702cf80e0e4SHerbert Xu req = ahash_request_alloc(hash, GFP_KERNEL); 3703cf80e0e4SHerbert Xu if (!req) 3704cf80e0e4SHerbert Xu return; 3705cf80e0e4SHerbert Xu 3706cf80e0e4SHerbert Xu ahash_request_set_callback(req, 0, NULL, NULL); 3707cf80e0e4SHerbert Xu 3708cf80e0e4SHerbert Xu per_cpu(tcp_md5sig_pool, cpu).md5_req = req; 3709349ce993SEric Dumazet } 3710349ce993SEric Dumazet /* before setting tcp_md5sig_pool_populated, we must commit all writes 3711349ce993SEric Dumazet * to memory. See smp_rmb() in tcp_get_md5sig_pool() 371271cea17eSEric Dumazet */ 371371cea17eSEric Dumazet smp_wmb(); 3714349ce993SEric Dumazet tcp_md5sig_pool_populated = true; 3715cfb6eeb4SYOSHIFUJI Hideaki } 3716cfb6eeb4SYOSHIFUJI Hideaki 371771cea17eSEric Dumazet bool tcp_alloc_md5sig_pool(void) 3718cfb6eeb4SYOSHIFUJI Hideaki { 3719349ce993SEric Dumazet if (unlikely(!tcp_md5sig_pool_populated)) { 372071cea17eSEric Dumazet mutex_lock(&tcp_md5sig_mutex); 3721cfb6eeb4SYOSHIFUJI Hideaki 37226015c71eSEric Dumazet if (!tcp_md5sig_pool_populated) { 372371cea17eSEric Dumazet __tcp_alloc_md5sig_pool(); 37246015c71eSEric Dumazet if (tcp_md5sig_pool_populated) 3725921f9a0fSEric Dumazet static_branch_inc(&tcp_md5_needed); 37266015c71eSEric Dumazet } 3727cfb6eeb4SYOSHIFUJI Hideaki 372871cea17eSEric Dumazet mutex_unlock(&tcp_md5sig_mutex); 3729cfb6eeb4SYOSHIFUJI Hideaki } 3730349ce993SEric Dumazet return tcp_md5sig_pool_populated; 3731cfb6eeb4SYOSHIFUJI Hideaki } 3732cfb6eeb4SYOSHIFUJI Hideaki EXPORT_SYMBOL(tcp_alloc_md5sig_pool); 3733cfb6eeb4SYOSHIFUJI Hideaki 373435790c04SEric Dumazet 373535790c04SEric Dumazet /** 373635790c04SEric Dumazet * tcp_get_md5sig_pool - get md5sig_pool for this user 373735790c04SEric Dumazet * 373835790c04SEric Dumazet * We use percpu structure, so if we succeed, we exit with preemption 373935790c04SEric Dumazet * and BH disabled, to make sure another thread or softirq handling 374035790c04SEric Dumazet * wont try to get same context. 374135790c04SEric Dumazet */ 374235790c04SEric Dumazet struct tcp_md5sig_pool *tcp_get_md5sig_pool(void) 3743cfb6eeb4SYOSHIFUJI Hideaki { 374435790c04SEric Dumazet local_bh_disable(); 374535790c04SEric Dumazet 3746349ce993SEric Dumazet if (tcp_md5sig_pool_populated) { 3747349ce993SEric Dumazet /* coupled with smp_wmb() in __tcp_alloc_md5sig_pool() */ 3748349ce993SEric Dumazet smp_rmb(); 3749349ce993SEric Dumazet return this_cpu_ptr(&tcp_md5sig_pool); 3750349ce993SEric Dumazet } 375135790c04SEric Dumazet local_bh_enable(); 375235790c04SEric Dumazet return NULL; 3753cfb6eeb4SYOSHIFUJI Hideaki } 375435790c04SEric Dumazet EXPORT_SYMBOL(tcp_get_md5sig_pool); 3755cfb6eeb4SYOSHIFUJI Hideaki 375649a72dfbSAdam Langley int tcp_md5_hash_skb_data(struct tcp_md5sig_pool *hp, 3757cf533ea5SEric Dumazet const struct sk_buff *skb, unsigned int header_len) 375849a72dfbSAdam Langley { 375949a72dfbSAdam Langley struct scatterlist sg; 376049a72dfbSAdam Langley const struct tcphdr *tp = tcp_hdr(skb); 3761cf80e0e4SHerbert Xu struct ahash_request *req = hp->md5_req; 376295c96174SEric Dumazet unsigned int i; 376395c96174SEric Dumazet const unsigned int head_data_len = skb_headlen(skb) > header_len ? 376449a72dfbSAdam Langley skb_headlen(skb) - header_len : 0; 376549a72dfbSAdam Langley const struct skb_shared_info *shi = skb_shinfo(skb); 3766d7fd1b57SEric Dumazet struct sk_buff *frag_iter; 376749a72dfbSAdam Langley 376849a72dfbSAdam Langley sg_init_table(&sg, 1); 376949a72dfbSAdam Langley 377049a72dfbSAdam Langley sg_set_buf(&sg, ((u8 *) tp) + header_len, head_data_len); 3771cf80e0e4SHerbert Xu ahash_request_set_crypt(req, &sg, NULL, head_data_len); 3772cf80e0e4SHerbert Xu if (crypto_ahash_update(req)) 377349a72dfbSAdam Langley return 1; 377449a72dfbSAdam Langley 377549a72dfbSAdam Langley for (i = 0; i < shi->nr_frags; ++i) { 377649a72dfbSAdam Langley const struct skb_frag_struct *f = &shi->frags[i]; 377754d27fcbSEric Dumazet unsigned int offset = f->page_offset; 377854d27fcbSEric Dumazet struct page *page = skb_frag_page(f) + (offset >> PAGE_SHIFT); 377954d27fcbSEric Dumazet 378054d27fcbSEric Dumazet sg_set_page(&sg, page, skb_frag_size(f), 378154d27fcbSEric Dumazet offset_in_page(offset)); 3782cf80e0e4SHerbert Xu ahash_request_set_crypt(req, &sg, NULL, skb_frag_size(f)); 3783cf80e0e4SHerbert Xu if (crypto_ahash_update(req)) 378449a72dfbSAdam Langley return 1; 378549a72dfbSAdam Langley } 378649a72dfbSAdam Langley 3787d7fd1b57SEric Dumazet skb_walk_frags(skb, frag_iter) 3788d7fd1b57SEric Dumazet if (tcp_md5_hash_skb_data(hp, frag_iter, 0)) 3789d7fd1b57SEric Dumazet return 1; 3790d7fd1b57SEric Dumazet 379149a72dfbSAdam Langley return 0; 379249a72dfbSAdam Langley } 379349a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_skb_data); 379449a72dfbSAdam Langley 3795cf533ea5SEric Dumazet int tcp_md5_hash_key(struct tcp_md5sig_pool *hp, const struct tcp_md5sig_key *key) 379649a72dfbSAdam Langley { 379749a72dfbSAdam Langley struct scatterlist sg; 379849a72dfbSAdam Langley 379949a72dfbSAdam Langley sg_init_one(&sg, key->key, key->keylen); 3800cf80e0e4SHerbert Xu ahash_request_set_crypt(hp->md5_req, &sg, NULL, key->keylen); 3801cf80e0e4SHerbert Xu return crypto_ahash_update(hp->md5_req); 380249a72dfbSAdam Langley } 380349a72dfbSAdam Langley EXPORT_SYMBOL(tcp_md5_hash_key); 380449a72dfbSAdam Langley 3805cfb6eeb4SYOSHIFUJI Hideaki #endif 3806cfb6eeb4SYOSHIFUJI Hideaki 38074ac02babSAndi Kleen void tcp_done(struct sock *sk) 38084ac02babSAndi Kleen { 38098336886fSJerry Chu struct request_sock *req = tcp_sk(sk)->fastopen_rsk; 38108336886fSJerry Chu 38114ac02babSAndi Kleen if (sk->sk_state == TCP_SYN_SENT || sk->sk_state == TCP_SYN_RECV) 3812c10d9310SEric Dumazet TCP_INC_STATS(sock_net(sk), TCP_MIB_ATTEMPTFAILS); 38134ac02babSAndi Kleen 38144ac02babSAndi Kleen tcp_set_state(sk, TCP_CLOSE); 38154ac02babSAndi Kleen tcp_clear_xmit_timers(sk); 381600db4124SIan Morris if (req) 38178336886fSJerry Chu reqsk_fastopen_remove(sk, req, false); 38184ac02babSAndi Kleen 38194ac02babSAndi Kleen sk->sk_shutdown = SHUTDOWN_MASK; 38204ac02babSAndi Kleen 38214ac02babSAndi Kleen if (!sock_flag(sk, SOCK_DEAD)) 38224ac02babSAndi Kleen sk->sk_state_change(sk); 38234ac02babSAndi Kleen else 38244ac02babSAndi Kleen inet_csk_destroy_sock(sk); 38254ac02babSAndi Kleen } 38264ac02babSAndi Kleen EXPORT_SYMBOL_GPL(tcp_done); 38274ac02babSAndi Kleen 3828c1e64e29SLorenzo Colitti int tcp_abort(struct sock *sk, int err) 3829c1e64e29SLorenzo Colitti { 3830c1e64e29SLorenzo Colitti if (!sk_fullsock(sk)) { 383107f6f4a3SEric Dumazet if (sk->sk_state == TCP_NEW_SYN_RECV) { 383207f6f4a3SEric Dumazet struct request_sock *req = inet_reqsk(sk); 383307f6f4a3SEric Dumazet 383407f6f4a3SEric Dumazet local_bh_disable(); 3835acc2cf4eSLorenzo Colitti inet_csk_reqsk_queue_drop(req->rsk_listener, req); 383607f6f4a3SEric Dumazet local_bh_enable(); 383707f6f4a3SEric Dumazet return 0; 383807f6f4a3SEric Dumazet } 3839c1e64e29SLorenzo Colitti return -EOPNOTSUPP; 3840c1e64e29SLorenzo Colitti } 3841c1e64e29SLorenzo Colitti 3842c1e64e29SLorenzo Colitti /* Don't race with userspace socket closes such as tcp_close. */ 3843c1e64e29SLorenzo Colitti lock_sock(sk); 3844c1e64e29SLorenzo Colitti 38452010b93eSLorenzo Colitti if (sk->sk_state == TCP_LISTEN) { 38462010b93eSLorenzo Colitti tcp_set_state(sk, TCP_CLOSE); 38472010b93eSLorenzo Colitti inet_csk_listen_stop(sk); 38482010b93eSLorenzo Colitti } 38492010b93eSLorenzo Colitti 3850c1e64e29SLorenzo Colitti /* Don't race with BH socket closes such as inet_csk_listen_stop. */ 3851c1e64e29SLorenzo Colitti local_bh_disable(); 3852c1e64e29SLorenzo Colitti bh_lock_sock(sk); 3853c1e64e29SLorenzo Colitti 3854c1e64e29SLorenzo Colitti if (!sock_flag(sk, SOCK_DEAD)) { 3855c1e64e29SLorenzo Colitti sk->sk_err = err; 3856c1e64e29SLorenzo Colitti /* This barrier is coupled with smp_rmb() in tcp_poll() */ 3857c1e64e29SLorenzo Colitti smp_wmb(); 3858c1e64e29SLorenzo Colitti sk->sk_error_report(sk); 3859c1e64e29SLorenzo Colitti if (tcp_need_reset(sk->sk_state)) 3860c1e64e29SLorenzo Colitti tcp_send_active_reset(sk, GFP_ATOMIC); 3861c1e64e29SLorenzo Colitti tcp_done(sk); 3862c1e64e29SLorenzo Colitti } 3863c1e64e29SLorenzo Colitti 3864c1e64e29SLorenzo Colitti bh_unlock_sock(sk); 3865c1e64e29SLorenzo Colitti local_bh_enable(); 3866e05836acSSoheil Hassas Yeganeh tcp_write_queue_purge(sk); 3867c1e64e29SLorenzo Colitti release_sock(sk); 3868c1e64e29SLorenzo Colitti return 0; 3869c1e64e29SLorenzo Colitti } 3870c1e64e29SLorenzo Colitti EXPORT_SYMBOL_GPL(tcp_abort); 3871c1e64e29SLorenzo Colitti 38725f8ef48dSStephen Hemminger extern struct tcp_congestion_ops tcp_reno; 38731da177e4SLinus Torvalds 38741da177e4SLinus Torvalds static __initdata unsigned long thash_entries; 38751da177e4SLinus Torvalds static int __init set_thash_entries(char *str) 38761da177e4SLinus Torvalds { 3877413c27d8SEldad Zack ssize_t ret; 3878413c27d8SEldad Zack 38791da177e4SLinus Torvalds if (!str) 38801da177e4SLinus Torvalds return 0; 3881413c27d8SEldad Zack 3882413c27d8SEldad Zack ret = kstrtoul(str, 0, &thash_entries); 3883413c27d8SEldad Zack if (ret) 3884413c27d8SEldad Zack return 0; 3885413c27d8SEldad Zack 38861da177e4SLinus Torvalds return 1; 38871da177e4SLinus Torvalds } 38881da177e4SLinus Torvalds __setup("thash_entries=", set_thash_entries); 38891da177e4SLinus Torvalds 389047d7a88cSFabian Frederick static void __init tcp_init_mem(void) 38914acb4190SGlauber Costa { 3892b66e91ccSEric Dumazet unsigned long limit = nr_free_buffer_pages() / 16; 3893b66e91ccSEric Dumazet 38944acb4190SGlauber Costa limit = max(limit, 128UL); 3895b66e91ccSEric Dumazet sysctl_tcp_mem[0] = limit / 4 * 3; /* 4.68 % */ 3896b66e91ccSEric Dumazet sysctl_tcp_mem[1] = limit; /* 6.25 % */ 3897b66e91ccSEric Dumazet sysctl_tcp_mem[2] = sysctl_tcp_mem[0] * 2; /* 9.37 % */ 38984acb4190SGlauber Costa } 38994acb4190SGlauber Costa 39001da177e4SLinus Torvalds void __init tcp_init(void) 39011da177e4SLinus Torvalds { 3902b49960a0SEric Dumazet int max_rshare, max_wshare, cnt; 3903b2d3ea4aSEric Dumazet unsigned long limit; 3904074b8517SDimitri Sivanich unsigned int i; 39051da177e4SLinus Torvalds 3906b2d3ea4aSEric Dumazet BUILD_BUG_ON(sizeof(struct tcp_skb_cb) > 3907b2d3ea4aSEric Dumazet FIELD_SIZEOF(struct sk_buff, cb)); 39081da177e4SLinus Torvalds 3909908c7f19STejun Heo percpu_counter_init(&tcp_sockets_allocated, 0, GFP_KERNEL); 3910908c7f19STejun Heo percpu_counter_init(&tcp_orphan_count, 0, GFP_KERNEL); 39111946e672SHaishuang Yan inet_hashinfo_init(&tcp_hashinfo); 391227da6d37SMartin KaFai Lau inet_hashinfo2_init(&tcp_hashinfo, "tcp_listen_portaddr_hash", 391327da6d37SMartin KaFai Lau thash_entries, 21, /* one slot per 2 MB*/ 391427da6d37SMartin KaFai Lau 0, 64 * 1024); 39156e04e021SArnaldo Carvalho de Melo tcp_hashinfo.bind_bucket_cachep = 39166e04e021SArnaldo Carvalho de Melo kmem_cache_create("tcp_bind_bucket", 39176e04e021SArnaldo Carvalho de Melo sizeof(struct inet_bind_bucket), 0, 391820c2df83SPaul Mundt SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL); 39191da177e4SLinus Torvalds 39201da177e4SLinus Torvalds /* Size and allocate the main established and bind bucket 39211da177e4SLinus Torvalds * hash tables. 39221da177e4SLinus Torvalds * 39231da177e4SLinus Torvalds * The methodology is similar to that of the buffer cache. 39241da177e4SLinus Torvalds */ 39256e04e021SArnaldo Carvalho de Melo tcp_hashinfo.ehash = 39261da177e4SLinus Torvalds alloc_large_system_hash("TCP established", 39270f7ff927SArnaldo Carvalho de Melo sizeof(struct inet_ehash_bucket), 39281da177e4SLinus Torvalds thash_entries, 3929fd90b29dSEric Dumazet 17, /* one slot per 128 KB of memory */ 39309e950efaSJohn Heffner 0, 39311da177e4SLinus Torvalds NULL, 3932f373b53bSEric Dumazet &tcp_hashinfo.ehash_mask, 393331fe62b9STim Bird 0, 39340ccfe618SJean Delvare thash_entries ? 0 : 512 * 1024); 393505dbc7b5SEric Dumazet for (i = 0; i <= tcp_hashinfo.ehash_mask; i++) 39363ab5aee7SEric Dumazet INIT_HLIST_NULLS_HEAD(&tcp_hashinfo.ehash[i].chain, i); 393705dbc7b5SEric Dumazet 3938230140cfSEric Dumazet if (inet_ehash_locks_alloc(&tcp_hashinfo)) 3939230140cfSEric Dumazet panic("TCP: failed to alloc ehash_locks"); 39406e04e021SArnaldo Carvalho de Melo tcp_hashinfo.bhash = 39411da177e4SLinus Torvalds alloc_large_system_hash("TCP bind", 39420f7ff927SArnaldo Carvalho de Melo sizeof(struct inet_bind_hashbucket), 3943f373b53bSEric Dumazet tcp_hashinfo.ehash_mask + 1, 3944fd90b29dSEric Dumazet 17, /* one slot per 128 KB of memory */ 39459e950efaSJohn Heffner 0, 39466e04e021SArnaldo Carvalho de Melo &tcp_hashinfo.bhash_size, 39471da177e4SLinus Torvalds NULL, 394831fe62b9STim Bird 0, 39491da177e4SLinus Torvalds 64 * 1024); 3950074b8517SDimitri Sivanich tcp_hashinfo.bhash_size = 1U << tcp_hashinfo.bhash_size; 39516e04e021SArnaldo Carvalho de Melo for (i = 0; i < tcp_hashinfo.bhash_size; i++) { 39526e04e021SArnaldo Carvalho de Melo spin_lock_init(&tcp_hashinfo.bhash[i].lock); 39536e04e021SArnaldo Carvalho de Melo INIT_HLIST_HEAD(&tcp_hashinfo.bhash[i].chain); 39541da177e4SLinus Torvalds } 39551da177e4SLinus Torvalds 3956c5ed63d6SEric Dumazet 3957c5ed63d6SEric Dumazet cnt = tcp_hashinfo.ehash_mask + 1; 3958c5ed63d6SEric Dumazet sysctl_tcp_max_orphans = cnt / 2; 39591da177e4SLinus Torvalds 3960a4fe34bfSEric W. Biederman tcp_init_mem(); 3961c43b874dSJason Wang /* Set per-socket limits to no more than 1/128 the pressure threshold */ 39625fb84b14SEric Dumazet limit = nr_free_buffer_pages() << (PAGE_SHIFT - 7); 3963b49960a0SEric Dumazet max_wshare = min(4UL*1024*1024, limit); 3964b49960a0SEric Dumazet max_rshare = min(6UL*1024*1024, limit); 39657b4f4b5eSJohn Heffner 3966356d1833SEric Dumazet init_net.ipv4.sysctl_tcp_wmem[0] = SK_MEM_QUANTUM; 3967356d1833SEric Dumazet init_net.ipv4.sysctl_tcp_wmem[1] = 16*1024; 3968356d1833SEric Dumazet init_net.ipv4.sysctl_tcp_wmem[2] = max(64*1024, max_wshare); 39697b4f4b5eSJohn Heffner 3970356d1833SEric Dumazet init_net.ipv4.sysctl_tcp_rmem[0] = SK_MEM_QUANTUM; 3971a337531bSYuchung Cheng init_net.ipv4.sysctl_tcp_rmem[1] = 131072; 3972a337531bSYuchung Cheng init_net.ipv4.sysctl_tcp_rmem[2] = max(131072, max_rshare); 39731da177e4SLinus Torvalds 3974afd46503SJoe Perches pr_info("Hash tables configured (established %u bind %u)\n", 3975f373b53bSEric Dumazet tcp_hashinfo.ehash_mask + 1, tcp_hashinfo.bhash_size); 3976317a76f9SStephen Hemminger 39771946e672SHaishuang Yan tcp_v4_init(); 397851c5d0c4SDavid S. Miller tcp_metrics_init(); 397955d8694fSFlorian Westphal BUG_ON(tcp_register_congestion_control(&tcp_reno) != 0); 398046d3ceabSEric Dumazet tcp_tasklet_init(); 39811da177e4SLinus Torvalds } 3982