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