]> bbs.cooldavid.org Git - net-next-2.6.git/blame - net/ipv4/tcp.c
[NETNS][IPV6]: inet6_addr - make ipv6_chk_home_addr namespace aware
[net-next-2.6.git] / net / ipv4 / tcp.c
CommitLineData
1da177e4
LT
1/*
2 * INET An implementation of the TCP/IP protocol suite for the LINUX
3 * operating system. INET is implemented using the BSD Socket
4 * interface as the means of communication with the user level.
5 *
6 * Implementation of the Transmission Control Protocol(TCP).
7 *
8 * Version: $Id: tcp.c,v 1.216 2002/02/01 22:01:04 davem Exp $
9 *
02c30a84 10 * Authors: Ross Biro
1da177e4
LT
11 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12 * Mark Evans, <evansmp@uhura.aston.ac.uk>
13 * Corey Minyard <wf-rch!minyard@relay.EU.net>
14 * Florian La Roche, <flla@stud.uni-sb.de>
15 * Charles Hedrick, <hedrick@klinzhai.rutgers.edu>
16 * Linus Torvalds, <torvalds@cs.helsinki.fi>
17 * Alan Cox, <gw4pts@gw4pts.ampr.org>
18 * Matthew Dillon, <dillon@apollo.west.oic.com>
19 * Arnt Gulbrandsen, <agulbra@nvg.unit.no>
20 * Jorge Cwik, <jorge@laser.satlink.net>
21 *
22 * Fixes:
23 * Alan Cox : Numerous verify_area() calls
24 * Alan Cox : Set the ACK bit on a reset
25 * Alan Cox : Stopped it crashing if it closed while
26 * sk->inuse=1 and was trying to connect
27 * (tcp_err()).
28 * Alan Cox : All icmp error handling was broken
29 * pointers passed where wrong and the
30 * socket was looked up backwards. Nobody
31 * tested any icmp error code obviously.
32 * Alan Cox : tcp_err() now handled properly. It
33 * wakes people on errors. poll
34 * behaves and the icmp error race
35 * has gone by moving it into sock.c
36 * Alan Cox : tcp_send_reset() fixed to work for
37 * everything not just packets for
38 * unknown sockets.
39 * Alan Cox : tcp option processing.
40 * Alan Cox : Reset tweaked (still not 100%) [Had
41 * syn rule wrong]
42 * Herp Rosmanith : More reset fixes
43 * Alan Cox : No longer acks invalid rst frames.
44 * Acking any kind of RST is right out.
45 * Alan Cox : Sets an ignore me flag on an rst
46 * receive otherwise odd bits of prattle
47 * escape still
48 * Alan Cox : Fixed another acking RST frame bug.
49 * Should stop LAN workplace lockups.
50 * Alan Cox : Some tidyups using the new skb list
51 * facilities
52 * Alan Cox : sk->keepopen now seems to work
53 * Alan Cox : Pulls options out correctly on accepts
54 * Alan Cox : Fixed assorted sk->rqueue->next errors
55 * Alan Cox : PSH doesn't end a TCP read. Switched a
56 * bit to skb ops.
57 * Alan Cox : Tidied tcp_data to avoid a potential
58 * nasty.
59 * Alan Cox : Added some better commenting, as the
60 * tcp is hard to follow
61 * Alan Cox : Removed incorrect check for 20 * psh
62 * Michael O'Reilly : ack < copied bug fix.
63 * Johannes Stille : Misc tcp fixes (not all in yet).
64 * Alan Cox : FIN with no memory -> CRASH
65 * Alan Cox : Added socket option proto entries.
66 * Also added awareness of them to accept.
67 * Alan Cox : Added TCP options (SOL_TCP)
68 * Alan Cox : Switched wakeup calls to callbacks,
69 * so the kernel can layer network
70 * sockets.
71 * Alan Cox : Use ip_tos/ip_ttl settings.
72 * Alan Cox : Handle FIN (more) properly (we hope).
73 * Alan Cox : RST frames sent on unsynchronised
74 * state ack error.
75 * Alan Cox : Put in missing check for SYN bit.
76 * Alan Cox : Added tcp_select_window() aka NET2E
77 * window non shrink trick.
78 * Alan Cox : Added a couple of small NET2E timer
79 * fixes
80 * Charles Hedrick : TCP fixes
81 * Toomas Tamm : TCP window fixes
82 * Alan Cox : Small URG fix to rlogin ^C ack fight
83 * Charles Hedrick : Rewrote most of it to actually work
84 * Linus : Rewrote tcp_read() and URG handling
85 * completely
86 * Gerhard Koerting: Fixed some missing timer handling
87 * Matthew Dillon : Reworked TCP machine states as per RFC
88 * Gerhard Koerting: PC/TCP workarounds
89 * Adam Caldwell : Assorted timer/timing errors
90 * Matthew Dillon : Fixed another RST bug
91 * Alan Cox : Move to kernel side addressing changes.
92 * Alan Cox : Beginning work on TCP fastpathing
93 * (not yet usable)
94 * Arnt Gulbrandsen: Turbocharged tcp_check() routine.
95 * Alan Cox : TCP fast path debugging
96 * Alan Cox : Window clamping
97 * Michael Riepe : Bug in tcp_check()
98 * Matt Dillon : More TCP improvements and RST bug fixes
99 * Matt Dillon : Yet more small nasties remove from the
100 * TCP code (Be very nice to this man if
101 * tcp finally works 100%) 8)
102 * Alan Cox : BSD accept semantics.
103 * Alan Cox : Reset on closedown bug.
104 * Peter De Schrijver : ENOTCONN check missing in tcp_sendto().
105 * Michael Pall : Handle poll() after URG properly in
106 * all cases.
107 * Michael Pall : Undo the last fix in tcp_read_urg()
108 * (multi URG PUSH broke rlogin).
109 * Michael Pall : Fix the multi URG PUSH problem in
110 * tcp_readable(), poll() after URG
111 * works now.
112 * Michael Pall : recv(...,MSG_OOB) never blocks in the
113 * BSD api.
114 * Alan Cox : Changed the semantics of sk->socket to
115 * fix a race and a signal problem with
116 * accept() and async I/O.
117 * Alan Cox : Relaxed the rules on tcp_sendto().
118 * Yury Shevchuk : Really fixed accept() blocking problem.
119 * Craig I. Hagan : Allow for BSD compatible TIME_WAIT for
120 * clients/servers which listen in on
121 * fixed ports.
122 * Alan Cox : Cleaned the above up and shrank it to
123 * a sensible code size.
124 * Alan Cox : Self connect lockup fix.
125 * Alan Cox : No connect to multicast.
126 * Ross Biro : Close unaccepted children on master
127 * socket close.
128 * Alan Cox : Reset tracing code.
129 * Alan Cox : Spurious resets on shutdown.
130 * Alan Cox : Giant 15 minute/60 second timer error
131 * Alan Cox : Small whoops in polling before an
132 * accept.
133 * Alan Cox : Kept the state trace facility since
134 * it's handy for debugging.
135 * Alan Cox : More reset handler fixes.
136 * Alan Cox : Started rewriting the code based on
137 * the RFC's for other useful protocol
138 * references see: Comer, KA9Q NOS, and
139 * for a reference on the difference
140 * between specifications and how BSD
141 * works see the 4.4lite source.
142 * A.N.Kuznetsov : Don't time wait on completion of tidy
143 * close.
144 * Linus Torvalds : Fin/Shutdown & copied_seq changes.
145 * Linus Torvalds : Fixed BSD port reuse to work first syn
146 * Alan Cox : Reimplemented timers as per the RFC
147 * and using multiple timers for sanity.
148 * Alan Cox : Small bug fixes, and a lot of new
149 * comments.
150 * Alan Cox : Fixed dual reader crash by locking
151 * the buffers (much like datagram.c)
152 * Alan Cox : Fixed stuck sockets in probe. A probe
153 * now gets fed up of retrying without
154 * (even a no space) answer.
155 * Alan Cox : Extracted closing code better
156 * Alan Cox : Fixed the closing state machine to
157 * resemble the RFC.
158 * Alan Cox : More 'per spec' fixes.
159 * Jorge Cwik : Even faster checksumming.
160 * Alan Cox : tcp_data() doesn't ack illegal PSH
161 * only frames. At least one pc tcp stack
162 * generates them.
163 * Alan Cox : Cache last socket.
164 * Alan Cox : Per route irtt.
165 * Matt Day : poll()->select() match BSD precisely on error
166 * Alan Cox : New buffers
167 * Marc Tamsky : Various sk->prot->retransmits and
168 * sk->retransmits misupdating fixed.
169 * Fixed tcp_write_timeout: stuck close,
170 * and TCP syn retries gets used now.
171 * Mark Yarvis : In tcp_read_wakeup(), don't send an
172 * ack if state is TCP_CLOSED.
173 * Alan Cox : Look up device on a retransmit - routes may
174 * change. Doesn't yet cope with MSS shrink right
175 * but it's a start!
176 * Marc Tamsky : Closing in closing fixes.
177 * Mike Shaver : RFC1122 verifications.
178 * Alan Cox : rcv_saddr errors.
179 * Alan Cox : Block double connect().
180 * Alan Cox : Small hooks for enSKIP.
181 * Alexey Kuznetsov: Path MTU discovery.
182 * Alan Cox : Support soft errors.
183 * Alan Cox : Fix MTU discovery pathological case
184 * when the remote claims no mtu!
185 * Marc Tamsky : TCP_CLOSE fix.
186 * Colin (G3TNE) : Send a reset on syn ack replies in
187 * window but wrong (fixes NT lpd problems)
188 * Pedro Roque : Better TCP window handling, delayed ack.
189 * Joerg Reuter : No modification of locked buffers in
190 * tcp_do_retransmit()
191 * Eric Schenk : Changed receiver side silly window
192 * avoidance algorithm to BSD style
193 * algorithm. This doubles throughput
194 * against machines running Solaris,
195 * and seems to result in general
196 * improvement.
197 * Stefan Magdalinski : adjusted tcp_readable() to fix FIONREAD
198 * Willy Konynenberg : Transparent proxying support.
199 * Mike McLagan : Routing by source
200 * Keith Owens : Do proper merging with partial SKB's in
201 * tcp_do_sendmsg to avoid burstiness.
202 * Eric Schenk : Fix fast close down bug with
203 * shutdown() followed by close().
204 * Andi Kleen : Make poll agree with SIGIO
205 * Salvatore Sanfilippo : Support SO_LINGER with linger == 1 and
206 * lingertime == 0 (RFC 793 ABORT Call)
207 * Hirokazu Takahashi : Use copy_from_user() instead of
208 * csum_and_copy_from_user() if possible.
209 *
210 * This program is free software; you can redistribute it and/or
211 * modify it under the terms of the GNU General Public License
212 * as published by the Free Software Foundation; either version
213 * 2 of the License, or(at your option) any later version.
214 *
215 * Description of States:
216 *
217 * TCP_SYN_SENT sent a connection request, waiting for ack
218 *
219 * TCP_SYN_RECV received a connection request, sent ack,
220 * waiting for final ack in three-way handshake.
221 *
222 * TCP_ESTABLISHED connection established
223 *
224 * TCP_FIN_WAIT1 our side has shutdown, waiting to complete
225 * transmission of remaining buffered data
226 *
227 * TCP_FIN_WAIT2 all buffered data sent, waiting for remote
228 * to shutdown
229 *
230 * TCP_CLOSING both sides have shutdown but we still have
231 * data we have to finish sending
232 *
233 * TCP_TIME_WAIT timeout to catch resent junk before entering
234 * closed, can only be entered from FIN_WAIT2
235 * or CLOSING. Required because the other end
236 * may not have gotten our last ACK causing it
237 * to retransmit the data packet (which we ignore)
238 *
239 * TCP_CLOSE_WAIT remote side has shutdown and is waiting for
240 * us to finish writing our data and to shutdown
241 * (we have to close() to move on to LAST_ACK)
242 *
243 * TCP_LAST_ACK out side has shutdown after remote has
244 * shutdown. There may still be data in our
245 * buffer that we have to finish sending
246 *
247 * TCP_CLOSE socket is finished
248 */
249
172589cc 250#include <linux/kernel.h>
1da177e4
LT
251#include <linux/module.h>
252#include <linux/types.h>
253#include <linux/fcntl.h>
254#include <linux/poll.h>
255#include <linux/init.h>
1da177e4 256#include <linux/fs.h>
9c55e01c
JA
257#include <linux/skbuff.h>
258#include <linux/splice.h>
259#include <linux/net.h>
260#include <linux/socket.h>
1da177e4
LT
261#include <linux/random.h>
262#include <linux/bootmem.h>
b8059ead 263#include <linux/cache.h>
f4c50d99 264#include <linux/err.h>
cfb6eeb4 265#include <linux/crypto.h>
1da177e4
LT
266
267#include <net/icmp.h>
268#include <net/tcp.h>
269#include <net/xfrm.h>
270#include <net/ip.h>
1a2449a8 271#include <net/netdma.h>
9c55e01c 272#include <net/sock.h>
1da177e4
LT
273
274#include <asm/uaccess.h>
275#include <asm/ioctls.h>
276
ab32ea5d 277int sysctl_tcp_fin_timeout __read_mostly = TCP_FIN_TIMEOUT;
1da177e4 278
ba89966c 279DEFINE_SNMP_STAT(struct tcp_mib, tcp_statistics) __read_mostly;
1da177e4 280
1da177e4
LT
281atomic_t tcp_orphan_count = ATOMIC_INIT(0);
282
0a5578cf
ACM
283EXPORT_SYMBOL_GPL(tcp_orphan_count);
284
b8059ead
DM
285int sysctl_tcp_mem[3] __read_mostly;
286int sysctl_tcp_wmem[3] __read_mostly;
287int sysctl_tcp_rmem[3] __read_mostly;
1da177e4
LT
288
289EXPORT_SYMBOL(sysctl_tcp_mem);
290EXPORT_SYMBOL(sysctl_tcp_rmem);
291EXPORT_SYMBOL(sysctl_tcp_wmem);
292
293atomic_t tcp_memory_allocated; /* Current allocated memory. */
294atomic_t tcp_sockets_allocated; /* Current number of TCP sockets. */
295
296EXPORT_SYMBOL(tcp_memory_allocated);
297EXPORT_SYMBOL(tcp_sockets_allocated);
298
9c55e01c
JA
299/*
300 * TCP splice context
301 */
302struct tcp_splice_state {
303 struct pipe_inode_info *pipe;
304 size_t len;
305 unsigned int flags;
306};
307
1da177e4
LT
308/*
309 * Pressure flag: try to collapse.
310 * Technical note: it is used by multiple contexts non atomically.
3ab224be 311 * All the __sk_mem_schedule() is of this nature: accounting
1da177e4
LT
312 * is strict, actions are advisory and have some latency.
313 */
4103f8cd 314int tcp_memory_pressure __read_mostly;
1da177e4
LT
315
316EXPORT_SYMBOL(tcp_memory_pressure);
317
318void tcp_enter_memory_pressure(void)
319{
320 if (!tcp_memory_pressure) {
321 NET_INC_STATS(LINUX_MIB_TCPMEMORYPRESSURES);
322 tcp_memory_pressure = 1;
323 }
324}
325
326EXPORT_SYMBOL(tcp_enter_memory_pressure);
327
1da177e4
LT
328/*
329 * Wait for a TCP event.
330 *
331 * Note that we don't need to lock the socket, as the upper poll layers
332 * take care of normal races (between the test and the event) and we don't
333 * go look at any of the socket buffers directly.
334 */
335unsigned int tcp_poll(struct file *file, struct socket *sock, poll_table *wait)
336{
337 unsigned int mask;
338 struct sock *sk = sock->sk;
339 struct tcp_sock *tp = tcp_sk(sk);
340
341 poll_wait(file, sk->sk_sleep, wait);
342 if (sk->sk_state == TCP_LISTEN)
dc40c7bc 343 return inet_csk_listen_poll(sk);
1da177e4
LT
344
345 /* Socket is not locked. We are protected from async events
346 by poll logic and correct handling of state changes
347 made by another threads is impossible in any case.
348 */
349
350 mask = 0;
351 if (sk->sk_err)
352 mask = POLLERR;
353
354 /*
355 * POLLHUP is certainly not done right. But poll() doesn't
356 * have a notion of HUP in just one direction, and for a
357 * socket the read side is more interesting.
358 *
359 * Some poll() documentation says that POLLHUP is incompatible
360 * with the POLLOUT/POLLWR flags, so somebody should check this
361 * all. But careful, it tends to be safer to return too many
362 * bits than too few, and you can easily break real applications
363 * if you don't tell them that something has hung up!
364 *
365 * Check-me.
366 *
367 * Check number 1. POLLHUP is _UNMASKABLE_ event (see UNIX98 and
368 * our fs/select.c). It means that after we received EOF,
369 * poll always returns immediately, making impossible poll() on write()
370 * in state CLOSE_WAIT. One solution is evident --- to set POLLHUP
371 * if and only if shutdown has been made in both directions.
372 * Actually, it is interesting to look how Solaris and DUX
373 * solve this dilemma. I would prefer, if PULLHUP were maskable,
374 * then we could set it on SND_SHUTDOWN. BTW examples given
375 * in Stevens' books assume exactly this behaviour, it explains
376 * why PULLHUP is incompatible with POLLOUT. --ANK
377 *
378 * NOTE. Check for TCP_CLOSE is added. The goal is to prevent
379 * blocking on fresh not-connected or disconnected socket. --ANK
380 */
381 if (sk->sk_shutdown == SHUTDOWN_MASK || sk->sk_state == TCP_CLOSE)
382 mask |= POLLHUP;
383 if (sk->sk_shutdown & RCV_SHUTDOWN)
f348d70a 384 mask |= POLLIN | POLLRDNORM | POLLRDHUP;
1da177e4
LT
385
386 /* Connected? */
387 if ((1 << sk->sk_state) & ~(TCPF_SYN_SENT | TCPF_SYN_RECV)) {
388 /* Potential race condition. If read of tp below will
389 * escape above sk->sk_state, we can be illegally awaken
390 * in SYN_* states. */
391 if ((tp->rcv_nxt != tp->copied_seq) &&
392 (tp->urg_seq != tp->copied_seq ||
393 tp->rcv_nxt != tp->copied_seq + 1 ||
394 sock_flag(sk, SOCK_URGINLINE) || !tp->urg_data))
395 mask |= POLLIN | POLLRDNORM;
396
397 if (!(sk->sk_shutdown & SEND_SHUTDOWN)) {
398 if (sk_stream_wspace(sk) >= sk_stream_min_wspace(sk)) {
399 mask |= POLLOUT | POLLWRNORM;
400 } else { /* send SIGIO later */
401 set_bit(SOCK_ASYNC_NOSPACE,
402 &sk->sk_socket->flags);
403 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
404
405 /* Race breaker. If space is freed after
406 * wspace test but before the flags are set,
407 * IO signal will be lost.
408 */
409 if (sk_stream_wspace(sk) >= sk_stream_min_wspace(sk))
410 mask |= POLLOUT | POLLWRNORM;
411 }
412 }
413
414 if (tp->urg_data & TCP_URG_VALID)
415 mask |= POLLPRI;
416 }
417 return mask;
418}
419
420int tcp_ioctl(struct sock *sk, int cmd, unsigned long arg)
421{
422 struct tcp_sock *tp = tcp_sk(sk);
423 int answ;
424
425 switch (cmd) {
426 case SIOCINQ:
427 if (sk->sk_state == TCP_LISTEN)
428 return -EINVAL;
429
430 lock_sock(sk);
431 if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
432 answ = 0;
433 else if (sock_flag(sk, SOCK_URGINLINE) ||
434 !tp->urg_data ||
435 before(tp->urg_seq, tp->copied_seq) ||
436 !before(tp->urg_seq, tp->rcv_nxt)) {
437 answ = tp->rcv_nxt - tp->copied_seq;
438
439 /* Subtract 1, if FIN is in queue. */
440 if (answ && !skb_queue_empty(&sk->sk_receive_queue))
441 answ -=
aa8223c7 442 tcp_hdr((struct sk_buff *)sk->sk_receive_queue.prev)->fin;
1da177e4
LT
443 } else
444 answ = tp->urg_seq - tp->copied_seq;
445 release_sock(sk);
446 break;
447 case SIOCATMARK:
448 answ = tp->urg_data && tp->urg_seq == tp->copied_seq;
449 break;
450 case SIOCOUTQ:
451 if (sk->sk_state == TCP_LISTEN)
452 return -EINVAL;
453
454 if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV))
455 answ = 0;
456 else
457 answ = tp->write_seq - tp->snd_una;
458 break;
459 default:
460 return -ENOIOCTLCMD;
3ff50b79 461 }
1da177e4
LT
462
463 return put_user(answ, (int __user *)arg);
464}
465
1da177e4
LT
466static inline void tcp_mark_push(struct tcp_sock *tp, struct sk_buff *skb)
467{
468 TCP_SKB_CB(skb)->flags |= TCPCB_FLAG_PSH;
469 tp->pushed_seq = tp->write_seq;
470}
471
472static inline int forced_push(struct tcp_sock *tp)
473{
474 return after(tp->write_seq, tp->pushed_seq + (tp->max_window >> 1));
475}
476
9e412ba7 477static inline void skb_entail(struct sock *sk, struct sk_buff *skb)
1da177e4 478{
9e412ba7 479 struct tcp_sock *tp = tcp_sk(sk);
352d4800
ACM
480 struct tcp_skb_cb *tcb = TCP_SKB_CB(skb);
481
482 skb->csum = 0;
483 tcb->seq = tcb->end_seq = tp->write_seq;
484 tcb->flags = TCPCB_FLAG_ACK;
485 tcb->sacked = 0;
1da177e4 486 skb_header_release(skb);
fe067e8a 487 tcp_add_write_queue_tail(sk, skb);
3ab224be
HA
488 sk->sk_wmem_queued += skb->truesize;
489 sk_mem_charge(sk, skb->truesize);
89ebd197 490 if (tp->nonagle & TCP_NAGLE_PUSH)
e905a9ed 491 tp->nonagle &= ~TCP_NAGLE_PUSH;
1da177e4
LT
492}
493
494static inline void tcp_mark_urg(struct tcp_sock *tp, int flags,
495 struct sk_buff *skb)
496{
497 if (flags & MSG_OOB) {
498 tp->urg_mode = 1;
499 tp->snd_up = tp->write_seq;
1da177e4
LT
500 }
501}
502
9e412ba7
IJ
503static inline void tcp_push(struct sock *sk, int flags, int mss_now,
504 int nonagle)
1da177e4 505{
9e412ba7
IJ
506 struct tcp_sock *tp = tcp_sk(sk);
507
fe067e8a
DM
508 if (tcp_send_head(sk)) {
509 struct sk_buff *skb = tcp_write_queue_tail(sk);
1da177e4
LT
510 if (!(flags & MSG_MORE) || forced_push(tp))
511 tcp_mark_push(tp, skb);
512 tcp_mark_urg(tp, flags, skb);
9e412ba7 513 __tcp_push_pending_frames(sk, mss_now,
1da177e4
LT
514 (flags & MSG_MORE) ? TCP_NAGLE_CORK : nonagle);
515 }
516}
517
6ff7751d
AB
518static int tcp_splice_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb,
519 unsigned int offset, size_t len)
9c55e01c
JA
520{
521 struct tcp_splice_state *tss = rd_desc->arg.data;
522
523 return skb_splice_bits(skb, offset, tss->pipe, tss->len, tss->flags);
524}
525
526static int __tcp_splice_read(struct sock *sk, struct tcp_splice_state *tss)
527{
528 /* Store TCP splice context information in read_descriptor_t. */
529 read_descriptor_t rd_desc = {
530 .arg.data = tss,
531 };
532
533 return tcp_read_sock(sk, &rd_desc, tcp_splice_data_recv);
534}
535
536/**
537 * tcp_splice_read - splice data from TCP socket to a pipe
538 * @sock: socket to splice from
539 * @ppos: position (not valid)
540 * @pipe: pipe to splice to
541 * @len: number of bytes to splice
542 * @flags: splice modifier flags
543 *
544 * Description:
545 * Will read pages from given socket and fill them into a pipe.
546 *
547 **/
548ssize_t tcp_splice_read(struct socket *sock, loff_t *ppos,
549 struct pipe_inode_info *pipe, size_t len,
550 unsigned int flags)
551{
552 struct sock *sk = sock->sk;
553 struct tcp_splice_state tss = {
554 .pipe = pipe,
555 .len = len,
556 .flags = flags,
557 };
558 long timeo;
559 ssize_t spliced;
560 int ret;
561
562 /*
563 * We can't seek on a socket input
564 */
565 if (unlikely(*ppos))
566 return -ESPIPE;
567
568 ret = spliced = 0;
569
570 lock_sock(sk);
571
572 timeo = sock_rcvtimeo(sk, flags & SPLICE_F_NONBLOCK);
573 while (tss.len) {
574 ret = __tcp_splice_read(sk, &tss);
575 if (ret < 0)
576 break;
577 else if (!ret) {
578 if (spliced)
579 break;
580 if (flags & SPLICE_F_NONBLOCK) {
581 ret = -EAGAIN;
582 break;
583 }
584 if (sock_flag(sk, SOCK_DONE))
585 break;
586 if (sk->sk_err) {
587 ret = sock_error(sk);
588 break;
589 }
590 if (sk->sk_shutdown & RCV_SHUTDOWN)
591 break;
592 if (sk->sk_state == TCP_CLOSE) {
593 /*
594 * This occurs when user tries to read
595 * from never connected socket.
596 */
597 if (!sock_flag(sk, SOCK_DONE))
598 ret = -ENOTCONN;
599 break;
600 }
601 if (!timeo) {
602 ret = -EAGAIN;
603 break;
604 }
605 sk_wait_data(sk, &timeo);
606 if (signal_pending(current)) {
607 ret = sock_intr_errno(timeo);
608 break;
609 }
610 continue;
611 }
612 tss.len -= ret;
613 spliced += ret;
614
615 release_sock(sk);
616 lock_sock(sk);
617
618 if (sk->sk_err || sk->sk_state == TCP_CLOSE ||
619 (sk->sk_shutdown & RCV_SHUTDOWN) || !timeo ||
620 signal_pending(current))
621 break;
622 }
623
624 release_sock(sk);
625
626 if (spliced)
627 return spliced;
628
629 return ret;
630}
631
df97c708 632struct sk_buff *sk_stream_alloc_skb(struct sock *sk, int size, gfp_t gfp)
f561d0f2
PE
633{
634 struct sk_buff *skb;
635
636 /* The TCP header must be at least 32-bit aligned. */
637 size = ALIGN(size, 4);
638
639 skb = alloc_skb_fclone(size + sk->sk_prot->max_header, gfp);
640 if (skb) {
3ab224be 641 if (sk_wmem_schedule(sk, skb->truesize)) {
f561d0f2
PE
642 /*
643 * Make sure that we have exactly size bytes
644 * available to the caller, no more, no less.
645 */
646 skb_reserve(skb, skb_tailroom(skb) - size);
647 return skb;
648 }
649 __kfree_skb(skb);
650 } else {
651 sk->sk_prot->enter_memory_pressure();
652 sk_stream_moderate_sndbuf(sk);
653 }
654 return NULL;
655}
656
1da177e4
LT
657static ssize_t do_tcp_sendpages(struct sock *sk, struct page **pages, int poffset,
658 size_t psize, int flags)
659{
660 struct tcp_sock *tp = tcp_sk(sk);
c1b4a7e6 661 int mss_now, size_goal;
1da177e4
LT
662 int err;
663 ssize_t copied;
664 long timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
665
666 /* Wait for a connection to finish. */
667 if ((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT))
668 if ((err = sk_stream_wait_connect(sk, &timeo)) != 0)
669 goto out_err;
670
671 clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
672
673 mss_now = tcp_current_mss(sk, !(flags&MSG_OOB));
c1b4a7e6 674 size_goal = tp->xmit_size_goal;
1da177e4
LT
675 copied = 0;
676
677 err = -EPIPE;
678 if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
679 goto do_error;
680
681 while (psize > 0) {
fe067e8a 682 struct sk_buff *skb = tcp_write_queue_tail(sk);
1da177e4
LT
683 struct page *page = pages[poffset / PAGE_SIZE];
684 int copy, i, can_coalesce;
685 int offset = poffset % PAGE_SIZE;
686 int size = min_t(size_t, psize, PAGE_SIZE - offset);
687
fe067e8a 688 if (!tcp_send_head(sk) || (copy = size_goal - skb->len) <= 0) {
1da177e4
LT
689new_segment:
690 if (!sk_stream_memory_free(sk))
691 goto wait_for_sndbuf;
692
df97c708 693 skb = sk_stream_alloc_skb(sk, 0, sk->sk_allocation);
1da177e4
LT
694 if (!skb)
695 goto wait_for_memory;
696
9e412ba7 697 skb_entail(sk, skb);
c1b4a7e6 698 copy = size_goal;
1da177e4
LT
699 }
700
701 if (copy > size)
702 copy = size;
703
704 i = skb_shinfo(skb)->nr_frags;
705 can_coalesce = skb_can_coalesce(skb, i, page, offset);
706 if (!can_coalesce && i >= MAX_SKB_FRAGS) {
707 tcp_mark_push(tp, skb);
708 goto new_segment;
709 }
3ab224be 710 if (!sk_wmem_schedule(sk, copy))
1da177e4 711 goto wait_for_memory;
e905a9ed 712
1da177e4
LT
713 if (can_coalesce) {
714 skb_shinfo(skb)->frags[i - 1].size += copy;
715 } else {
716 get_page(page);
717 skb_fill_page_desc(skb, i, page, offset, copy);
718 }
719
720 skb->len += copy;
721 skb->data_len += copy;
722 skb->truesize += copy;
723 sk->sk_wmem_queued += copy;
3ab224be 724 sk_mem_charge(sk, copy);
84fa7933 725 skb->ip_summed = CHECKSUM_PARTIAL;
1da177e4
LT
726 tp->write_seq += copy;
727 TCP_SKB_CB(skb)->end_seq += copy;
7967168c 728 skb_shinfo(skb)->gso_segs = 0;
1da177e4
LT
729
730 if (!copied)
731 TCP_SKB_CB(skb)->flags &= ~TCPCB_FLAG_PSH;
732
733 copied += copy;
734 poffset += copy;
735 if (!(psize -= copy))
736 goto out;
737
c1b4a7e6 738 if (skb->len < mss_now || (flags & MSG_OOB))
1da177e4
LT
739 continue;
740
741 if (forced_push(tp)) {
742 tcp_mark_push(tp, skb);
9e412ba7 743 __tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);
fe067e8a 744 } else if (skb == tcp_send_head(sk))
1da177e4
LT
745 tcp_push_one(sk, mss_now);
746 continue;
747
748wait_for_sndbuf:
749 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
750wait_for_memory:
751 if (copied)
9e412ba7 752 tcp_push(sk, flags & ~MSG_MORE, mss_now, TCP_NAGLE_PUSH);
1da177e4
LT
753
754 if ((err = sk_stream_wait_memory(sk, &timeo)) != 0)
755 goto do_error;
756
757 mss_now = tcp_current_mss(sk, !(flags&MSG_OOB));
c1b4a7e6 758 size_goal = tp->xmit_size_goal;
1da177e4
LT
759 }
760
761out:
762 if (copied)
9e412ba7 763 tcp_push(sk, flags, mss_now, tp->nonagle);
1da177e4
LT
764 return copied;
765
766do_error:
767 if (copied)
768 goto out;
769out_err:
770 return sk_stream_error(sk, flags, err);
771}
772
773ssize_t tcp_sendpage(struct socket *sock, struct page *page, int offset,
774 size_t size, int flags)
775{
776 ssize_t res;
777 struct sock *sk = sock->sk;
778
1da177e4 779 if (!(sk->sk_route_caps & NETIF_F_SG) ||
8648b305 780 !(sk->sk_route_caps & NETIF_F_ALL_CSUM))
1da177e4
LT
781 return sock_no_sendpage(sock, page, offset, size, flags);
782
1da177e4
LT
783 lock_sock(sk);
784 TCP_CHECK_TIMER(sk);
785 res = do_tcp_sendpages(sk, &page, offset, size, flags);
786 TCP_CHECK_TIMER(sk);
787 release_sock(sk);
788 return res;
789}
790
791#define TCP_PAGE(sk) (sk->sk_sndmsg_page)
792#define TCP_OFF(sk) (sk->sk_sndmsg_off)
793
9e412ba7 794static inline int select_size(struct sock *sk)
1da177e4 795{
9e412ba7 796 struct tcp_sock *tp = tcp_sk(sk);
c1b4a7e6 797 int tmp = tp->mss_cache;
1da177e4 798
b4e26f5e 799 if (sk->sk_route_caps & NETIF_F_SG) {
bcd76111 800 if (sk_can_gso(sk))
b4e26f5e
DM
801 tmp = 0;
802 else {
803 int pgbreak = SKB_MAX_HEAD(MAX_TCP_HEADER);
804
805 if (tmp >= pgbreak &&
806 tmp <= pgbreak + (MAX_SKB_FRAGS - 1) * PAGE_SIZE)
807 tmp = pgbreak;
808 }
809 }
1da177e4 810
1da177e4
LT
811 return tmp;
812}
813
3516ffb0 814int tcp_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg,
1da177e4
LT
815 size_t size)
816{
3516ffb0 817 struct sock *sk = sock->sk;
1da177e4
LT
818 struct iovec *iov;
819 struct tcp_sock *tp = tcp_sk(sk);
820 struct sk_buff *skb;
821 int iovlen, flags;
c1b4a7e6 822 int mss_now, size_goal;
1da177e4
LT
823 int err, copied;
824 long timeo;
825
826 lock_sock(sk);
827 TCP_CHECK_TIMER(sk);
828
829 flags = msg->msg_flags;
830 timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
831
832 /* Wait for a connection to finish. */
833 if ((1 << sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT))
834 if ((err = sk_stream_wait_connect(sk, &timeo)) != 0)
835 goto out_err;
836
837 /* This should be in poll */
838 clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
839
840 mss_now = tcp_current_mss(sk, !(flags&MSG_OOB));
c1b4a7e6 841 size_goal = tp->xmit_size_goal;
1da177e4
LT
842
843 /* Ok commence sending. */
844 iovlen = msg->msg_iovlen;
845 iov = msg->msg_iov;
846 copied = 0;
847
848 err = -EPIPE;
849 if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
850 goto do_error;
851
852 while (--iovlen >= 0) {
853 int seglen = iov->iov_len;
854 unsigned char __user *from = iov->iov_base;
855
856 iov++;
857
858 while (seglen > 0) {
859 int copy;
860
fe067e8a 861 skb = tcp_write_queue_tail(sk);
1da177e4 862
fe067e8a 863 if (!tcp_send_head(sk) ||
c1b4a7e6 864 (copy = size_goal - skb->len) <= 0) {
1da177e4
LT
865
866new_segment:
867 /* Allocate new segment. If the interface is SG,
868 * allocate skb fitting to single page.
869 */
870 if (!sk_stream_memory_free(sk))
871 goto wait_for_sndbuf;
872
df97c708
PE
873 skb = sk_stream_alloc_skb(sk, select_size(sk),
874 sk->sk_allocation);
1da177e4
LT
875 if (!skb)
876 goto wait_for_memory;
877
878 /*
879 * Check whether we can use HW checksum.
880 */
8648b305 881 if (sk->sk_route_caps & NETIF_F_ALL_CSUM)
84fa7933 882 skb->ip_summed = CHECKSUM_PARTIAL;
1da177e4 883
9e412ba7 884 skb_entail(sk, skb);
c1b4a7e6 885 copy = size_goal;
1da177e4
LT
886 }
887
888 /* Try to append data to the end of skb. */
889 if (copy > seglen)
890 copy = seglen;
891
892 /* Where to copy to? */
893 if (skb_tailroom(skb) > 0) {
894 /* We have some space in skb head. Superb! */
895 if (copy > skb_tailroom(skb))
896 copy = skb_tailroom(skb);
897 if ((err = skb_add_data(skb, from, copy)) != 0)
898 goto do_fault;
899 } else {
900 int merge = 0;
901 int i = skb_shinfo(skb)->nr_frags;
902 struct page *page = TCP_PAGE(sk);
903 int off = TCP_OFF(sk);
904
905 if (skb_can_coalesce(skb, i, page, off) &&
906 off != PAGE_SIZE) {
907 /* We can extend the last page
908 * fragment. */
909 merge = 1;
910 } else if (i == MAX_SKB_FRAGS ||
911 (!i &&
912 !(sk->sk_route_caps & NETIF_F_SG))) {
913 /* Need to add new fragment and cannot
914 * do this because interface is non-SG,
915 * or because all the page slots are
916 * busy. */
917 tcp_mark_push(tp, skb);
918 goto new_segment;
919 } else if (page) {
1da177e4
LT
920 if (off == PAGE_SIZE) {
921 put_page(page);
922 TCP_PAGE(sk) = page = NULL;
fb5f5e6e 923 off = 0;
1da177e4 924 }
ef015786 925 } else
fb5f5e6e 926 off = 0;
ef015786
HX
927
928 if (copy > PAGE_SIZE - off)
929 copy = PAGE_SIZE - off;
930
3ab224be 931 if (!sk_wmem_schedule(sk, copy))
ef015786 932 goto wait_for_memory;
1da177e4
LT
933
934 if (!page) {
935 /* Allocate new cache page. */
936 if (!(page = sk_stream_alloc_page(sk)))
937 goto wait_for_memory;
1da177e4
LT
938 }
939
1da177e4
LT
940 /* Time to copy data. We are close to
941 * the end! */
942 err = skb_copy_to_page(sk, from, skb, page,
943 off, copy);
944 if (err) {
945 /* If this page was new, give it to the
946 * socket so it does not get leaked.
947 */
948 if (!TCP_PAGE(sk)) {
949 TCP_PAGE(sk) = page;
950 TCP_OFF(sk) = 0;
951 }
952 goto do_error;
953 }
954
955 /* Update the skb. */
956 if (merge) {
957 skb_shinfo(skb)->frags[i - 1].size +=
958 copy;
959 } else {
960 skb_fill_page_desc(skb, i, page, off, copy);
961 if (TCP_PAGE(sk)) {
962 get_page(page);
963 } else if (off + copy < PAGE_SIZE) {
964 get_page(page);
965 TCP_PAGE(sk) = page;
966 }
967 }
968
969 TCP_OFF(sk) = off + copy;
970 }
971
972 if (!copied)
973 TCP_SKB_CB(skb)->flags &= ~TCPCB_FLAG_PSH;
974
975 tp->write_seq += copy;
976 TCP_SKB_CB(skb)->end_seq += copy;
7967168c 977 skb_shinfo(skb)->gso_segs = 0;
1da177e4
LT
978
979 from += copy;
980 copied += copy;
981 if ((seglen -= copy) == 0 && iovlen == 0)
982 goto out;
983
c1b4a7e6 984 if (skb->len < mss_now || (flags & MSG_OOB))
1da177e4
LT
985 continue;
986
987 if (forced_push(tp)) {
988 tcp_mark_push(tp, skb);
9e412ba7 989 __tcp_push_pending_frames(sk, mss_now, TCP_NAGLE_PUSH);
fe067e8a 990 } else if (skb == tcp_send_head(sk))
1da177e4
LT
991 tcp_push_one(sk, mss_now);
992 continue;
993
994wait_for_sndbuf:
995 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
996wait_for_memory:
997 if (copied)
9e412ba7 998 tcp_push(sk, flags & ~MSG_MORE, mss_now, TCP_NAGLE_PUSH);
1da177e4
LT
999
1000 if ((err = sk_stream_wait_memory(sk, &timeo)) != 0)
1001 goto do_error;
1002
1003 mss_now = tcp_current_mss(sk, !(flags&MSG_OOB));
c1b4a7e6 1004 size_goal = tp->xmit_size_goal;
1da177e4
LT
1005 }
1006 }
1007
1008out:
1009 if (copied)
9e412ba7 1010 tcp_push(sk, flags, mss_now, tp->nonagle);
1da177e4
LT
1011 TCP_CHECK_TIMER(sk);
1012 release_sock(sk);
1013 return copied;
1014
1015do_fault:
1016 if (!skb->len) {
fe067e8a
DM
1017 tcp_unlink_write_queue(skb, sk);
1018 /* It is the one place in all of TCP, except connection
1019 * reset, where we can be unlinking the send_head.
1020 */
1021 tcp_check_send_head(sk, skb);
3ab224be 1022 sk_wmem_free_skb(sk, skb);
1da177e4
LT
1023 }
1024
1025do_error:
1026 if (copied)
1027 goto out;
1028out_err:
1029 err = sk_stream_error(sk, flags, err);
1030 TCP_CHECK_TIMER(sk);
1031 release_sock(sk);
1032 return err;
1033}
1034
1035/*
1036 * Handle reading urgent data. BSD has very simple semantics for
1037 * this, no blocking and very strange errors 8)
1038 */
1039
1040static int tcp_recv_urg(struct sock *sk, long timeo,
1041 struct msghdr *msg, int len, int flags,
1042 int *addr_len)
1043{
1044 struct tcp_sock *tp = tcp_sk(sk);
1045
1046 /* No URG data to read. */
1047 if (sock_flag(sk, SOCK_URGINLINE) || !tp->urg_data ||
1048 tp->urg_data == TCP_URG_READ)
1049 return -EINVAL; /* Yes this is right ! */
1050
1051 if (sk->sk_state == TCP_CLOSE && !sock_flag(sk, SOCK_DONE))
1052 return -ENOTCONN;
1053
1054 if (tp->urg_data & TCP_URG_VALID) {
1055 int err = 0;
1056 char c = tp->urg_data;
1057
1058 if (!(flags & MSG_PEEK))
1059 tp->urg_data = TCP_URG_READ;
1060
1061 /* Read urgent data. */
1062 msg->msg_flags |= MSG_OOB;
1063
1064 if (len > 0) {
1065 if (!(flags & MSG_TRUNC))
1066 err = memcpy_toiovec(msg->msg_iov, &c, 1);
1067 len = 1;
1068 } else
1069 msg->msg_flags |= MSG_TRUNC;
1070
1071 return err ? -EFAULT : len;
1072 }
1073
1074 if (sk->sk_state == TCP_CLOSE || (sk->sk_shutdown & RCV_SHUTDOWN))
1075 return 0;
1076
1077 /* Fixed the recv(..., MSG_OOB) behaviour. BSD docs and
1078 * the available implementations agree in this case:
1079 * this call should never block, independent of the
1080 * blocking state of the socket.
1081 * Mike <pall@rz.uni-karlsruhe.de>
1082 */
1083 return -EAGAIN;
1084}
1085
1086/* Clean up the receive buffer for full frames taken by the user,
1087 * then send an ACK if necessary. COPIED is the number of bytes
1088 * tcp_recvmsg has given to the user so far, it speeds up the
1089 * calculation of whether or not we must ACK for the sake of
1090 * a window update.
1091 */
0e4b4992 1092void tcp_cleanup_rbuf(struct sock *sk, int copied)
1da177e4
LT
1093{
1094 struct tcp_sock *tp = tcp_sk(sk);
1095 int time_to_ack = 0;
1096
1097#if TCP_DEBUG
1098 struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
1099
1100 BUG_TRAP(!skb || before(tp->copied_seq, TCP_SKB_CB(skb)->end_seq));
1101#endif
1102
463c84b9
ACM
1103 if (inet_csk_ack_scheduled(sk)) {
1104 const struct inet_connection_sock *icsk = inet_csk(sk);
1da177e4
LT
1105 /* Delayed ACKs frequently hit locked sockets during bulk
1106 * receive. */
463c84b9 1107 if (icsk->icsk_ack.blocked ||
1da177e4 1108 /* Once-per-two-segments ACK was not sent by tcp_input.c */
463c84b9 1109 tp->rcv_nxt - tp->rcv_wup > icsk->icsk_ack.rcv_mss ||
1da177e4
LT
1110 /*
1111 * If this read emptied read buffer, we send ACK, if
1112 * connection is not bidirectional, user drained
1113 * receive buffer and there was a small segment
1114 * in queue.
1115 */
1ef9696c
AK
1116 (copied > 0 &&
1117 ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED2) ||
1118 ((icsk->icsk_ack.pending & ICSK_ACK_PUSHED) &&
1119 !icsk->icsk_ack.pingpong)) &&
1120 !atomic_read(&sk->sk_rmem_alloc)))
1da177e4
LT
1121 time_to_ack = 1;
1122 }
1123
1124 /* We send an ACK if we can now advertise a non-zero window
1125 * which has been raised "significantly".
1126 *
1127 * Even if window raised up to infinity, do not send window open ACK
1128 * in states, where we will not receive more. It is useless.
1129 */
1130 if (copied > 0 && !time_to_ack && !(sk->sk_shutdown & RCV_SHUTDOWN)) {
1131 __u32 rcv_window_now = tcp_receive_window(tp);
1132
1133 /* Optimize, __tcp_select_window() is not cheap. */
1134 if (2*rcv_window_now <= tp->window_clamp) {
1135 __u32 new_window = __tcp_select_window(sk);
1136
1137 /* Send ACK now, if this read freed lots of space
1138 * in our buffer. Certainly, new_window is new window.
1139 * We can advertise it now, if it is not less than current one.
1140 * "Lots" means "at least twice" here.
1141 */
1142 if (new_window && new_window >= 2 * rcv_window_now)
1143 time_to_ack = 1;
1144 }
1145 }
1146 if (time_to_ack)
1147 tcp_send_ack(sk);
1148}
1149
1150static void tcp_prequeue_process(struct sock *sk)
1151{
1152 struct sk_buff *skb;
1153 struct tcp_sock *tp = tcp_sk(sk);
1154
b03efcfb 1155 NET_INC_STATS_USER(LINUX_MIB_TCPPREQUEUED);
1da177e4
LT
1156
1157 /* RX process wants to run with disabled BHs, though it is not
1158 * necessary */
1159 local_bh_disable();
1160 while ((skb = __skb_dequeue(&tp->ucopy.prequeue)) != NULL)
1161 sk->sk_backlog_rcv(sk, skb);
1162 local_bh_enable();
1163
1164 /* Clear memory counter. */
1165 tp->ucopy.memory = 0;
1166}
1167
1168static inline struct sk_buff *tcp_recv_skb(struct sock *sk, u32 seq, u32 *off)
1169{
1170 struct sk_buff *skb;
1171 u32 offset;
1172
1173 skb_queue_walk(&sk->sk_receive_queue, skb) {
1174 offset = seq - TCP_SKB_CB(skb)->seq;
aa8223c7 1175 if (tcp_hdr(skb)->syn)
1da177e4 1176 offset--;
aa8223c7 1177 if (offset < skb->len || tcp_hdr(skb)->fin) {
1da177e4
LT
1178 *off = offset;
1179 return skb;
1180 }
1181 }
1182 return NULL;
1183}
1184
1185/*
1186 * This routine provides an alternative to tcp_recvmsg() for routines
1187 * that would like to handle copying from skbuffs directly in 'sendfile'
1188 * fashion.
1189 * Note:
1190 * - It is assumed that the socket was locked by the caller.
1191 * - The routine does not block.
1192 * - At present, there is no support for reading OOB data
1193 * or for 'peeking' the socket using this routine
1194 * (although both would be easy to implement).
1195 */
1196int tcp_read_sock(struct sock *sk, read_descriptor_t *desc,
1197 sk_read_actor_t recv_actor)
1198{
1199 struct sk_buff *skb;
1200 struct tcp_sock *tp = tcp_sk(sk);
1201 u32 seq = tp->copied_seq;
1202 u32 offset;
1203 int copied = 0;
1204
1205 if (sk->sk_state == TCP_LISTEN)
1206 return -ENOTCONN;
1207 while ((skb = tcp_recv_skb(sk, seq, &offset)) != NULL) {
1208 if (offset < skb->len) {
1209 size_t used, len;
1210
1211 len = skb->len - offset;
1212 /* Stop reading if we hit a patch of urgent data */
1213 if (tp->urg_data) {
1214 u32 urg_offset = tp->urg_seq - seq;
1215 if (urg_offset < len)
1216 len = urg_offset;
1217 if (!len)
1218 break;
1219 }
1220 used = recv_actor(desc, skb, offset, len);
ddb61a57
JA
1221 if (used < 0) {
1222 if (!copied)
1223 copied = used;
1224 break;
1225 } else if (used <= len) {
1da177e4
LT
1226 seq += used;
1227 copied += used;
1228 offset += used;
1229 }
1230 if (offset != skb->len)
1231 break;
1232 }
aa8223c7 1233 if (tcp_hdr(skb)->fin) {
624d1164 1234 sk_eat_skb(sk, skb, 0);
1da177e4
LT
1235 ++seq;
1236 break;
1237 }
624d1164 1238 sk_eat_skb(sk, skb, 0);
1da177e4
LT
1239 if (!desc->count)
1240 break;
1241 }
1242 tp->copied_seq = seq;
1243
1244 tcp_rcv_space_adjust(sk);
1245
1246 /* Clean up data we have read: This will do ACK frames. */
ddb61a57 1247 if (copied > 0)
0e4b4992 1248 tcp_cleanup_rbuf(sk, copied);
1da177e4
LT
1249 return copied;
1250}
1251
1252/*
1253 * This routine copies from a sock struct into the user buffer.
1254 *
1255 * Technical note: in 2.3 we work on _locked_ socket, so that
1256 * tricks with *seq access order and skb->users are not required.
1257 * Probably, code can be easily improved even more.
1258 */
1259
1260int tcp_recvmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
1261 size_t len, int nonblock, int flags, int *addr_len)
1262{
1263 struct tcp_sock *tp = tcp_sk(sk);
1264 int copied = 0;
1265 u32 peek_seq;
1266 u32 *seq;
1267 unsigned long used;
1268 int err;
1269 int target; /* Read at least this many bytes */
1270 long timeo;
1271 struct task_struct *user_recv = NULL;
1a2449a8 1272 int copied_early = 0;
2b1244a4 1273 struct sk_buff *skb;
1da177e4
LT
1274
1275 lock_sock(sk);
1276
1277 TCP_CHECK_TIMER(sk);
1278
1279 err = -ENOTCONN;
1280 if (sk->sk_state == TCP_LISTEN)
1281 goto out;
1282
1283 timeo = sock_rcvtimeo(sk, nonblock);
1284
1285 /* Urgent data needs to be handled specially. */
1286 if (flags & MSG_OOB)
1287 goto recv_urg;
1288
1289 seq = &tp->copied_seq;
1290 if (flags & MSG_PEEK) {
1291 peek_seq = tp->copied_seq;
1292 seq = &peek_seq;
1293 }
1294
1295 target = sock_rcvlowat(sk, flags & MSG_WAITALL, len);
1296
1a2449a8
CL
1297#ifdef CONFIG_NET_DMA
1298 tp->ucopy.dma_chan = NULL;
1299 preempt_disable();
2b1244a4 1300 skb = skb_peek_tail(&sk->sk_receive_queue);
e00c5d8b
AM
1301 {
1302 int available = 0;
1303
1304 if (skb)
1305 available = TCP_SKB_CB(skb)->seq + skb->len - (*seq);
1306 if ((available < target) &&
1307 (len > sysctl_tcp_dma_copybreak) && !(flags & MSG_PEEK) &&
1308 !sysctl_tcp_low_latency &&
1309 __get_cpu_var(softnet_data).net_dma) {
1310 preempt_enable_no_resched();
1311 tp->ucopy.pinned_list =
1312 dma_pin_iovec_pages(msg->msg_iov, len);
1313 } else {
1314 preempt_enable_no_resched();
1315 }
1316 }
1a2449a8
CL
1317#endif
1318
1da177e4 1319 do {
1da177e4
LT
1320 u32 offset;
1321
1322 /* Are we at urgent data? Stop if we have read anything or have SIGURG pending. */
1323 if (tp->urg_data && tp->urg_seq == *seq) {
1324 if (copied)
1325 break;
1326 if (signal_pending(current)) {
1327 copied = timeo ? sock_intr_errno(timeo) : -EAGAIN;
1328 break;
1329 }
1330 }
1331
1332 /* Next get a buffer. */
1333
1334 skb = skb_peek(&sk->sk_receive_queue);
1335 do {
1336 if (!skb)
1337 break;
1338
1339 /* Now that we have two receive queues this
1340 * shouldn't happen.
1341 */
1342 if (before(*seq, TCP_SKB_CB(skb)->seq)) {
1343 printk(KERN_INFO "recvmsg bug: copied %X "
1344 "seq %X\n", *seq, TCP_SKB_CB(skb)->seq);
1345 break;
1346 }
1347 offset = *seq - TCP_SKB_CB(skb)->seq;
aa8223c7 1348 if (tcp_hdr(skb)->syn)
1da177e4
LT
1349 offset--;
1350 if (offset < skb->len)
1351 goto found_ok_skb;
aa8223c7 1352 if (tcp_hdr(skb)->fin)
1da177e4
LT
1353 goto found_fin_ok;
1354 BUG_TRAP(flags & MSG_PEEK);
1355 skb = skb->next;
1356 } while (skb != (struct sk_buff *)&sk->sk_receive_queue);
1357
1358 /* Well, if we have backlog, try to process it now yet. */
1359
1360 if (copied >= target && !sk->sk_backlog.tail)
1361 break;
1362
1363 if (copied) {
1364 if (sk->sk_err ||
1365 sk->sk_state == TCP_CLOSE ||
1366 (sk->sk_shutdown & RCV_SHUTDOWN) ||
1367 !timeo ||
1368 signal_pending(current) ||
1369 (flags & MSG_PEEK))
1370 break;
1371 } else {
1372 if (sock_flag(sk, SOCK_DONE))
1373 break;
1374
1375 if (sk->sk_err) {
1376 copied = sock_error(sk);
1377 break;
1378 }
1379
1380 if (sk->sk_shutdown & RCV_SHUTDOWN)
1381 break;
1382
1383 if (sk->sk_state == TCP_CLOSE) {
1384 if (!sock_flag(sk, SOCK_DONE)) {
1385 /* This occurs when user tries to read
1386 * from never connected socket.
1387 */
1388 copied = -ENOTCONN;
1389 break;
1390 }
1391 break;
1392 }
1393
1394 if (!timeo) {
1395 copied = -EAGAIN;
1396 break;
1397 }
1398
1399 if (signal_pending(current)) {
1400 copied = sock_intr_errno(timeo);
1401 break;
1402 }
1403 }
1404
0e4b4992 1405 tcp_cleanup_rbuf(sk, copied);
1da177e4 1406
7df55125 1407 if (!sysctl_tcp_low_latency && tp->ucopy.task == user_recv) {
1da177e4
LT
1408 /* Install new reader */
1409 if (!user_recv && !(flags & (MSG_TRUNC | MSG_PEEK))) {
1410 user_recv = current;
1411 tp->ucopy.task = user_recv;
1412 tp->ucopy.iov = msg->msg_iov;
1413 }
1414
1415 tp->ucopy.len = len;
1416
1417 BUG_TRAP(tp->copied_seq == tp->rcv_nxt ||
1418 (flags & (MSG_PEEK | MSG_TRUNC)));
1419
1420 /* Ugly... If prequeue is not empty, we have to
1421 * process it before releasing socket, otherwise
1422 * order will be broken at second iteration.
1423 * More elegant solution is required!!!
1424 *
1425 * Look: we have the following (pseudo)queues:
1426 *
1427 * 1. packets in flight
1428 * 2. backlog
1429 * 3. prequeue
1430 * 4. receive_queue
1431 *
1432 * Each queue can be processed only if the next ones
1433 * are empty. At this point we have empty receive_queue.
1434 * But prequeue _can_ be not empty after 2nd iteration,
1435 * when we jumped to start of loop because backlog
1436 * processing added something to receive_queue.
1437 * We cannot release_sock(), because backlog contains
1438 * packets arrived _after_ prequeued ones.
1439 *
1440 * Shortly, algorithm is clear --- to process all
1441 * the queues in order. We could make it more directly,
1442 * requeueing packets from backlog to prequeue, if
1443 * is not empty. It is more elegant, but eats cycles,
1444 * unfortunately.
1445 */
b03efcfb 1446 if (!skb_queue_empty(&tp->ucopy.prequeue))
1da177e4
LT
1447 goto do_prequeue;
1448
1449 /* __ Set realtime policy in scheduler __ */
1450 }
1451
1452 if (copied >= target) {
1453 /* Do not sleep, just process backlog. */
1454 release_sock(sk);
1455 lock_sock(sk);
1456 } else
1457 sk_wait_data(sk, &timeo);
1458
1a2449a8
CL
1459#ifdef CONFIG_NET_DMA
1460 tp->ucopy.wakeup = 0;
1461#endif
1462
1da177e4
LT
1463 if (user_recv) {
1464 int chunk;
1465
1466 /* __ Restore normal policy in scheduler __ */
1467
1468 if ((chunk = len - tp->ucopy.len) != 0) {
1469 NET_ADD_STATS_USER(LINUX_MIB_TCPDIRECTCOPYFROMBACKLOG, chunk);
1470 len -= chunk;
1471 copied += chunk;
1472 }
1473
1474 if (tp->rcv_nxt == tp->copied_seq &&
b03efcfb 1475 !skb_queue_empty(&tp->ucopy.prequeue)) {
1da177e4
LT
1476do_prequeue:
1477 tcp_prequeue_process(sk);
1478
1479 if ((chunk = len - tp->ucopy.len) != 0) {
1480 NET_ADD_STATS_USER(LINUX_MIB_TCPDIRECTCOPYFROMPREQUEUE, chunk);
1481 len -= chunk;
1482 copied += chunk;
1483 }
1484 }
1485 }
1486 if ((flags & MSG_PEEK) && peek_seq != tp->copied_seq) {
1487 if (net_ratelimit())
1488 printk(KERN_DEBUG "TCP(%s:%d): Application bug, race in MSG_PEEK.\n",
ba25f9dc 1489 current->comm, task_pid_nr(current));
1da177e4
LT
1490 peek_seq = tp->copied_seq;
1491 }
1492 continue;
1493
1494 found_ok_skb:
1495 /* Ok so how much can we use? */
1496 used = skb->len - offset;
1497 if (len < used)
1498 used = len;
1499
1500 /* Do we have urgent data here? */
1501 if (tp->urg_data) {
1502 u32 urg_offset = tp->urg_seq - *seq;
1503 if (urg_offset < used) {
1504 if (!urg_offset) {
1505 if (!sock_flag(sk, SOCK_URGINLINE)) {
1506 ++*seq;
1507 offset++;
1508 used--;
1509 if (!used)
1510 goto skip_copy;
1511 }
1512 } else
1513 used = urg_offset;
1514 }
1515 }
1516
1517 if (!(flags & MSG_TRUNC)) {
1a2449a8
CL
1518#ifdef CONFIG_NET_DMA
1519 if (!tp->ucopy.dma_chan && tp->ucopy.pinned_list)
1520 tp->ucopy.dma_chan = get_softnet_dma();
1521
1522 if (tp->ucopy.dma_chan) {
1523 tp->ucopy.dma_cookie = dma_skb_copy_datagram_iovec(
1524 tp->ucopy.dma_chan, skb, offset,
1525 msg->msg_iov, used,
1526 tp->ucopy.pinned_list);
1527
1528 if (tp->ucopy.dma_cookie < 0) {
1529
1530 printk(KERN_ALERT "dma_cookie < 0\n");
1531
1532 /* Exception. Bailout! */
1533 if (!copied)
1534 copied = -EFAULT;
1535 break;
1536 }
1537 if ((offset + used) == skb->len)
1538 copied_early = 1;
1539
1540 } else
1541#endif
1542 {
1543 err = skb_copy_datagram_iovec(skb, offset,
1544 msg->msg_iov, used);
1545 if (err) {
1546 /* Exception. Bailout! */
1547 if (!copied)
1548 copied = -EFAULT;
1549 break;
1550 }
1da177e4
LT
1551 }
1552 }
1553
1554 *seq += used;
1555 copied += used;
1556 len -= used;
1557
1558 tcp_rcv_space_adjust(sk);
1559
1560skip_copy:
1561 if (tp->urg_data && after(tp->copied_seq, tp->urg_seq)) {
1562 tp->urg_data = 0;
9e412ba7 1563 tcp_fast_path_check(sk);
1da177e4
LT
1564 }
1565 if (used + offset < skb->len)
1566 continue;
1567
aa8223c7 1568 if (tcp_hdr(skb)->fin)
1da177e4 1569 goto found_fin_ok;
1a2449a8
CL
1570 if (!(flags & MSG_PEEK)) {
1571 sk_eat_skb(sk, skb, copied_early);
1572 copied_early = 0;
1573 }
1da177e4
LT
1574 continue;
1575
1576 found_fin_ok:
1577 /* Process the FIN. */
1578 ++*seq;
1a2449a8
CL
1579 if (!(flags & MSG_PEEK)) {
1580 sk_eat_skb(sk, skb, copied_early);
1581 copied_early = 0;
1582 }
1da177e4
LT
1583 break;
1584 } while (len > 0);
1585
1586 if (user_recv) {
b03efcfb 1587 if (!skb_queue_empty(&tp->ucopy.prequeue)) {
1da177e4
LT
1588 int chunk;
1589
1590 tp->ucopy.len = copied > 0 ? len : 0;
1591
1592 tcp_prequeue_process(sk);
1593
1594 if (copied > 0 && (chunk = len - tp->ucopy.len) != 0) {
1595 NET_ADD_STATS_USER(LINUX_MIB_TCPDIRECTCOPYFROMPREQUEUE, chunk);
1596 len -= chunk;
1597 copied += chunk;
1598 }
1599 }
1600
1601 tp->ucopy.task = NULL;
1602 tp->ucopy.len = 0;
1603 }
1604
1a2449a8
CL
1605#ifdef CONFIG_NET_DMA
1606 if (tp->ucopy.dma_chan) {
1a2449a8
CL
1607 dma_cookie_t done, used;
1608
1609 dma_async_memcpy_issue_pending(tp->ucopy.dma_chan);
1610
1611 while (dma_async_memcpy_complete(tp->ucopy.dma_chan,
e905a9ed
YH
1612 tp->ucopy.dma_cookie, &done,
1613 &used) == DMA_IN_PROGRESS) {
1a2449a8
CL
1614 /* do partial cleanup of sk_async_wait_queue */
1615 while ((skb = skb_peek(&sk->sk_async_wait_queue)) &&
1616 (dma_async_is_complete(skb->dma_cookie, done,
e905a9ed 1617 used) == DMA_SUCCESS)) {
1a2449a8
CL
1618 __skb_dequeue(&sk->sk_async_wait_queue);
1619 kfree_skb(skb);
1620 }
1621 }
1622
1623 /* Safe to free early-copied skbs now */
1624 __skb_queue_purge(&sk->sk_async_wait_queue);
1625 dma_chan_put(tp->ucopy.dma_chan);
1626 tp->ucopy.dma_chan = NULL;
1627 }
1628 if (tp->ucopy.pinned_list) {
1629 dma_unpin_iovec_pages(tp->ucopy.pinned_list);
1630 tp->ucopy.pinned_list = NULL;
1631 }
1632#endif
1633
1da177e4
LT
1634 /* According to UNIX98, msg_name/msg_namelen are ignored
1635 * on connected socket. I was just happy when found this 8) --ANK
1636 */
1637
1638 /* Clean up data we have read: This will do ACK frames. */
0e4b4992 1639 tcp_cleanup_rbuf(sk, copied);
1da177e4
LT
1640
1641 TCP_CHECK_TIMER(sk);
1642 release_sock(sk);
1643 return copied;
1644
1645out:
1646 TCP_CHECK_TIMER(sk);
1647 release_sock(sk);
1648 return err;
1649
1650recv_urg:
1651 err = tcp_recv_urg(sk, timeo, msg, len, flags, addr_len);
1652 goto out;
1653}
1654
1655/*
1656 * State processing on a close. This implements the state shift for
1657 * sending our FIN frame. Note that we only send a FIN for some
1658 * states. A shutdown() may have already sent the FIN, or we may be
1659 * closed.
1660 */
1661
9b5b5cff 1662static const unsigned char new_state[16] = {
1da177e4
LT
1663 /* current state: new state: action: */
1664 /* (Invalid) */ TCP_CLOSE,
1665 /* TCP_ESTABLISHED */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
1666 /* TCP_SYN_SENT */ TCP_CLOSE,
1667 /* TCP_SYN_RECV */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
1668 /* TCP_FIN_WAIT1 */ TCP_FIN_WAIT1,
1669 /* TCP_FIN_WAIT2 */ TCP_FIN_WAIT2,
1670 /* TCP_TIME_WAIT */ TCP_CLOSE,
1671 /* TCP_CLOSE */ TCP_CLOSE,
1672 /* TCP_CLOSE_WAIT */ TCP_LAST_ACK | TCP_ACTION_FIN,
1673 /* TCP_LAST_ACK */ TCP_LAST_ACK,
1674 /* TCP_LISTEN */ TCP_CLOSE,
1675 /* TCP_CLOSING */ TCP_CLOSING,
1676};
1677
1678static int tcp_close_state(struct sock *sk)
1679{
1680 int next = (int)new_state[sk->sk_state];
1681 int ns = next & TCP_STATE_MASK;
1682
1683 tcp_set_state(sk, ns);
1684
1685 return next & TCP_ACTION_FIN;
1686}
1687
1688/*
1689 * Shutdown the sending side of a connection. Much like close except
1690 * that we don't receive shut down or set_sock_flag(sk, SOCK_DEAD).
1691 */
1692
1693void tcp_shutdown(struct sock *sk, int how)
1694{
1695 /* We need to grab some memory, and put together a FIN,
1696 * and then put it into the queue to be sent.
1697 * Tim MacKenzie(tym@dibbler.cs.monash.edu.au) 4 Dec '92.
1698 */
1699 if (!(how & SEND_SHUTDOWN))
1700 return;
1701
1702 /* If we've already sent a FIN, or it's a closed state, skip this. */
1703 if ((1 << sk->sk_state) &
1704 (TCPF_ESTABLISHED | TCPF_SYN_SENT |
1705 TCPF_SYN_RECV | TCPF_CLOSE_WAIT)) {
1706 /* Clear out any half completed packets. FIN if needed. */
1707 if (tcp_close_state(sk))
1708 tcp_send_fin(sk);
1709 }
1710}
1711
1da177e4
LT
1712void tcp_close(struct sock *sk, long timeout)
1713{
1714 struct sk_buff *skb;
1715 int data_was_unread = 0;
75c2d907 1716 int state;
1da177e4
LT
1717
1718 lock_sock(sk);
1719 sk->sk_shutdown = SHUTDOWN_MASK;
1720
1721 if (sk->sk_state == TCP_LISTEN) {
1722 tcp_set_state(sk, TCP_CLOSE);
1723
1724 /* Special case. */
0a5578cf 1725 inet_csk_listen_stop(sk);
1da177e4
LT
1726
1727 goto adjudge_to_death;
1728 }
1729
1730 /* We need to flush the recv. buffs. We do this only on the
1731 * descriptor close, not protocol-sourced closes, because the
1732 * reader process may not have drained the data yet!
1733 */
1734 while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
1735 u32 len = TCP_SKB_CB(skb)->end_seq - TCP_SKB_CB(skb)->seq -
aa8223c7 1736 tcp_hdr(skb)->fin;
1da177e4
LT
1737 data_was_unread += len;
1738 __kfree_skb(skb);
1739 }
1740
3ab224be 1741 sk_mem_reclaim(sk);
1da177e4 1742
65bb723c
GR
1743 /* As outlined in RFC 2525, section 2.17, we send a RST here because
1744 * data was lost. To witness the awful effects of the old behavior of
1745 * always doing a FIN, run an older 2.1.x kernel or 2.0.x, start a bulk
1746 * GET in an FTP client, suspend the process, wait for the client to
1747 * advertise a zero window, then kill -9 the FTP client, wheee...
1748 * Note: timeout is always zero in such a case.
1da177e4
LT
1749 */
1750 if (data_was_unread) {
1751 /* Unread data was tossed, zap the connection. */
1752 NET_INC_STATS_USER(LINUX_MIB_TCPABORTONCLOSE);
1753 tcp_set_state(sk, TCP_CLOSE);
1754 tcp_send_active_reset(sk, GFP_KERNEL);
1755 } else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
1756 /* Check zero linger _after_ checking for unread data. */
1757 sk->sk_prot->disconnect(sk, 0);
1758 NET_INC_STATS_USER(LINUX_MIB_TCPABORTONDATA);
1759 } else if (tcp_close_state(sk)) {
1760 /* We FIN if the application ate all the data before
1761 * zapping the connection.
1762 */
1763
1764 /* RED-PEN. Formally speaking, we have broken TCP state
1765 * machine. State transitions:
1766 *
1767 * TCP_ESTABLISHED -> TCP_FIN_WAIT1
1768 * TCP_SYN_RECV -> TCP_FIN_WAIT1 (forget it, it's impossible)
1769 * TCP_CLOSE_WAIT -> TCP_LAST_ACK
1770 *
1771 * are legal only when FIN has been sent (i.e. in window),
1772 * rather than queued out of window. Purists blame.
1773 *
1774 * F.e. "RFC state" is ESTABLISHED,
1775 * if Linux state is FIN-WAIT-1, but FIN is still not sent.
1776 *
1777 * The visible declinations are that sometimes
1778 * we enter time-wait state, when it is not required really
1779 * (harmless), do not send active resets, when they are
1780 * required by specs (TCP_ESTABLISHED, TCP_CLOSE_WAIT, when
1781 * they look as CLOSING or LAST_ACK for Linux)
1782 * Probably, I missed some more holelets.
1783 * --ANK
1784 */
1785 tcp_send_fin(sk);
1786 }
1787
1788 sk_stream_wait_close(sk, timeout);
1789
1790adjudge_to_death:
75c2d907
HX
1791 state = sk->sk_state;
1792 sock_hold(sk);
1793 sock_orphan(sk);
1794 atomic_inc(sk->sk_prot->orphan_count);
1795
1da177e4
LT
1796 /* It is the last release_sock in its life. It will remove backlog. */
1797 release_sock(sk);
1798
1799
1800 /* Now socket is owned by kernel and we acquire BH lock
1801 to finish close. No need to check for user refs.
1802 */
1803 local_bh_disable();
1804 bh_lock_sock(sk);
1805 BUG_TRAP(!sock_owned_by_user(sk));
1806
75c2d907
HX
1807 /* Have we already been destroyed by a softirq or backlog? */
1808 if (state != TCP_CLOSE && sk->sk_state == TCP_CLOSE)
1809 goto out;
1da177e4
LT
1810
1811 /* This is a (useful) BSD violating of the RFC. There is a
1812 * problem with TCP as specified in that the other end could
1813 * keep a socket open forever with no application left this end.
1814 * We use a 3 minute timeout (about the same as BSD) then kill
1815 * our end. If they send after that then tough - BUT: long enough
1816 * that we won't make the old 4*rto = almost no time - whoops
1817 * reset mistake.
1818 *
1819 * Nope, it was not mistake. It is really desired behaviour
1820 * f.e. on http servers, when such sockets are useless, but
1821 * consume significant resources. Let's do it with special
1822 * linger2 option. --ANK
1823 */
1824
1825 if (sk->sk_state == TCP_FIN_WAIT2) {
1826 struct tcp_sock *tp = tcp_sk(sk);
1827 if (tp->linger2 < 0) {
1828 tcp_set_state(sk, TCP_CLOSE);
1829 tcp_send_active_reset(sk, GFP_ATOMIC);
1830 NET_INC_STATS_BH(LINUX_MIB_TCPABORTONLINGER);
1831 } else {
463c84b9 1832 const int tmo = tcp_fin_time(sk);
1da177e4
LT
1833
1834 if (tmo > TCP_TIMEWAIT_LEN) {
52499afe
DM
1835 inet_csk_reset_keepalive_timer(sk,
1836 tmo - TCP_TIMEWAIT_LEN);
1da177e4 1837 } else {
1da177e4
LT
1838 tcp_time_wait(sk, TCP_FIN_WAIT2, tmo);
1839 goto out;
1840 }
1841 }
1842 }
1843 if (sk->sk_state != TCP_CLOSE) {
3ab224be 1844 sk_mem_reclaim(sk);
e4fd5da3
PE
1845 if (tcp_too_many_orphans(sk,
1846 atomic_read(sk->sk_prot->orphan_count))) {
1da177e4
LT
1847 if (net_ratelimit())
1848 printk(KERN_INFO "TCP: too many of orphaned "
1849 "sockets\n");
1850 tcp_set_state(sk, TCP_CLOSE);
1851 tcp_send_active_reset(sk, GFP_ATOMIC);
1852 NET_INC_STATS_BH(LINUX_MIB_TCPABORTONMEMORY);
1853 }
1854 }
1da177e4
LT
1855
1856 if (sk->sk_state == TCP_CLOSE)
0a5578cf 1857 inet_csk_destroy_sock(sk);
1da177e4
LT
1858 /* Otherwise, socket is reprieved until protocol close. */
1859
1860out:
1861 bh_unlock_sock(sk);
1862 local_bh_enable();
1863 sock_put(sk);
1864}
1865
1866/* These states need RST on ABORT according to RFC793 */
1867
1868static inline int tcp_need_reset(int state)
1869{
1870 return (1 << state) &
1871 (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT | TCPF_FIN_WAIT1 |
1872 TCPF_FIN_WAIT2 | TCPF_SYN_RECV);
1873}
1874
1875int tcp_disconnect(struct sock *sk, int flags)
1876{
1877 struct inet_sock *inet = inet_sk(sk);
463c84b9 1878 struct inet_connection_sock *icsk = inet_csk(sk);
1da177e4
LT
1879 struct tcp_sock *tp = tcp_sk(sk);
1880 int err = 0;
1881 int old_state = sk->sk_state;
1882
1883 if (old_state != TCP_CLOSE)
1884 tcp_set_state(sk, TCP_CLOSE);
1885
1886 /* ABORT function of RFC793 */
1887 if (old_state == TCP_LISTEN) {
0a5578cf 1888 inet_csk_listen_stop(sk);
1da177e4
LT
1889 } else if (tcp_need_reset(old_state) ||
1890 (tp->snd_nxt != tp->write_seq &&
1891 (1 << old_state) & (TCPF_CLOSING | TCPF_LAST_ACK))) {
caa20d9a 1892 /* The last check adjusts for discrepancy of Linux wrt. RFC
1da177e4
LT
1893 * states
1894 */
1895 tcp_send_active_reset(sk, gfp_any());
1896 sk->sk_err = ECONNRESET;
1897 } else if (old_state == TCP_SYN_SENT)
1898 sk->sk_err = ECONNRESET;
1899
1900 tcp_clear_xmit_timers(sk);
1901 __skb_queue_purge(&sk->sk_receive_queue);
fe067e8a 1902 tcp_write_queue_purge(sk);
1da177e4 1903 __skb_queue_purge(&tp->out_of_order_queue);
1a2449a8
CL
1904#ifdef CONFIG_NET_DMA
1905 __skb_queue_purge(&sk->sk_async_wait_queue);
1906#endif
1da177e4
LT
1907
1908 inet->dport = 0;
1909
1910 if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK))
1911 inet_reset_saddr(sk);
1912
1913 sk->sk_shutdown = 0;
1914 sock_reset_flag(sk, SOCK_DONE);
1915 tp->srtt = 0;
1916 if ((tp->write_seq += tp->max_window + 2) == 0)
1917 tp->write_seq = 1;
463c84b9 1918 icsk->icsk_backoff = 0;
1da177e4 1919 tp->snd_cwnd = 2;
6687e988 1920 icsk->icsk_probes_out = 0;
1da177e4
LT
1921 tp->packets_out = 0;
1922 tp->snd_ssthresh = 0x7fffffff;
1923 tp->snd_cwnd_cnt = 0;
9772efb9 1924 tp->bytes_acked = 0;
6687e988 1925 tcp_set_ca_state(sk, TCP_CA_Open);
1da177e4 1926 tcp_clear_retrans(tp);
463c84b9 1927 inet_csk_delack_init(sk);
fe067e8a 1928 tcp_init_send_head(sk);
b40b4f79 1929 memset(&tp->rx_opt, 0, sizeof(tp->rx_opt));
1da177e4
LT
1930 __sk_dst_reset(sk);
1931
463c84b9 1932 BUG_TRAP(!inet->num || icsk->icsk_bind_hash);
1da177e4
LT
1933
1934 sk->sk_error_report(sk);
1935 return err;
1936}
1937
1da177e4
LT
1938/*
1939 * Socket option code for TCP.
1940 */
3fdadf7d
DM
1941static int do_tcp_setsockopt(struct sock *sk, int level,
1942 int optname, char __user *optval, int optlen)
1da177e4
LT
1943{
1944 struct tcp_sock *tp = tcp_sk(sk);
463c84b9 1945 struct inet_connection_sock *icsk = inet_csk(sk);
1da177e4
LT
1946 int val;
1947 int err = 0;
1948
5f8ef48d
SH
1949 /* This is a string value all the others are int's */
1950 if (optname == TCP_CONGESTION) {
1951 char name[TCP_CA_NAME_MAX];
1952
1953 if (optlen < 1)
1954 return -EINVAL;
1955
1956 val = strncpy_from_user(name, optval,
1957 min(TCP_CA_NAME_MAX-1, optlen));
1958 if (val < 0)
1959 return -EFAULT;
1960 name[val] = 0;
1961
1962 lock_sock(sk);
6687e988 1963 err = tcp_set_congestion_control(sk, name);
5f8ef48d
SH
1964 release_sock(sk);
1965 return err;
1966 }
1967
1da177e4
LT
1968 if (optlen < sizeof(int))
1969 return -EINVAL;
1970
1971 if (get_user(val, (int __user *)optval))
1972 return -EFAULT;
1973
1974 lock_sock(sk);
1975
1976 switch (optname) {
1977 case TCP_MAXSEG:
1978 /* Values greater than interface MTU won't take effect. However
1979 * at the point when this call is done we typically don't yet
1980 * know which interface is going to be used */
1981 if (val < 8 || val > MAX_TCP_WINDOW) {
1982 err = -EINVAL;
1983 break;
1984 }
1985 tp->rx_opt.user_mss = val;
1986 break;
1987
1988 case TCP_NODELAY:
1989 if (val) {
1990 /* TCP_NODELAY is weaker than TCP_CORK, so that
1991 * this option on corked socket is remembered, but
1992 * it is not activated until cork is cleared.
1993 *
1994 * However, when TCP_NODELAY is set we make
1995 * an explicit push, which overrides even TCP_CORK
1996 * for currently queued segments.
1997 */
1998 tp->nonagle |= TCP_NAGLE_OFF|TCP_NAGLE_PUSH;
9e412ba7 1999 tcp_push_pending_frames(sk);
1da177e4
LT
2000 } else {
2001 tp->nonagle &= ~TCP_NAGLE_OFF;
2002 }
2003 break;
2004
2005 case TCP_CORK:
2006 /* When set indicates to always queue non-full frames.
2007 * Later the user clears this option and we transmit
2008 * any pending partial frames in the queue. This is
2009 * meant to be used alongside sendfile() to get properly
2010 * filled frames when the user (for example) must write
2011 * out headers with a write() call first and then use
2012 * sendfile to send out the data parts.
2013 *
2014 * TCP_CORK can be set together with TCP_NODELAY and it is
2015 * stronger than TCP_NODELAY.
2016 */
2017 if (val) {
2018 tp->nonagle |= TCP_NAGLE_CORK;
2019 } else {
2020 tp->nonagle &= ~TCP_NAGLE_CORK;
2021 if (tp->nonagle&TCP_NAGLE_OFF)
2022 tp->nonagle |= TCP_NAGLE_PUSH;
9e412ba7 2023 tcp_push_pending_frames(sk);
1da177e4
LT
2024 }
2025 break;
2026
2027 case TCP_KEEPIDLE:
2028 if (val < 1 || val > MAX_TCP_KEEPIDLE)
2029 err = -EINVAL;
2030 else {
2031 tp->keepalive_time = val * HZ;
2032 if (sock_flag(sk, SOCK_KEEPOPEN) &&
2033 !((1 << sk->sk_state) &
2034 (TCPF_CLOSE | TCPF_LISTEN))) {
2035 __u32 elapsed = tcp_time_stamp - tp->rcv_tstamp;
2036 if (tp->keepalive_time > elapsed)
2037 elapsed = tp->keepalive_time - elapsed;
2038 else
2039 elapsed = 0;
463c84b9 2040 inet_csk_reset_keepalive_timer(sk, elapsed);
1da177e4
LT
2041 }
2042 }
2043 break;
2044 case TCP_KEEPINTVL:
2045 if (val < 1 || val > MAX_TCP_KEEPINTVL)
2046 err = -EINVAL;
2047 else
2048 tp->keepalive_intvl = val * HZ;
2049 break;
2050 case TCP_KEEPCNT:
2051 if (val < 1 || val > MAX_TCP_KEEPCNT)
2052 err = -EINVAL;
2053 else
2054 tp->keepalive_probes = val;
2055 break;
2056 case TCP_SYNCNT:
2057 if (val < 1 || val > MAX_TCP_SYNCNT)
2058 err = -EINVAL;
2059 else
463c84b9 2060 icsk->icsk_syn_retries = val;
1da177e4
LT
2061 break;
2062
2063 case TCP_LINGER2:
2064 if (val < 0)
2065 tp->linger2 = -1;
2066 else if (val > sysctl_tcp_fin_timeout / HZ)
2067 tp->linger2 = 0;
2068 else
2069 tp->linger2 = val * HZ;
2070 break;
2071
2072 case TCP_DEFER_ACCEPT:
295f7324 2073 icsk->icsk_accept_queue.rskq_defer_accept = 0;
1da177e4
LT
2074 if (val > 0) {
2075 /* Translate value in seconds to number of
2076 * retransmits */
295f7324 2077 while (icsk->icsk_accept_queue.rskq_defer_accept < 32 &&
1da177e4 2078 val > ((TCP_TIMEOUT_INIT / HZ) <<
295f7324
ACM
2079 icsk->icsk_accept_queue.rskq_defer_accept))
2080 icsk->icsk_accept_queue.rskq_defer_accept++;
2081 icsk->icsk_accept_queue.rskq_defer_accept++;
1da177e4
LT
2082 }
2083 break;
2084
2085 case TCP_WINDOW_CLAMP:
2086 if (!val) {
2087 if (sk->sk_state != TCP_CLOSE) {
2088 err = -EINVAL;
2089 break;
2090 }
2091 tp->window_clamp = 0;
2092 } else
2093 tp->window_clamp = val < SOCK_MIN_RCVBUF / 2 ?
2094 SOCK_MIN_RCVBUF / 2 : val;
2095 break;
2096
2097 case TCP_QUICKACK:
2098 if (!val) {
463c84b9 2099 icsk->icsk_ack.pingpong = 1;
1da177e4 2100 } else {
463c84b9 2101 icsk->icsk_ack.pingpong = 0;
1da177e4
LT
2102 if ((1 << sk->sk_state) &
2103 (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT) &&
463c84b9
ACM
2104 inet_csk_ack_scheduled(sk)) {
2105 icsk->icsk_ack.pending |= ICSK_ACK_PUSHED;
0e4b4992 2106 tcp_cleanup_rbuf(sk, 1);
1da177e4 2107 if (!(val & 1))
463c84b9 2108 icsk->icsk_ack.pingpong = 1;
1da177e4
LT
2109 }
2110 }
2111 break;
2112
cfb6eeb4
YH
2113#ifdef CONFIG_TCP_MD5SIG
2114 case TCP_MD5SIG:
2115 /* Read the IP->Key mappings from userspace */
2116 err = tp->af_specific->md5_parse(sk, optval, optlen);
2117 break;
2118#endif
2119
1da177e4
LT
2120 default:
2121 err = -ENOPROTOOPT;
2122 break;
3ff50b79
SH
2123 }
2124
1da177e4
LT
2125 release_sock(sk);
2126 return err;
2127}
2128
3fdadf7d
DM
2129int tcp_setsockopt(struct sock *sk, int level, int optname, char __user *optval,
2130 int optlen)
2131{
2132 struct inet_connection_sock *icsk = inet_csk(sk);
2133
2134 if (level != SOL_TCP)
2135 return icsk->icsk_af_ops->setsockopt(sk, level, optname,
2136 optval, optlen);
2137 return do_tcp_setsockopt(sk, level, optname, optval, optlen);
2138}
2139
2140#ifdef CONFIG_COMPAT
543d9cfe
ACM
2141int compat_tcp_setsockopt(struct sock *sk, int level, int optname,
2142 char __user *optval, int optlen)
3fdadf7d 2143{
dec73ff0
ACM
2144 if (level != SOL_TCP)
2145 return inet_csk_compat_setsockopt(sk, level, optname,
2146 optval, optlen);
3fdadf7d
DM
2147 return do_tcp_setsockopt(sk, level, optname, optval, optlen);
2148}
543d9cfe
ACM
2149
2150EXPORT_SYMBOL(compat_tcp_setsockopt);
3fdadf7d
DM
2151#endif
2152
1da177e4
LT
2153/* Return information about state of tcp endpoint in API format. */
2154void tcp_get_info(struct sock *sk, struct tcp_info *info)
2155{
2156 struct tcp_sock *tp = tcp_sk(sk);
463c84b9 2157 const struct inet_connection_sock *icsk = inet_csk(sk);
1da177e4
LT
2158 u32 now = tcp_time_stamp;
2159
2160 memset(info, 0, sizeof(*info));
2161
2162 info->tcpi_state = sk->sk_state;
6687e988 2163 info->tcpi_ca_state = icsk->icsk_ca_state;
463c84b9 2164 info->tcpi_retransmits = icsk->icsk_retransmits;
6687e988 2165 info->tcpi_probes = icsk->icsk_probes_out;
463c84b9 2166 info->tcpi_backoff = icsk->icsk_backoff;
1da177e4
LT
2167
2168 if (tp->rx_opt.tstamp_ok)
2169 info->tcpi_options |= TCPI_OPT_TIMESTAMPS;
e60402d0 2170 if (tcp_is_sack(tp))
1da177e4
LT
2171 info->tcpi_options |= TCPI_OPT_SACK;
2172 if (tp->rx_opt.wscale_ok) {
2173 info->tcpi_options |= TCPI_OPT_WSCALE;
2174 info->tcpi_snd_wscale = tp->rx_opt.snd_wscale;
2175 info->tcpi_rcv_wscale = tp->rx_opt.rcv_wscale;
e905a9ed 2176 }
1da177e4
LT
2177
2178 if (tp->ecn_flags&TCP_ECN_OK)
2179 info->tcpi_options |= TCPI_OPT_ECN;
2180
463c84b9
ACM
2181 info->tcpi_rto = jiffies_to_usecs(icsk->icsk_rto);
2182 info->tcpi_ato = jiffies_to_usecs(icsk->icsk_ack.ato);
c1b4a7e6 2183 info->tcpi_snd_mss = tp->mss_cache;
463c84b9 2184 info->tcpi_rcv_mss = icsk->icsk_ack.rcv_mss;
1da177e4 2185
5ee3afba
RJ
2186 if (sk->sk_state == TCP_LISTEN) {
2187 info->tcpi_unacked = sk->sk_ack_backlog;
2188 info->tcpi_sacked = sk->sk_max_ack_backlog;
2189 } else {
2190 info->tcpi_unacked = tp->packets_out;
2191 info->tcpi_sacked = tp->sacked_out;
2192 }
1da177e4
LT
2193 info->tcpi_lost = tp->lost_out;
2194 info->tcpi_retrans = tp->retrans_out;
2195 info->tcpi_fackets = tp->fackets_out;
2196
2197 info->tcpi_last_data_sent = jiffies_to_msecs(now - tp->lsndtime);
463c84b9 2198 info->tcpi_last_data_recv = jiffies_to_msecs(now - icsk->icsk_ack.lrcvtime);
1da177e4
LT
2199 info->tcpi_last_ack_recv = jiffies_to_msecs(now - tp->rcv_tstamp);
2200
d83d8461 2201 info->tcpi_pmtu = icsk->icsk_pmtu_cookie;
1da177e4
LT
2202 info->tcpi_rcv_ssthresh = tp->rcv_ssthresh;
2203 info->tcpi_rtt = jiffies_to_usecs(tp->srtt)>>3;
2204 info->tcpi_rttvar = jiffies_to_usecs(tp->mdev)>>2;
2205 info->tcpi_snd_ssthresh = tp->snd_ssthresh;
2206 info->tcpi_snd_cwnd = tp->snd_cwnd;
2207 info->tcpi_advmss = tp->advmss;
2208 info->tcpi_reordering = tp->reordering;
2209
2210 info->tcpi_rcv_rtt = jiffies_to_usecs(tp->rcv_rtt_est.rtt)>>3;
2211 info->tcpi_rcv_space = tp->rcvq_space.space;
2212
2213 info->tcpi_total_retrans = tp->total_retrans;
2214}
2215
2216EXPORT_SYMBOL_GPL(tcp_get_info);
2217
3fdadf7d
DM
2218static int do_tcp_getsockopt(struct sock *sk, int level,
2219 int optname, char __user *optval, int __user *optlen)
1da177e4 2220{
295f7324 2221 struct inet_connection_sock *icsk = inet_csk(sk);
1da177e4
LT
2222 struct tcp_sock *tp = tcp_sk(sk);
2223 int val, len;
2224
1da177e4
LT
2225 if (get_user(len, optlen))
2226 return -EFAULT;
2227
2228 len = min_t(unsigned int, len, sizeof(int));
2229
2230 if (len < 0)
2231 return -EINVAL;
2232
2233 switch (optname) {
2234 case TCP_MAXSEG:
c1b4a7e6 2235 val = tp->mss_cache;
1da177e4
LT
2236 if (!val && ((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_LISTEN)))
2237 val = tp->rx_opt.user_mss;
2238 break;
2239 case TCP_NODELAY:
2240 val = !!(tp->nonagle&TCP_NAGLE_OFF);
2241 break;
2242 case TCP_CORK:
2243 val = !!(tp->nonagle&TCP_NAGLE_CORK);
2244 break;
2245 case TCP_KEEPIDLE:
2246 val = (tp->keepalive_time ? : sysctl_tcp_keepalive_time) / HZ;
2247 break;
2248 case TCP_KEEPINTVL:
2249 val = (tp->keepalive_intvl ? : sysctl_tcp_keepalive_intvl) / HZ;
2250 break;
2251 case TCP_KEEPCNT:
2252 val = tp->keepalive_probes ? : sysctl_tcp_keepalive_probes;
2253 break;
2254 case TCP_SYNCNT:
295f7324 2255 val = icsk->icsk_syn_retries ? : sysctl_tcp_syn_retries;
1da177e4
LT
2256 break;
2257 case TCP_LINGER2:
2258 val = tp->linger2;
2259 if (val >= 0)
2260 val = (val ? : sysctl_tcp_fin_timeout) / HZ;
2261 break;
2262 case TCP_DEFER_ACCEPT:
295f7324
ACM
2263 val = !icsk->icsk_accept_queue.rskq_defer_accept ? 0 :
2264 ((TCP_TIMEOUT_INIT / HZ) << (icsk->icsk_accept_queue.rskq_defer_accept - 1));
1da177e4
LT
2265 break;
2266 case TCP_WINDOW_CLAMP:
2267 val = tp->window_clamp;
2268 break;
2269 case TCP_INFO: {
2270 struct tcp_info info;
2271
2272 if (get_user(len, optlen))
2273 return -EFAULT;
2274
2275 tcp_get_info(sk, &info);
2276
2277 len = min_t(unsigned int, len, sizeof(info));
2278 if (put_user(len, optlen))
2279 return -EFAULT;
2280 if (copy_to_user(optval, &info, len))
2281 return -EFAULT;
2282 return 0;
2283 }
2284 case TCP_QUICKACK:
295f7324 2285 val = !icsk->icsk_ack.pingpong;
1da177e4 2286 break;
5f8ef48d
SH
2287
2288 case TCP_CONGESTION:
2289 if (get_user(len, optlen))
2290 return -EFAULT;
2291 len = min_t(unsigned int, len, TCP_CA_NAME_MAX);
2292 if (put_user(len, optlen))
2293 return -EFAULT;
6687e988 2294 if (copy_to_user(optval, icsk->icsk_ca_ops->name, len))
5f8ef48d
SH
2295 return -EFAULT;
2296 return 0;
1da177e4
LT
2297 default:
2298 return -ENOPROTOOPT;
3ff50b79 2299 }
1da177e4
LT
2300
2301 if (put_user(len, optlen))
2302 return -EFAULT;
2303 if (copy_to_user(optval, &val, len))
2304 return -EFAULT;
2305 return 0;
2306}
2307
3fdadf7d
DM
2308int tcp_getsockopt(struct sock *sk, int level, int optname, char __user *optval,
2309 int __user *optlen)
2310{
2311 struct inet_connection_sock *icsk = inet_csk(sk);
2312
2313 if (level != SOL_TCP)
2314 return icsk->icsk_af_ops->getsockopt(sk, level, optname,
2315 optval, optlen);
2316 return do_tcp_getsockopt(sk, level, optname, optval, optlen);
2317}
2318
2319#ifdef CONFIG_COMPAT
543d9cfe
ACM
2320int compat_tcp_getsockopt(struct sock *sk, int level, int optname,
2321 char __user *optval, int __user *optlen)
3fdadf7d 2322{
dec73ff0
ACM
2323 if (level != SOL_TCP)
2324 return inet_csk_compat_getsockopt(sk, level, optname,
2325 optval, optlen);
3fdadf7d
DM
2326 return do_tcp_getsockopt(sk, level, optname, optval, optlen);
2327}
543d9cfe
ACM
2328
2329EXPORT_SYMBOL(compat_tcp_getsockopt);
3fdadf7d 2330#endif
1da177e4 2331
576a30eb 2332struct sk_buff *tcp_tso_segment(struct sk_buff *skb, int features)
f4c50d99
HX
2333{
2334 struct sk_buff *segs = ERR_PTR(-EINVAL);
2335 struct tcphdr *th;
2336 unsigned thlen;
2337 unsigned int seq;
d3bc23e7 2338 __be32 delta;
f4c50d99
HX
2339 unsigned int oldlen;
2340 unsigned int len;
2341
2342 if (!pskb_may_pull(skb, sizeof(*th)))
2343 goto out;
2344
aa8223c7 2345 th = tcp_hdr(skb);
f4c50d99
HX
2346 thlen = th->doff * 4;
2347 if (thlen < sizeof(*th))
2348 goto out;
2349
2350 if (!pskb_may_pull(skb, thlen))
2351 goto out;
2352
0718bcc0 2353 oldlen = (u16)~skb->len;
f4c50d99
HX
2354 __skb_pull(skb, thlen);
2355
3820c3f3
HX
2356 if (skb_gso_ok(skb, features | NETIF_F_GSO_ROBUST)) {
2357 /* Packet is from an untrusted source, reset gso_segs. */
bbcf467d
HX
2358 int type = skb_shinfo(skb)->gso_type;
2359 int mss;
2360
2361 if (unlikely(type &
2362 ~(SKB_GSO_TCPV4 |
2363 SKB_GSO_DODGY |
2364 SKB_GSO_TCP_ECN |
2365 SKB_GSO_TCPV6 |
2366 0) ||
2367 !(type & (SKB_GSO_TCPV4 | SKB_GSO_TCPV6))))
2368 goto out;
3820c3f3 2369
bbcf467d 2370 mss = skb_shinfo(skb)->gso_size;
172589cc 2371 skb_shinfo(skb)->gso_segs = DIV_ROUND_UP(skb->len, mss);
3820c3f3
HX
2372
2373 segs = NULL;
2374 goto out;
2375 }
2376
576a30eb 2377 segs = skb_segment(skb, features);
f4c50d99
HX
2378 if (IS_ERR(segs))
2379 goto out;
2380
2381 len = skb_shinfo(skb)->gso_size;
0718bcc0 2382 delta = htonl(oldlen + (thlen + len));
f4c50d99
HX
2383
2384 skb = segs;
aa8223c7 2385 th = tcp_hdr(skb);
f4c50d99
HX
2386 seq = ntohl(th->seq);
2387
2388 do {
2389 th->fin = th->psh = 0;
2390
d3bc23e7
AV
2391 th->check = ~csum_fold((__force __wsum)((__force u32)th->check +
2392 (__force u32)delta));
84fa7933 2393 if (skb->ip_summed != CHECKSUM_PARTIAL)
9c70220b
ACM
2394 th->check =
2395 csum_fold(csum_partial(skb_transport_header(skb),
2396 thlen, skb->csum));
f4c50d99
HX
2397
2398 seq += len;
2399 skb = skb->next;
aa8223c7 2400 th = tcp_hdr(skb);
f4c50d99
HX
2401
2402 th->seq = htonl(seq);
2403 th->cwr = 0;
2404 } while (skb->next);
2405
27a884dc 2406 delta = htonl(oldlen + (skb->tail - skb->transport_header) +
9c70220b 2407 skb->data_len);
d3bc23e7
AV
2408 th->check = ~csum_fold((__force __wsum)((__force u32)th->check +
2409 (__force u32)delta));
84fa7933 2410 if (skb->ip_summed != CHECKSUM_PARTIAL)
9c70220b
ACM
2411 th->check = csum_fold(csum_partial(skb_transport_header(skb),
2412 thlen, skb->csum));
f4c50d99
HX
2413
2414out:
2415 return segs;
2416}
adcfc7d0 2417EXPORT_SYMBOL(tcp_tso_segment);
f4c50d99 2418
cfb6eeb4
YH
2419#ifdef CONFIG_TCP_MD5SIG
2420static unsigned long tcp_md5sig_users;
2421static struct tcp_md5sig_pool **tcp_md5sig_pool;
2422static DEFINE_SPINLOCK(tcp_md5sig_pool_lock);
2423
2424static void __tcp_free_md5sig_pool(struct tcp_md5sig_pool **pool)
2425{
2426 int cpu;
2427 for_each_possible_cpu(cpu) {
2428 struct tcp_md5sig_pool *p = *per_cpu_ptr(pool, cpu);
2429 if (p) {
2430 if (p->md5_desc.tfm)
2431 crypto_free_hash(p->md5_desc.tfm);
2432 kfree(p);
2433 p = NULL;
2434 }
2435 }
2436 free_percpu(pool);
2437}
2438
2439void tcp_free_md5sig_pool(void)
2440{
2441 struct tcp_md5sig_pool **pool = NULL;
2442
2c4f6219 2443 spin_lock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2444 if (--tcp_md5sig_users == 0) {
2445 pool = tcp_md5sig_pool;
2446 tcp_md5sig_pool = NULL;
2447 }
2c4f6219 2448 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2449 if (pool)
2450 __tcp_free_md5sig_pool(pool);
2451}
2452
2453EXPORT_SYMBOL(tcp_free_md5sig_pool);
2454
f5b99bcd 2455static struct tcp_md5sig_pool **__tcp_alloc_md5sig_pool(void)
cfb6eeb4
YH
2456{
2457 int cpu;
2458 struct tcp_md5sig_pool **pool;
2459
2460 pool = alloc_percpu(struct tcp_md5sig_pool *);
2461 if (!pool)
2462 return NULL;
2463
2464 for_each_possible_cpu(cpu) {
2465 struct tcp_md5sig_pool *p;
2466 struct crypto_hash *hash;
2467
2468 p = kzalloc(sizeof(*p), GFP_KERNEL);
2469 if (!p)
2470 goto out_free;
2471 *per_cpu_ptr(pool, cpu) = p;
2472
2473 hash = crypto_alloc_hash("md5", 0, CRYPTO_ALG_ASYNC);
2474 if (!hash || IS_ERR(hash))
2475 goto out_free;
2476
2477 p->md5_desc.tfm = hash;
2478 }
2479 return pool;
2480out_free:
2481 __tcp_free_md5sig_pool(pool);
2482 return NULL;
2483}
2484
2485struct tcp_md5sig_pool **tcp_alloc_md5sig_pool(void)
2486{
2487 struct tcp_md5sig_pool **pool;
2488 int alloc = 0;
2489
2490retry:
2c4f6219 2491 spin_lock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2492 pool = tcp_md5sig_pool;
2493 if (tcp_md5sig_users++ == 0) {
2494 alloc = 1;
2c4f6219 2495 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2496 } else if (!pool) {
2497 tcp_md5sig_users--;
2c4f6219 2498 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2499 cpu_relax();
2500 goto retry;
2501 } else
2c4f6219 2502 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2503
2504 if (alloc) {
2505 /* we cannot hold spinlock here because this may sleep. */
2506 struct tcp_md5sig_pool **p = __tcp_alloc_md5sig_pool();
2c4f6219 2507 spin_lock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2508 if (!p) {
2509 tcp_md5sig_users--;
2c4f6219 2510 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2511 return NULL;
2512 }
2513 pool = tcp_md5sig_pool;
2514 if (pool) {
2515 /* oops, it has already been assigned. */
2c4f6219 2516 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2517 __tcp_free_md5sig_pool(p);
2518 } else {
2519 tcp_md5sig_pool = pool = p;
2c4f6219 2520 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2521 }
2522 }
2523 return pool;
2524}
2525
2526EXPORT_SYMBOL(tcp_alloc_md5sig_pool);
2527
2528struct tcp_md5sig_pool *__tcp_get_md5sig_pool(int cpu)
2529{
2530 struct tcp_md5sig_pool **p;
2c4f6219 2531 spin_lock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2532 p = tcp_md5sig_pool;
2533 if (p)
2534 tcp_md5sig_users++;
2c4f6219 2535 spin_unlock_bh(&tcp_md5sig_pool_lock);
cfb6eeb4
YH
2536 return (p ? *per_cpu_ptr(p, cpu) : NULL);
2537}
2538
2539EXPORT_SYMBOL(__tcp_get_md5sig_pool);
2540
6931ba7c
DM
2541void __tcp_put_md5sig_pool(void)
2542{
2543 tcp_free_md5sig_pool();
cfb6eeb4
YH
2544}
2545
2546EXPORT_SYMBOL(__tcp_put_md5sig_pool);
2547#endif
2548
4ac02bab
AK
2549void tcp_done(struct sock *sk)
2550{
2551 if(sk->sk_state == TCP_SYN_SENT || sk->sk_state == TCP_SYN_RECV)
2552 TCP_INC_STATS_BH(TCP_MIB_ATTEMPTFAILS);
2553
2554 tcp_set_state(sk, TCP_CLOSE);
2555 tcp_clear_xmit_timers(sk);
2556
2557 sk->sk_shutdown = SHUTDOWN_MASK;
2558
2559 if (!sock_flag(sk, SOCK_DEAD))
2560 sk->sk_state_change(sk);
2561 else
2562 inet_csk_destroy_sock(sk);
2563}
2564EXPORT_SYMBOL_GPL(tcp_done);
2565
5f8ef48d 2566extern struct tcp_congestion_ops tcp_reno;
1da177e4
LT
2567
2568static __initdata unsigned long thash_entries;
2569static int __init set_thash_entries(char *str)
2570{
2571 if (!str)
2572 return 0;
2573 thash_entries = simple_strtoul(str, &str, 0);
2574 return 1;
2575}
2576__setup("thash_entries=", set_thash_entries);
2577
2578void __init tcp_init(void)
2579{
2580 struct sk_buff *skb = NULL;
7b4f4b5e
JH
2581 unsigned long limit;
2582 int order, i, max_share;
1da177e4 2583
1f9e636e 2584 BUILD_BUG_ON(sizeof(struct tcp_skb_cb) > sizeof(skb->cb));
1da177e4 2585
6e04e021
ACM
2586 tcp_hashinfo.bind_bucket_cachep =
2587 kmem_cache_create("tcp_bind_bucket",
2588 sizeof(struct inet_bind_bucket), 0,
20c2df83 2589 SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL);
1da177e4 2590
1da177e4
LT
2591 /* Size and allocate the main established and bind bucket
2592 * hash tables.
2593 *
2594 * The methodology is similar to that of the buffer cache.
2595 */
6e04e021 2596 tcp_hashinfo.ehash =
1da177e4 2597 alloc_large_system_hash("TCP established",
0f7ff927 2598 sizeof(struct inet_ehash_bucket),
1da177e4
LT
2599 thash_entries,
2600 (num_physpages >= 128 * 1024) ?
18955cfc 2601 13 : 15,
9e950efa 2602 0,
6e04e021 2603 &tcp_hashinfo.ehash_size,
1da177e4 2604 NULL,
0ccfe618 2605 thash_entries ? 0 : 512 * 1024);
dbca9b27
ED
2606 tcp_hashinfo.ehash_size = 1 << tcp_hashinfo.ehash_size;
2607 for (i = 0; i < tcp_hashinfo.ehash_size; i++) {
6e04e021 2608 INIT_HLIST_HEAD(&tcp_hashinfo.ehash[i].chain);
dbca9b27 2609 INIT_HLIST_HEAD(&tcp_hashinfo.ehash[i].twchain);
1da177e4 2610 }
230140cf
ED
2611 if (inet_ehash_locks_alloc(&tcp_hashinfo))
2612 panic("TCP: failed to alloc ehash_locks");
6e04e021 2613 tcp_hashinfo.bhash =
1da177e4 2614 alloc_large_system_hash("TCP bind",
0f7ff927 2615 sizeof(struct inet_bind_hashbucket),
6e04e021 2616 tcp_hashinfo.ehash_size,
1da177e4 2617 (num_physpages >= 128 * 1024) ?
18955cfc 2618 13 : 15,
9e950efa 2619 0,
6e04e021 2620 &tcp_hashinfo.bhash_size,
1da177e4
LT
2621 NULL,
2622 64 * 1024);
6e04e021
ACM
2623 tcp_hashinfo.bhash_size = 1 << tcp_hashinfo.bhash_size;
2624 for (i = 0; i < tcp_hashinfo.bhash_size; i++) {
2625 spin_lock_init(&tcp_hashinfo.bhash[i].lock);
2626 INIT_HLIST_HEAD(&tcp_hashinfo.bhash[i].chain);
1da177e4
LT
2627 }
2628
2629 /* Try to be a bit smarter and adjust defaults depending
2630 * on available memory.
2631 */
2632 for (order = 0; ((1 << order) << PAGE_SHIFT) <
6e04e021 2633 (tcp_hashinfo.bhash_size * sizeof(struct inet_bind_hashbucket));
1da177e4
LT
2634 order++)
2635 ;
e7626486 2636 if (order >= 4) {
295ff7ed 2637 tcp_death_row.sysctl_max_tw_buckets = 180000;
1da177e4
LT
2638 sysctl_tcp_max_orphans = 4096 << (order - 4);
2639 sysctl_max_syn_backlog = 1024;
2640 } else if (order < 3) {
295ff7ed 2641 tcp_death_row.sysctl_max_tw_buckets >>= (3 - order);
1da177e4
LT
2642 sysctl_tcp_max_orphans >>= (3 - order);
2643 sysctl_max_syn_backlog = 128;
2644 }
1da177e4 2645
53cdcc04
JH
2646 /* Set the pressure threshold to be a fraction of global memory that
2647 * is up to 1/2 at 256 MB, decreasing toward zero with the amount of
2648 * memory, with a floor of 128 pages.
2649 */
2650 limit = min(nr_all_pages, 1UL<<(28-PAGE_SHIFT)) >> (20-PAGE_SHIFT);
2651 limit = (limit * (nr_all_pages >> (20-PAGE_SHIFT))) >> (PAGE_SHIFT-11);
2652 limit = max(limit, 128UL);
2653 sysctl_tcp_mem[0] = limit / 4 * 3;
2654 sysctl_tcp_mem[1] = limit;
52bf376c 2655 sysctl_tcp_mem[2] = sysctl_tcp_mem[0] * 2;
1da177e4 2656
53cdcc04 2657 /* Set per-socket limits to no more than 1/128 the pressure threshold */
7b4f4b5e
JH
2658 limit = ((unsigned long)sysctl_tcp_mem[1]) << (PAGE_SHIFT - 7);
2659 max_share = min(4UL*1024*1024, limit);
2660
3ab224be 2661 sysctl_tcp_wmem[0] = SK_MEM_QUANTUM;
7b4f4b5e
JH
2662 sysctl_tcp_wmem[1] = 16*1024;
2663 sysctl_tcp_wmem[2] = max(64*1024, max_share);
2664
3ab224be 2665 sysctl_tcp_rmem[0] = SK_MEM_QUANTUM;
7b4f4b5e
JH
2666 sysctl_tcp_rmem[1] = 87380;
2667 sysctl_tcp_rmem[2] = max(87380, max_share);
1da177e4
LT
2668
2669 printk(KERN_INFO "TCP: Hash tables configured "
2670 "(established %d bind %d)\n",
dbca9b27 2671 tcp_hashinfo.ehash_size, tcp_hashinfo.bhash_size);
317a76f9
SH
2672
2673 tcp_register_congestion_control(&tcp_reno);
1da177e4
LT
2674}
2675
1da177e4 2676EXPORT_SYMBOL(tcp_close);
1da177e4
LT
2677EXPORT_SYMBOL(tcp_disconnect);
2678EXPORT_SYMBOL(tcp_getsockopt);
2679EXPORT_SYMBOL(tcp_ioctl);
1da177e4
LT
2680EXPORT_SYMBOL(tcp_poll);
2681EXPORT_SYMBOL(tcp_read_sock);
2682EXPORT_SYMBOL(tcp_recvmsg);
2683EXPORT_SYMBOL(tcp_sendmsg);
9c55e01c 2684EXPORT_SYMBOL(tcp_splice_read);
1da177e4
LT
2685EXPORT_SYMBOL(tcp_sendpage);
2686EXPORT_SYMBOL(tcp_setsockopt);
2687EXPORT_SYMBOL(tcp_shutdown);
2688EXPORT_SYMBOL(tcp_statistics);