tcp_ipv4.c 66.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
/*
 * INET		An implementation of the TCP/IP protocol suite for the LINUX
 *		operating system.  INET is implemented using the  BSD Socket
 *		interface as the means of communication with the user level.
 *
 *		Implementation of the Transmission Control Protocol(TCP).
 *
 *		IPv4 specific functions
 *
 *
 *		code split from:
 *		linux/ipv4/tcp.c
 *		linux/ipv4/tcp_input.c
 *		linux/ipv4/tcp_output.c
 *
 *		See tcp.c for author information
 *
 *	This program is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU General Public License
 *      as published by the Free Software Foundation; either version
 *      2 of the License, or (at your option) any later version.
 */

/*
 * Changes:
 *		David S. Miller	:	New socket lookup architecture.
 *					This code is dedicated to John Dyson.
 *		David S. Miller :	Change semantics of established hash,
 *					half is devoted to TIME_WAIT sockets
 *					and the rest go in the other half.
 *		Andi Kleen :		Add support for syncookies and fixed
 *					some bugs: ip options weren't passed to
 *					the TCP layer, missed a check for an
 *					ACK bit.
 *		Andi Kleen :		Implemented fast path mtu discovery.
 *	     				Fixed many serious bugs in the
37
 *					request_sock handling and moved
L
Linus Torvalds 已提交
38 39
 *					most of it into the af independent code.
 *					Added tail drop and some other bugfixes.
S
Stephen Hemminger 已提交
40
 *					Added new listen semantics.
L
Linus Torvalds 已提交
41 42 43 44 45 46 47 48 49 50 51 52 53
 *		Mike McLagan	:	Routing by source
 *	Juan Jose Ciarlante:		ip_dynaddr bits
 *		Andi Kleen:		various fixes.
 *	Vitaly E. Lavrov	:	Transparent proxy revived after year
 *					coma.
 *	Andi Kleen		:	Fix new listen.
 *	Andi Kleen		:	Fix accept error reporting.
 *	YOSHIFUJI Hideaki @USAGI and:	Support IPV6_V6ONLY socket option, which
 *	Alexey Kuznetsov		allow both IPv4 and IPv6 sockets to bind
 *					a single port at the same time.
 */


H
Herbert Xu 已提交
54
#include <linux/bottom_half.h>
L
Linus Torvalds 已提交
55 56 57 58 59 60 61 62
#include <linux/types.h>
#include <linux/fcntl.h>
#include <linux/module.h>
#include <linux/random.h>
#include <linux/cache.h>
#include <linux/jhash.h>
#include <linux/init.h>
#include <linux/times.h>
63
#include <linux/slab.h>
L
Linus Torvalds 已提交
64

65
#include <net/net_namespace.h>
L
Linus Torvalds 已提交
66
#include <net/icmp.h>
67
#include <net/inet_hashtables.h>
L
Linus Torvalds 已提交
68
#include <net/tcp.h>
69
#include <net/transp_v6.h>
L
Linus Torvalds 已提交
70 71
#include <net/ipv6.h>
#include <net/inet_common.h>
72
#include <net/timewait_sock.h>
L
Linus Torvalds 已提交
73
#include <net/xfrm.h>
C
Chris Leech 已提交
74
#include <net/netdma.h>
75
#include <net/secure_seq.h>
L
Linus Torvalds 已提交
76 77 78 79 80 81 82

#include <linux/inet.h>
#include <linux/ipv6.h>
#include <linux/stddef.h>
#include <linux/proc_fs.h>
#include <linux/seq_file.h>

83 84 85
#include <linux/crypto.h>
#include <linux/scatterlist.h>

86 87
int sysctl_tcp_tw_reuse __read_mostly;
int sysctl_tcp_low_latency __read_mostly;
E
Eric Dumazet 已提交
88
EXPORT_SYMBOL(sysctl_tcp_low_latency);
L
Linus Torvalds 已提交
89 90


91
#ifdef CONFIG_TCP_MD5SIG
92 93
static struct tcp_md5sig_key *tcp_v4_md5_do_lookup(struct sock *sk,
						   __be32 addr);
94 95
static int tcp_v4_md5_hash_hdr(char *md5_hash, struct tcp_md5sig_key *key,
			       __be32 daddr, __be32 saddr, struct tcphdr *th);
96 97 98 99 100 101
#else
static inline
struct tcp_md5sig_key *tcp_v4_md5_do_lookup(struct sock *sk, __be32 addr)
{
	return NULL;
}
102 103
#endif

104
struct inet_hashinfo tcp_hashinfo;
E
Eric Dumazet 已提交
105
EXPORT_SYMBOL(tcp_hashinfo);
L
Linus Torvalds 已提交
106

107
static inline __u32 tcp_v4_init_sequence(const struct sk_buff *skb)
L
Linus Torvalds 已提交
108
{
109 110
	return secure_tcp_sequence_number(ip_hdr(skb)->daddr,
					  ip_hdr(skb)->saddr,
111 112
					  tcp_hdr(skb)->dest,
					  tcp_hdr(skb)->source);
L
Linus Torvalds 已提交
113 114
}

115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
int tcp_twsk_unique(struct sock *sk, struct sock *sktw, void *twp)
{
	const struct tcp_timewait_sock *tcptw = tcp_twsk(sktw);
	struct tcp_sock *tp = tcp_sk(sk);

	/* With PAWS, it is safe from the viewpoint
	   of data integrity. Even without PAWS it is safe provided sequence
	   spaces do not overlap i.e. at data rates <= 80Mbit/sec.

	   Actually, the idea is close to VJ's one, only timestamp cache is
	   held not per host, but per port pair and TW bucket is used as state
	   holder.

	   If TW bucket has been already destroyed we fall back to VJ's scheme
	   and use initial timestamp retrieved from peer table.
	 */
	if (tcptw->tw_ts_recent_stamp &&
	    (twp == NULL || (sysctl_tcp_tw_reuse &&
133
			     get_seconds() - tcptw->tw_ts_recent_stamp > 1))) {
134 135 136 137 138 139 140 141 142 143 144 145 146
		tp->write_seq = tcptw->tw_snd_nxt + 65535 + 2;
		if (tp->write_seq == 0)
			tp->write_seq = 1;
		tp->rx_opt.ts_recent	   = tcptw->tw_ts_recent;
		tp->rx_opt.ts_recent_stamp = tcptw->tw_ts_recent_stamp;
		sock_hold(sktw);
		return 1;
	}

	return 0;
}
EXPORT_SYMBOL_GPL(tcp_twsk_unique);

L
Linus Torvalds 已提交
147 148 149
/* This will initiate an outgoing connection. */
int tcp_v4_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
{
150
	struct sockaddr_in *usin = (struct sockaddr_in *)uaddr;
L
Linus Torvalds 已提交
151 152
	struct inet_sock *inet = inet_sk(sk);
	struct tcp_sock *tp = tcp_sk(sk);
153
	__be16 orig_sport, orig_dport;
154
	__be32 daddr, nexthop;
155
	struct flowi4 *fl4;
156
	struct rtable *rt;
L
Linus Torvalds 已提交
157
	int err;
158
	struct ip_options_rcu *inet_opt;
L
Linus Torvalds 已提交
159 160 161 162 163 164 165 166

	if (addr_len < sizeof(struct sockaddr_in))
		return -EINVAL;

	if (usin->sin_family != AF_INET)
		return -EAFNOSUPPORT;

	nexthop = daddr = usin->sin_addr.s_addr;
167 168 169
	inet_opt = rcu_dereference_protected(inet->inet_opt,
					     sock_owned_by_user(sk));
	if (inet_opt && inet_opt->opt.srr) {
L
Linus Torvalds 已提交
170 171
		if (!daddr)
			return -EINVAL;
172
		nexthop = inet_opt->opt.faddr;
L
Linus Torvalds 已提交
173 174
	}

175 176
	orig_sport = inet->inet_sport;
	orig_dport = usin->sin_port;
177 178
	fl4 = &inet->cork.fl.u.ip4;
	rt = ip_route_connect(fl4, nexthop, inet->inet_saddr,
179 180 181 182 183 184
			      RT_CONN_FLAGS(sk), sk->sk_bound_dev_if,
			      IPPROTO_TCP,
			      orig_sport, orig_dport, sk, true);
	if (IS_ERR(rt)) {
		err = PTR_ERR(rt);
		if (err == -ENETUNREACH)
185
			IP_INC_STATS_BH(sock_net(sk), IPSTATS_MIB_OUTNOROUTES);
186
		return err;
187
	}
L
Linus Torvalds 已提交
188 189 190 191 192 193

	if (rt->rt_flags & (RTCF_MULTICAST | RTCF_BROADCAST)) {
		ip_rt_put(rt);
		return -ENETUNREACH;
	}

194
	if (!inet_opt || !inet_opt->opt.srr)
195
		daddr = fl4->daddr;
L
Linus Torvalds 已提交
196

E
Eric Dumazet 已提交
197
	if (!inet->inet_saddr)
198
		inet->inet_saddr = fl4->saddr;
E
Eric Dumazet 已提交
199
	inet->inet_rcv_saddr = inet->inet_saddr;
L
Linus Torvalds 已提交
200

E
Eric Dumazet 已提交
201
	if (tp->rx_opt.ts_recent_stamp && inet->inet_daddr != daddr) {
L
Linus Torvalds 已提交
202 203 204 205 206 207
		/* Reset inherited state */
		tp->rx_opt.ts_recent	   = 0;
		tp->rx_opt.ts_recent_stamp = 0;
		tp->write_seq		   = 0;
	}

208
	if (tcp_death_row.sysctl_tw_recycle &&
209
	    !tp->rx_opt.ts_recent_stamp && fl4->daddr == daddr) {
210
		struct inet_peer *peer = rt_get_peer(rt, fl4->daddr);
211 212 213 214 215
		/*
		 * VJ's idea. We save last timestamp seen from
		 * the destination in peer table, when entering state
		 * TIME-WAIT * and initialize rx_opt.ts_recent from it,
		 * when trying new connection.
L
Linus Torvalds 已提交
216
		 */
217 218 219 220 221 222
		if (peer) {
			inet_peer_refcheck(peer);
			if ((u32)get_seconds() - peer->tcp_ts_stamp <= TCP_PAWS_MSL) {
				tp->rx_opt.ts_recent_stamp = peer->tcp_ts_stamp;
				tp->rx_opt.ts_recent = peer->tcp_ts;
			}
L
Linus Torvalds 已提交
223 224 225
		}
	}

E
Eric Dumazet 已提交
226 227
	inet->inet_dport = usin->sin_port;
	inet->inet_daddr = daddr;
L
Linus Torvalds 已提交
228

229
	inet_csk(sk)->icsk_ext_hdr_len = 0;
230 231
	if (inet_opt)
		inet_csk(sk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
L
Linus Torvalds 已提交
232

233
	tp->rx_opt.mss_clamp = TCP_MSS_DEFAULT;
L
Linus Torvalds 已提交
234 235 236 237 238 239 240

	/* Socket identity is still unknown (sport may be zero).
	 * However we set state to SYN-SENT and not releasing socket
	 * lock select source port, enter ourselves into the hash tables and
	 * complete initialization after this.
	 */
	tcp_set_state(sk, TCP_SYN_SENT);
241
	err = inet_hash_connect(&tcp_death_row, sk);
L
Linus Torvalds 已提交
242 243 244
	if (err)
		goto failure;

245
	rt = ip_route_newports(fl4, rt, orig_sport, orig_dport,
246 247 248 249
			       inet->inet_sport, inet->inet_dport, sk);
	if (IS_ERR(rt)) {
		err = PTR_ERR(rt);
		rt = NULL;
L
Linus Torvalds 已提交
250
		goto failure;
251
	}
L
Linus Torvalds 已提交
252
	/* OK, now commit destination to socket.  */
253
	sk->sk_gso_type = SKB_GSO_TCPV4;
254
	sk_setup_caps(sk, &rt->dst);
L
Linus Torvalds 已提交
255 256

	if (!tp->write_seq)
E
Eric Dumazet 已提交
257 258 259
		tp->write_seq = secure_tcp_sequence_number(inet->inet_saddr,
							   inet->inet_daddr,
							   inet->inet_sport,
L
Linus Torvalds 已提交
260 261
							   usin->sin_port);

E
Eric Dumazet 已提交
262
	inet->inet_id = tp->write_seq ^ jiffies;
L
Linus Torvalds 已提交
263 264 265 266 267 268 269 270 271

	err = tcp_connect(sk);
	rt = NULL;
	if (err)
		goto failure;

	return 0;

failure:
272 273 274 275
	/*
	 * This unhashes the socket and releases the local port,
	 * if necessary.
	 */
L
Linus Torvalds 已提交
276 277 278
	tcp_set_state(sk, TCP_CLOSE);
	ip_rt_put(rt);
	sk->sk_route_caps = 0;
E
Eric Dumazet 已提交
279
	inet->inet_dport = 0;
L
Linus Torvalds 已提交
280 281
	return err;
}
E
Eric Dumazet 已提交
282
EXPORT_SYMBOL(tcp_v4_connect);
L
Linus Torvalds 已提交
283 284 285 286

/*
 * This routine does path mtu discovery as defined in RFC1191.
 */
287
static void do_pmtu_discovery(struct sock *sk, const struct iphdr *iph, u32 mtu)
L
Linus Torvalds 已提交
288 289 290 291 292 293 294 295 296 297 298 299 300 301
{
	struct dst_entry *dst;
	struct inet_sock *inet = inet_sk(sk);

	/* We are not interested in TCP_LISTEN and open_requests (SYN-ACKs
	 * send out by Linux are always <576bytes so they should go through
	 * unfragmented).
	 */
	if (sk->sk_state == TCP_LISTEN)
		return;

	/* We don't check in the destentry if pmtu discovery is forbidden
	 * on this route. We just assume that no packet_to_big packets
	 * are send back when pmtu discovery is not active.
302
	 * There is a small race when the user changes this flag in the
L
Linus Torvalds 已提交
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318
	 * route, but I think that's acceptable.
	 */
	if ((dst = __sk_dst_check(sk, 0)) == NULL)
		return;

	dst->ops->update_pmtu(dst, mtu);

	/* Something is about to be wrong... Remember soft error
	 * for the case, if this connection will not able to recover.
	 */
	if (mtu < dst_mtu(dst) && ip_dont_fragment(sk, dst))
		sk->sk_err_soft = EMSGSIZE;

	mtu = dst_mtu(dst);

	if (inet->pmtudisc != IP_PMTUDISC_DONT &&
319
	    inet_csk(sk)->icsk_pmtu_cookie > mtu) {
L
Linus Torvalds 已提交
320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346
		tcp_sync_mss(sk, mtu);

		/* Resend the TCP packet because it's
		 * clear that the old packet has been
		 * dropped. This is the new "fast" path mtu
		 * discovery.
		 */
		tcp_simple_retransmit(sk);
	} /* else let the usual retransmit timer handle it */
}

/*
 * This routine is called by the ICMP module when it gets some
 * sort of error condition.  If err < 0 then the socket should
 * be closed and the error returned to the user.  If err > 0
 * it's just the icmp type << 8 | icmp code.  After adjustment
 * header points to the first 8 bytes of the tcp header.  We need
 * to find the appropriate port.
 *
 * The locking strategy used here is very "optimistic". When
 * someone else accesses the socket the ICMP is just dropped
 * and for some paths there is no check at all.
 * A more general error queue to queue errors for later handling
 * is probably better.
 *
 */

347
void tcp_v4_err(struct sk_buff *icmp_skb, u32 info)
L
Linus Torvalds 已提交
348
{
349
	const struct iphdr *iph = (const struct iphdr *)icmp_skb->data;
350
	struct tcphdr *th = (struct tcphdr *)(icmp_skb->data + (iph->ihl << 2));
351
	struct inet_connection_sock *icsk;
L
Linus Torvalds 已提交
352 353
	struct tcp_sock *tp;
	struct inet_sock *inet;
354 355
	const int type = icmp_hdr(icmp_skb)->type;
	const int code = icmp_hdr(icmp_skb)->code;
L
Linus Torvalds 已提交
356
	struct sock *sk;
357
	struct sk_buff *skb;
L
Linus Torvalds 已提交
358
	__u32 seq;
359
	__u32 remaining;
L
Linus Torvalds 已提交
360
	int err;
361
	struct net *net = dev_net(icmp_skb->dev);
L
Linus Torvalds 已提交
362

363
	if (icmp_skb->len < (iph->ihl << 2) + 8) {
364
		ICMP_INC_STATS_BH(net, ICMP_MIB_INERRORS);
L
Linus Torvalds 已提交
365 366 367
		return;
	}

368
	sk = inet_lookup(net, &tcp_hashinfo, iph->daddr, th->dest,
369
			iph->saddr, th->source, inet_iif(icmp_skb));
L
Linus Torvalds 已提交
370
	if (!sk) {
371
		ICMP_INC_STATS_BH(net, ICMP_MIB_INERRORS);
L
Linus Torvalds 已提交
372 373 374
		return;
	}
	if (sk->sk_state == TCP_TIME_WAIT) {
375
		inet_twsk_put(inet_twsk(sk));
L
Linus Torvalds 已提交
376 377 378 379 380 381 382 383
		return;
	}

	bh_lock_sock(sk);
	/* If too many ICMPs get dropped on busy
	 * servers this needs to be solved differently.
	 */
	if (sock_owned_by_user(sk))
384
		NET_INC_STATS_BH(net, LINUX_MIB_LOCKDROPPEDICMPS);
L
Linus Torvalds 已提交
385 386 387 388

	if (sk->sk_state == TCP_CLOSE)
		goto out;

389 390 391 392 393
	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
		goto out;
	}

394
	icsk = inet_csk(sk);
L
Linus Torvalds 已提交
395 396 397 398
	tp = tcp_sk(sk);
	seq = ntohl(th->seq);
	if (sk->sk_state != TCP_LISTEN &&
	    !between(seq, tp->snd_una, tp->snd_nxt)) {
399
		NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
L
Linus Torvalds 已提交
400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420
		goto out;
	}

	switch (type) {
	case ICMP_SOURCE_QUENCH:
		/* Just silently ignore these. */
		goto out;
	case ICMP_PARAMETERPROB:
		err = EPROTO;
		break;
	case ICMP_DEST_UNREACH:
		if (code > NR_ICMP_UNREACH)
			goto out;

		if (code == ICMP_FRAG_NEEDED) { /* PMTU discovery (RFC1191) */
			if (!sock_owned_by_user(sk))
				do_pmtu_discovery(sk, iph, info);
			goto out;
		}

		err = icmp_err_convert[code].errno;
421 422 423 424 425 426 427 428
		/* check if icmp_skb allows revert of backoff
		 * (see draft-zimmermann-tcp-lcd) */
		if (code != ICMP_NET_UNREACH && code != ICMP_HOST_UNREACH)
			break;
		if (seq != tp->snd_una  || !icsk->icsk_retransmits ||
		    !icsk->icsk_backoff)
			break;

429 430 431
		if (sock_owned_by_user(sk))
			break;

432
		icsk->icsk_backoff--;
433 434
		inet_csk(sk)->icsk_rto = (tp->srtt ? __tcp_set_rto(tp) :
			TCP_TIMEOUT_INIT) << icsk->icsk_backoff;
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451
		tcp_bound_rto(sk);

		skb = tcp_write_queue_head(sk);
		BUG_ON(!skb);

		remaining = icsk->icsk_rto - min(icsk->icsk_rto,
				tcp_time_stamp - TCP_SKB_CB(skb)->when);

		if (remaining) {
			inet_csk_reset_xmit_timer(sk, ICSK_TIME_RETRANS,
						  remaining, TCP_RTO_MAX);
		} else {
			/* RTO revert clocked out retransmission.
			 * Will retransmit now */
			tcp_retransmit_timer(sk);
		}

L
Linus Torvalds 已提交
452 453 454 455 456 457 458 459 460
		break;
	case ICMP_TIME_EXCEEDED:
		err = EHOSTUNREACH;
		break;
	default:
		goto out;
	}

	switch (sk->sk_state) {
461
		struct request_sock *req, **prev;
L
Linus Torvalds 已提交
462 463 464 465
	case TCP_LISTEN:
		if (sock_owned_by_user(sk))
			goto out;

466 467
		req = inet_csk_search_req(sk, &prev, th->dest,
					  iph->daddr, iph->saddr);
L
Linus Torvalds 已提交
468 469 470 471 472 473
		if (!req)
			goto out;

		/* ICMPs are not backlogged, hence we cannot get
		   an established socket here.
		 */
474
		WARN_ON(req->sk);
L
Linus Torvalds 已提交
475

476
		if (seq != tcp_rsk(req)->snt_isn) {
477
			NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
L
Linus Torvalds 已提交
478 479 480 481 482 483 484 485 486
			goto out;
		}

		/*
		 * Still in SYN_RECV, just remove it silently.
		 * There is no good way to pass the error to the newly
		 * created socket, and POSIX does not want network
		 * errors returned from accept().
		 */
487
		inet_csk_reqsk_queue_drop(sk, req, prev);
L
Linus Torvalds 已提交
488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534
		goto out;

	case TCP_SYN_SENT:
	case TCP_SYN_RECV:  /* Cannot happen.
			       It can f.e. if SYNs crossed.
			     */
		if (!sock_owned_by_user(sk)) {
			sk->sk_err = err;

			sk->sk_error_report(sk);

			tcp_done(sk);
		} else {
			sk->sk_err_soft = err;
		}
		goto out;
	}

	/* If we've already connected we will keep trying
	 * until we time out, or the user gives up.
	 *
	 * rfc1122 4.2.3.9 allows to consider as hard errors
	 * only PROTO_UNREACH and PORT_UNREACH (well, FRAG_FAILED too,
	 * but it is obsoleted by pmtu discovery).
	 *
	 * Note, that in modern internet, where routing is unreliable
	 * and in each dark corner broken firewalls sit, sending random
	 * errors ordered by their masters even this two messages finally lose
	 * their original sense (even Linux sends invalid PORT_UNREACHs)
	 *
	 * Now we are in compliance with RFCs.
	 *							--ANK (980905)
	 */

	inet = inet_sk(sk);
	if (!sock_owned_by_user(sk) && inet->recverr) {
		sk->sk_err = err;
		sk->sk_error_report(sk);
	} else	{ /* Only an error on timeout */
		sk->sk_err_soft = err;
	}

out:
	bh_unlock_sock(sk);
	sock_put(sk);
}

535 536
static void __tcp_v4_send_check(struct sk_buff *skb,
				__be32 saddr, __be32 daddr)
L
Linus Torvalds 已提交
537
{
538
	struct tcphdr *th = tcp_hdr(skb);
L
Linus Torvalds 已提交
539

540
	if (skb->ip_summed == CHECKSUM_PARTIAL) {
541
		th->check = ~tcp_v4_check(skb->len, saddr, daddr, 0);
542
		skb->csum_start = skb_transport_header(skb) - skb->head;
A
Al Viro 已提交
543
		skb->csum_offset = offsetof(struct tcphdr, check);
L
Linus Torvalds 已提交
544
	} else {
545
		th->check = tcp_v4_check(skb->len, saddr, daddr,
546
					 csum_partial(th,
L
Linus Torvalds 已提交
547 548 549 550 551
						      th->doff << 2,
						      skb->csum));
	}
}

552
/* This routine computes an IPv4 TCP checksum. */
553
void tcp_v4_send_check(struct sock *sk, struct sk_buff *skb)
554
{
555
	const struct inet_sock *inet = inet_sk(sk);
556 557 558

	__tcp_v4_send_check(skb, inet->inet_saddr, inet->inet_daddr);
}
E
Eric Dumazet 已提交
559
EXPORT_SYMBOL(tcp_v4_send_check);
560

561 562
int tcp_v4_gso_send_check(struct sk_buff *skb)
{
563
	const struct iphdr *iph;
564 565 566 567 568
	struct tcphdr *th;

	if (!pskb_may_pull(skb, sizeof(*th)))
		return -EINVAL;

569
	iph = ip_hdr(skb);
570
	th = tcp_hdr(skb);
571 572

	th->check = 0;
573
	skb->ip_summed = CHECKSUM_PARTIAL;
574
	__tcp_v4_send_check(skb, iph->saddr, iph->daddr);
575 576 577
	return 0;
}

L
Linus Torvalds 已提交
578 579 580 581 582 583 584 585 586 587 588 589 590
/*
 *	This routine will send an RST to the other tcp.
 *
 *	Someone asks: why I NEVER use socket parameters (TOS, TTL etc.)
 *		      for reset.
 *	Answer: if a packet caused RST, it is not for a socket
 *		existing in our system, if it is matched to a socket,
 *		it is just duplicate segment or bug in other side's TCP.
 *		So that we build reply only basing on parameters
 *		arrived with segment.
 *	Exception: precedence violation. We do not implement it in any case.
 */

591
static void tcp_v4_send_reset(struct sock *sk, struct sk_buff *skb)
L
Linus Torvalds 已提交
592
{
593
	const struct tcphdr *th = tcp_hdr(skb);
594 595 596
	struct {
		struct tcphdr th;
#ifdef CONFIG_TCP_MD5SIG
597
		__be32 opt[(TCPOLEN_MD5SIG_ALIGNED >> 2)];
598 599
#endif
	} rep;
L
Linus Torvalds 已提交
600
	struct ip_reply_arg arg;
601 602 603
#ifdef CONFIG_TCP_MD5SIG
	struct tcp_md5sig_key *key;
#endif
604
	struct net *net;
L
Linus Torvalds 已提交
605 606 607 608 609

	/* Never send a reset in response to a reset. */
	if (th->rst)
		return;

E
Eric Dumazet 已提交
610
	if (skb_rtable(skb)->rt_type != RTN_LOCAL)
L
Linus Torvalds 已提交
611 612 613
		return;

	/* Swap the send and the receive. */
614 615 616 617 618
	memset(&rep, 0, sizeof(rep));
	rep.th.dest   = th->source;
	rep.th.source = th->dest;
	rep.th.doff   = sizeof(struct tcphdr) / 4;
	rep.th.rst    = 1;
L
Linus Torvalds 已提交
619 620

	if (th->ack) {
621
		rep.th.seq = th->ack_seq;
L
Linus Torvalds 已提交
622
	} else {
623 624 625
		rep.th.ack = 1;
		rep.th.ack_seq = htonl(ntohl(th->seq) + th->syn + th->fin +
				       skb->len - (th->doff << 2));
L
Linus Torvalds 已提交
626 627
	}

628
	memset(&arg, 0, sizeof(arg));
629 630 631 632
	arg.iov[0].iov_base = (unsigned char *)&rep;
	arg.iov[0].iov_len  = sizeof(rep.th);

#ifdef CONFIG_TCP_MD5SIG
633
	key = sk ? tcp_v4_md5_do_lookup(sk, ip_hdr(skb)->daddr) : NULL;
634 635 636 637 638 639 640 641 642
	if (key) {
		rep.opt[0] = htonl((TCPOPT_NOP << 24) |
				   (TCPOPT_NOP << 16) |
				   (TCPOPT_MD5SIG << 8) |
				   TCPOLEN_MD5SIG);
		/* Update length and the length the header thinks exists */
		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
		rep.th.doff = arg.iov[0].iov_len / 4;

643
		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[1],
644 645
				     key, ip_hdr(skb)->saddr,
				     ip_hdr(skb)->daddr, &rep.th);
646 647
	}
#endif
648 649
	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
				      ip_hdr(skb)->saddr, /* XXX */
650
				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
L
Linus Torvalds 已提交
651
	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
652
	arg.flags = (sk && inet_sk(sk)->transparent) ? IP_REPLY_ARG_NOSRCCHECK : 0;
L
Linus Torvalds 已提交
653

E
Eric Dumazet 已提交
654
	net = dev_net(skb_dst(skb)->dev);
655
	ip_send_reply(net->ipv4.tcp_sock, skb, ip_hdr(skb)->saddr,
656
		      &arg, arg.iov[0].iov_len);
L
Linus Torvalds 已提交
657

658 659
	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
	TCP_INC_STATS_BH(net, TCP_MIB_OUTRSTS);
L
Linus Torvalds 已提交
660 661 662 663 664 665
}

/* The code following below sending ACKs in SYN-RECV and TIME-WAIT states
   outside socket context is ugly, certainly. What can I do?
 */

666 667
static void tcp_v4_send_ack(struct sk_buff *skb, u32 seq, u32 ack,
			    u32 win, u32 ts, int oif,
668 669
			    struct tcp_md5sig_key *key,
			    int reply_flags)
L
Linus Torvalds 已提交
670
{
671
	const struct tcphdr *th = tcp_hdr(skb);
L
Linus Torvalds 已提交
672 673
	struct {
		struct tcphdr th;
674
		__be32 opt[(TCPOLEN_TSTAMP_ALIGNED >> 2)
675
#ifdef CONFIG_TCP_MD5SIG
676
			   + (TCPOLEN_MD5SIG_ALIGNED >> 2)
677 678
#endif
			];
L
Linus Torvalds 已提交
679 680
	} rep;
	struct ip_reply_arg arg;
E
Eric Dumazet 已提交
681
	struct net *net = dev_net(skb_dst(skb)->dev);
L
Linus Torvalds 已提交
682 683

	memset(&rep.th, 0, sizeof(struct tcphdr));
684
	memset(&arg, 0, sizeof(arg));
L
Linus Torvalds 已提交
685 686 687 688

	arg.iov[0].iov_base = (unsigned char *)&rep;
	arg.iov[0].iov_len  = sizeof(rep.th);
	if (ts) {
689 690 691 692 693
		rep.opt[0] = htonl((TCPOPT_NOP << 24) | (TCPOPT_NOP << 16) |
				   (TCPOPT_TIMESTAMP << 8) |
				   TCPOLEN_TIMESTAMP);
		rep.opt[1] = htonl(tcp_time_stamp);
		rep.opt[2] = htonl(ts);
694
		arg.iov[0].iov_len += TCPOLEN_TSTAMP_ALIGNED;
L
Linus Torvalds 已提交
695 696 697 698 699 700 701 702 703 704 705
	}

	/* Swap the send and the receive. */
	rep.th.dest    = th->source;
	rep.th.source  = th->dest;
	rep.th.doff    = arg.iov[0].iov_len / 4;
	rep.th.seq     = htonl(seq);
	rep.th.ack_seq = htonl(ack);
	rep.th.ack     = 1;
	rep.th.window  = htons(win);

706 707 708 709 710 711 712 713 714 715 716
#ifdef CONFIG_TCP_MD5SIG
	if (key) {
		int offset = (ts) ? 3 : 0;

		rep.opt[offset++] = htonl((TCPOPT_NOP << 24) |
					  (TCPOPT_NOP << 16) |
					  (TCPOPT_MD5SIG << 8) |
					  TCPOLEN_MD5SIG);
		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
		rep.th.doff = arg.iov[0].iov_len/4;

717
		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[offset],
718 719
				    key, ip_hdr(skb)->saddr,
				    ip_hdr(skb)->daddr, &rep.th);
720 721
	}
#endif
722
	arg.flags = reply_flags;
723 724
	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
				      ip_hdr(skb)->saddr, /* XXX */
L
Linus Torvalds 已提交
725 726
				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
727 728
	if (oif)
		arg.bound_dev_if = oif;
L
Linus Torvalds 已提交
729

730
	ip_send_reply(net->ipv4.tcp_sock, skb, ip_hdr(skb)->saddr,
731
		      &arg, arg.iov[0].iov_len);
L
Linus Torvalds 已提交
732

733
	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
L
Linus Torvalds 已提交
734 735 736 737
}

static void tcp_v4_timewait_ack(struct sock *sk, struct sk_buff *skb)
{
738
	struct inet_timewait_sock *tw = inet_twsk(sk);
739
	struct tcp_timewait_sock *tcptw = tcp_twsk(sk);
L
Linus Torvalds 已提交
740

741
	tcp_v4_send_ack(skb, tcptw->tw_snd_nxt, tcptw->tw_rcv_nxt,
742
			tcptw->tw_rcv_wnd >> tw->tw_rcv_wscale,
743 744
			tcptw->tw_ts_recent,
			tw->tw_bound_dev_if,
745 746
			tcp_twsk_md5_key(tcptw),
			tw->tw_transparent ? IP_REPLY_ARG_NOSRCCHECK : 0
747
			);
L
Linus Torvalds 已提交
748

749
	inet_twsk_put(tw);
L
Linus Torvalds 已提交
750 751
}

752
static void tcp_v4_reqsk_send_ack(struct sock *sk, struct sk_buff *skb,
753
				  struct request_sock *req)
L
Linus Torvalds 已提交
754
{
755
	tcp_v4_send_ack(skb, tcp_rsk(req)->snt_isn + 1,
756
			tcp_rsk(req)->rcv_isn + 1, req->rcv_wnd,
757 758
			req->ts_recent,
			0,
759 760
			tcp_v4_md5_do_lookup(sk, ip_hdr(skb)->daddr),
			inet_rsk(req)->no_srccheck ? IP_REPLY_ARG_NOSRCCHECK : 0);
L
Linus Torvalds 已提交
761 762 763
}

/*
764
 *	Send a SYN-ACK after having received a SYN.
765
 *	This still operates on a request_sock only, not on a big
L
Linus Torvalds 已提交
766 767
 *	socket.
 */
768 769 770
static int tcp_v4_send_synack(struct sock *sk, struct dst_entry *dst,
			      struct request_sock *req,
			      struct request_values *rvp)
L
Linus Torvalds 已提交
771
{
772
	const struct inet_request_sock *ireq = inet_rsk(req);
773
	struct flowi4 fl4;
L
Linus Torvalds 已提交
774 775 776 777
	int err = -1;
	struct sk_buff * skb;

	/* First, grab a route. */
778
	if (!dst && (dst = inet_csk_route_req(sk, &fl4, req)) == NULL)
779
		return -1;
L
Linus Torvalds 已提交
780

781
	skb = tcp_make_synack(sk, dst, req, rvp);
L
Linus Torvalds 已提交
782 783

	if (skb) {
784
		__tcp_v4_send_check(skb, ireq->loc_addr, ireq->rmt_addr);
L
Linus Torvalds 已提交
785

786 787 788
		err = ip_build_and_send_pkt(skb, sk, ireq->loc_addr,
					    ireq->rmt_addr,
					    ireq->opt);
789
		err = net_xmit_eval(err);
L
Linus Torvalds 已提交
790 791 792 793 794 795
	}

	dst_release(dst);
	return err;
}

796
static int tcp_v4_rtx_synack(struct sock *sk, struct request_sock *req,
797
			      struct request_values *rvp)
798
{
799 800
	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_RETRANSSEGS);
	return tcp_v4_send_synack(sk, NULL, req, rvp);
801 802
}

L
Linus Torvalds 已提交
803
/*
804
 *	IPv4 request_sock destructor.
L
Linus Torvalds 已提交
805
 */
806
static void tcp_v4_reqsk_destructor(struct request_sock *req)
L
Linus Torvalds 已提交
807
{
J
Jesper Juhl 已提交
808
	kfree(inet_rsk(req)->opt);
L
Linus Torvalds 已提交
809 810
}

811 812 813 814 815 816
/*
 * Return 1 if a syncookie should be sent
 */
int tcp_syn_flood_action(struct sock *sk,
			 const struct sk_buff *skb,
			 const char *proto)
L
Linus Torvalds 已提交
817
{
818 819 820 821 822
	const char *msg = "Dropping request";
	int want_cookie = 0;
	struct listen_sock *lopt;


L
Linus Torvalds 已提交
823

824
#ifdef CONFIG_SYN_COOKIES
825
	if (sysctl_tcp_syncookies) {
826
		msg = "Sending cookies";
827 828 829
		want_cookie = 1;
		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPREQQFULLDOCOOKIES);
	} else
830
#endif
831 832 833 834 835 836 837 838 839 840
		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPREQQFULLDROP);

	lopt = inet_csk(sk)->icsk_accept_queue.listen_opt;
	if (!lopt->synflood_warned) {
		lopt->synflood_warned = 1;
		pr_info("%s: Possible SYN flooding on port %d. %s. "
			" Check SNMP counters.\n",
			proto, ntohs(tcp_hdr(skb)->dest), msg);
	}
	return want_cookie;
841
}
842
EXPORT_SYMBOL(tcp_syn_flood_action);
L
Linus Torvalds 已提交
843 844

/*
845
 * Save and compile IPv4 options into the request_sock if needed.
L
Linus Torvalds 已提交
846
 */
847 848
static struct ip_options_rcu *tcp_v4_save_options(struct sock *sk,
						  struct sk_buff *skb)
L
Linus Torvalds 已提交
849
{
850 851
	const struct ip_options *opt = &(IPCB(skb)->opt);
	struct ip_options_rcu *dopt = NULL;
L
Linus Torvalds 已提交
852 853

	if (opt && opt->optlen) {
854 855
		int opt_size = sizeof(*dopt) + opt->optlen;

L
Linus Torvalds 已提交
856 857
		dopt = kmalloc(opt_size, GFP_ATOMIC);
		if (dopt) {
858
			if (ip_options_echo(&dopt->opt, skb)) {
L
Linus Torvalds 已提交
859 860 861 862 863 864 865 866
				kfree(dopt);
				dopt = NULL;
			}
		}
	}
	return dopt;
}

867 868 869 870 871 872 873 874
#ifdef CONFIG_TCP_MD5SIG
/*
 * RFC2385 MD5 checksumming requires a mapping of
 * IP address->MD5 Key.
 * We need to maintain these in the sk structure.
 */

/* Find the Key structure for an address.  */
875 876
static struct tcp_md5sig_key *
			tcp_v4_md5_do_lookup(struct sock *sk, __be32 addr)
877 878 879 880 881 882 883 884
{
	struct tcp_sock *tp = tcp_sk(sk);
	int i;

	if (!tp->md5sig_info || !tp->md5sig_info->entries4)
		return NULL;
	for (i = 0; i < tp->md5sig_info->entries4; i++) {
		if (tp->md5sig_info->keys4[i].addr == addr)
885
			return &tp->md5sig_info->keys4[i].base;
886 887 888 889 890 891 892
	}
	return NULL;
}

struct tcp_md5sig_key *tcp_v4_md5_lookup(struct sock *sk,
					 struct sock *addr_sk)
{
E
Eric Dumazet 已提交
893
	return tcp_v4_md5_do_lookup(sk, inet_sk(addr_sk)->inet_daddr);
894 895 896
}
EXPORT_SYMBOL(tcp_v4_md5_lookup);

A
Adrian Bunk 已提交
897 898
static struct tcp_md5sig_key *tcp_v4_reqsk_md5_lookup(struct sock *sk,
						      struct request_sock *req)
899 900 901 902 903 904 905 906 907
{
	return tcp_v4_md5_do_lookup(sk, inet_rsk(req)->rmt_addr);
}

/* This can be called on a newly created socket, from other files */
int tcp_v4_md5_do_add(struct sock *sk, __be32 addr,
		      u8 *newkey, u8 newkeylen)
{
	/* Add Key to the list */
908
	struct tcp_md5sig_key *key;
909 910 911
	struct tcp_sock *tp = tcp_sk(sk);
	struct tcp4_md5sig_key *keys;

912
	key = tcp_v4_md5_do_lookup(sk, addr);
913 914
	if (key) {
		/* Pre-existing entry - just update that one. */
915 916 917
		kfree(key->key);
		key->key = newkey;
		key->keylen = newkeylen;
918
	} else {
919 920
		struct tcp_md5sig_info *md5sig;

921
		if (!tp->md5sig_info) {
922 923
			tp->md5sig_info = kzalloc(sizeof(*tp->md5sig_info),
						  GFP_ATOMIC);
924 925 926 927
			if (!tp->md5sig_info) {
				kfree(newkey);
				return -ENOMEM;
			}
E
Eric Dumazet 已提交
928
			sk_nocaps_add(sk, NETIF_F_GSO_MASK);
929
		}
930 931 932 933

		md5sig = tp->md5sig_info;
		if (md5sig->entries4 == 0 &&
		    tcp_alloc_md5sig_pool(sk) == NULL) {
934 935 936
			kfree(newkey);
			return -ENOMEM;
		}
937 938 939

		if (md5sig->alloced4 == md5sig->entries4) {
			keys = kmalloc((sizeof(*keys) *
940
					(md5sig->entries4 + 1)), GFP_ATOMIC);
941 942
			if (!keys) {
				kfree(newkey);
943 944
				if (md5sig->entries4 == 0)
					tcp_free_md5sig_pool();
945 946 947
				return -ENOMEM;
			}

948 949 950
			if (md5sig->entries4)
				memcpy(keys, md5sig->keys4,
				       sizeof(*keys) * md5sig->entries4);
951 952

			/* Free old key list, and reference new one */
953
			kfree(md5sig->keys4);
954 955
			md5sig->keys4 = keys;
			md5sig->alloced4++;
956
		}
957
		md5sig->entries4++;
958 959 960
		md5sig->keys4[md5sig->entries4 - 1].addr        = addr;
		md5sig->keys4[md5sig->entries4 - 1].base.key    = newkey;
		md5sig->keys4[md5sig->entries4 - 1].base.keylen = newkeylen;
961 962 963 964 965 966 967 968
	}
	return 0;
}
EXPORT_SYMBOL(tcp_v4_md5_do_add);

static int tcp_v4_md5_add_func(struct sock *sk, struct sock *addr_sk,
			       u8 *newkey, u8 newkeylen)
{
E
Eric Dumazet 已提交
969
	return tcp_v4_md5_do_add(sk, inet_sk(addr_sk)->inet_daddr,
970 971 972 973 974 975 976 977 978 979 980
				 newkey, newkeylen);
}

int tcp_v4_md5_do_del(struct sock *sk, __be32 addr)
{
	struct tcp_sock *tp = tcp_sk(sk);
	int i;

	for (i = 0; i < tp->md5sig_info->entries4; i++) {
		if (tp->md5sig_info->keys4[i].addr == addr) {
			/* Free the key */
981
			kfree(tp->md5sig_info->keys4[i].base.key);
982 983 984 985 986
			tp->md5sig_info->entries4--;

			if (tp->md5sig_info->entries4 == 0) {
				kfree(tp->md5sig_info->keys4);
				tp->md5sig_info->keys4 = NULL;
987
				tp->md5sig_info->alloced4 = 0;
988
				tcp_free_md5sig_pool();
989
			} else if (tp->md5sig_info->entries4 != i) {
990
				/* Need to do some manipulation */
991 992 993 994
				memmove(&tp->md5sig_info->keys4[i],
					&tp->md5sig_info->keys4[i+1],
					(tp->md5sig_info->entries4 - i) *
					 sizeof(struct tcp4_md5sig_key));
995 996 997 998 999 1000 1001 1002
			}
			return 0;
		}
	}
	return -ENOENT;
}
EXPORT_SYMBOL(tcp_v4_md5_do_del);

1003
static void tcp_v4_clear_md5_list(struct sock *sk)
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
{
	struct tcp_sock *tp = tcp_sk(sk);

	/* Free each key, then the set of key keys,
	 * the crypto element, and then decrement our
	 * hold on the last resort crypto.
	 */
	if (tp->md5sig_info->entries4) {
		int i;
		for (i = 0; i < tp->md5sig_info->entries4; i++)
1014
			kfree(tp->md5sig_info->keys4[i].base.key);
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024
		tp->md5sig_info->entries4 = 0;
		tcp_free_md5sig_pool();
	}
	if (tp->md5sig_info->keys4) {
		kfree(tp->md5sig_info->keys4);
		tp->md5sig_info->keys4 = NULL;
		tp->md5sig_info->alloced4  = 0;
	}
}

1025 1026
static int tcp_v4_parse_md5_keys(struct sock *sk, char __user *optval,
				 int optlen)
1027 1028 1029 1030 1031 1032 1033 1034
{
	struct tcp_md5sig cmd;
	struct sockaddr_in *sin = (struct sockaddr_in *)&cmd.tcpm_addr;
	u8 *newkey;

	if (optlen < sizeof(cmd))
		return -EINVAL;

1035
	if (copy_from_user(&cmd, optval, sizeof(cmd)))
1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051
		return -EFAULT;

	if (sin->sin_family != AF_INET)
		return -EINVAL;

	if (!cmd.tcpm_key || !cmd.tcpm_keylen) {
		if (!tcp_sk(sk)->md5sig_info)
			return -ENOENT;
		return tcp_v4_md5_do_del(sk, sin->sin_addr.s_addr);
	}

	if (cmd.tcpm_keylen > TCP_MD5SIG_MAXKEYLEN)
		return -EINVAL;

	if (!tcp_sk(sk)->md5sig_info) {
		struct tcp_sock *tp = tcp_sk(sk);
1052
		struct tcp_md5sig_info *p;
1053

1054
		p = kzalloc(sizeof(*p), sk->sk_allocation);
1055 1056 1057 1058
		if (!p)
			return -EINVAL;

		tp->md5sig_info = p;
E
Eric Dumazet 已提交
1059
		sk_nocaps_add(sk, NETIF_F_GSO_MASK);
1060 1061
	}

1062
	newkey = kmemdup(cmd.tcpm_key, cmd.tcpm_keylen, sk->sk_allocation);
1063 1064 1065 1066 1067 1068
	if (!newkey)
		return -ENOMEM;
	return tcp_v4_md5_do_add(sk, sin->sin_addr.s_addr,
				 newkey, cmd.tcpm_keylen);
}

1069 1070
static int tcp_v4_md5_hash_pseudoheader(struct tcp_md5sig_pool *hp,
					__be32 daddr, __be32 saddr, int nbytes)
1071 1072
{
	struct tcp4_pseudohdr *bp;
1073
	struct scatterlist sg;
1074 1075 1076 1077

	bp = &hp->md5_blk.ip4;

	/*
1078
	 * 1. the TCP pseudo-header (in the order: source IP address,
1079 1080 1081 1082 1083 1084
	 * destination IP address, zero-padded protocol number, and
	 * segment length)
	 */
	bp->saddr = saddr;
	bp->daddr = daddr;
	bp->pad = 0;
1085
	bp->protocol = IPPROTO_TCP;
1086
	bp->len = cpu_to_be16(nbytes);
1087

1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111
	sg_init_one(&sg, bp, sizeof(*bp));
	return crypto_hash_update(&hp->md5_desc, &sg, sizeof(*bp));
}

static int tcp_v4_md5_hash_hdr(char *md5_hash, struct tcp_md5sig_key *key,
			       __be32 daddr, __be32 saddr, struct tcphdr *th)
{
	struct tcp_md5sig_pool *hp;
	struct hash_desc *desc;

	hp = tcp_get_md5sig_pool();
	if (!hp)
		goto clear_hash_noput;
	desc = &hp->md5_desc;

	if (crypto_hash_init(desc))
		goto clear_hash;
	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, th->doff << 2))
		goto clear_hash;
	if (tcp_md5_hash_header(hp, th))
		goto clear_hash;
	if (tcp_md5_hash_key(hp, key))
		goto clear_hash;
	if (crypto_hash_final(desc, md5_hash))
1112 1113 1114 1115
		goto clear_hash;

	tcp_put_md5sig_pool();
	return 0;
1116

1117 1118 1119 1120
clear_hash:
	tcp_put_md5sig_pool();
clear_hash_noput:
	memset(md5_hash, 0, 16);
1121
	return 1;
1122 1123
}

1124 1125 1126
int tcp_v4_md5_hash_skb(char *md5_hash, struct tcp_md5sig_key *key,
			struct sock *sk, struct request_sock *req,
			struct sk_buff *skb)
1127
{
1128 1129 1130
	struct tcp_md5sig_pool *hp;
	struct hash_desc *desc;
	struct tcphdr *th = tcp_hdr(skb);
1131 1132 1133
	__be32 saddr, daddr;

	if (sk) {
E
Eric Dumazet 已提交
1134 1135
		saddr = inet_sk(sk)->inet_saddr;
		daddr = inet_sk(sk)->inet_daddr;
1136 1137 1138
	} else if (req) {
		saddr = inet_rsk(req)->loc_addr;
		daddr = inet_rsk(req)->rmt_addr;
1139
	} else {
1140 1141 1142
		const struct iphdr *iph = ip_hdr(skb);
		saddr = iph->saddr;
		daddr = iph->daddr;
1143
	}
1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171

	hp = tcp_get_md5sig_pool();
	if (!hp)
		goto clear_hash_noput;
	desc = &hp->md5_desc;

	if (crypto_hash_init(desc))
		goto clear_hash;

	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, skb->len))
		goto clear_hash;
	if (tcp_md5_hash_header(hp, th))
		goto clear_hash;
	if (tcp_md5_hash_skb_data(hp, skb, th->doff << 2))
		goto clear_hash;
	if (tcp_md5_hash_key(hp, key))
		goto clear_hash;
	if (crypto_hash_final(desc, md5_hash))
		goto clear_hash;

	tcp_put_md5sig_pool();
	return 0;

clear_hash:
	tcp_put_md5sig_pool();
clear_hash_noput:
	memset(md5_hash, 0, 16);
	return 1;
1172
}
1173
EXPORT_SYMBOL(tcp_v4_md5_hash_skb);
1174

1175
static int tcp_v4_inbound_md5_hash(struct sock *sk, struct sk_buff *skb)
1176 1177 1178 1179 1180 1181 1182 1183 1184
{
	/*
	 * This gets called for each TCP segment that arrives
	 * so we want to be efficient.
	 * We have 3 drop cases:
	 * o No MD5 hash and one expected.
	 * o MD5 hash and we're not expecting one.
	 * o MD5 hash and its wrong.
	 */
1185
	const __u8 *hash_location = NULL;
1186
	struct tcp_md5sig_key *hash_expected;
1187
	const struct iphdr *iph = ip_hdr(skb);
1188
	const struct tcphdr *th = tcp_hdr(skb);
1189 1190 1191 1192
	int genhash;
	unsigned char newhash[16];

	hash_expected = tcp_v4_md5_do_lookup(sk, iph->saddr);
1193
	hash_location = tcp_parse_md5sig_option(th);
1194 1195 1196 1197 1198 1199

	/* We've parsed the options - do we have a hash? */
	if (!hash_expected && !hash_location)
		return 0;

	if (hash_expected && !hash_location) {
1200
		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5NOTFOUND);
1201 1202 1203 1204
		return 1;
	}

	if (!hash_expected && hash_location) {
1205
		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5UNEXPECTED);
1206 1207 1208 1209 1210 1211
		return 1;
	}

	/* Okay, so this is hash_expected and hash_location -
	 * so we need to calculate the checksum.
	 */
1212 1213 1214
	genhash = tcp_v4_md5_hash_skb(newhash,
				      hash_expected,
				      NULL, NULL, skb);
1215 1216 1217

	if (genhash || memcmp(hash_location, newhash, 16) != 0) {
		if (net_ratelimit()) {
1218 1219 1220
			printk(KERN_INFO "MD5 Hash failed for (%pI4, %d)->(%pI4, %d)%s\n",
			       &iph->saddr, ntohs(th->source),
			       &iph->daddr, ntohs(th->dest),
1221 1222 1223 1224 1225 1226 1227 1228 1229
			       genhash ? " tcp_v4_calc_md5_hash failed" : "");
		}
		return 1;
	}
	return 0;
}

#endif

1230
struct request_sock_ops tcp_request_sock_ops __read_mostly = {
L
Linus Torvalds 已提交
1231
	.family		=	PF_INET,
1232
	.obj_size	=	sizeof(struct tcp_request_sock),
1233
	.rtx_syn_ack	=	tcp_v4_rtx_synack,
1234 1235
	.send_ack	=	tcp_v4_reqsk_send_ack,
	.destructor	=	tcp_v4_reqsk_destructor,
L
Linus Torvalds 已提交
1236
	.send_reset	=	tcp_v4_send_reset,
1237
	.syn_ack_timeout = 	tcp_syn_ack_timeout,
L
Linus Torvalds 已提交
1238 1239
};

1240
#ifdef CONFIG_TCP_MD5SIG
1241
static const struct tcp_request_sock_ops tcp_request_sock_ipv4_ops = {
1242
	.md5_lookup	=	tcp_v4_reqsk_md5_lookup,
1243
	.calc_md5_hash	=	tcp_v4_md5_hash_skb,
1244
};
1245
#endif
1246

L
Linus Torvalds 已提交
1247 1248
int tcp_v4_conn_request(struct sock *sk, struct sk_buff *skb)
{
1249
	struct tcp_extend_values tmp_ext;
L
Linus Torvalds 已提交
1250
	struct tcp_options_received tmp_opt;
1251
	const u8 *hash_location;
1252
	struct request_sock *req;
1253
	struct inet_request_sock *ireq;
1254
	struct tcp_sock *tp = tcp_sk(sk);
1255
	struct dst_entry *dst = NULL;
1256 1257
	__be32 saddr = ip_hdr(skb)->saddr;
	__be32 daddr = ip_hdr(skb)->daddr;
L
Linus Torvalds 已提交
1258 1259 1260 1261
	__u32 isn = TCP_SKB_CB(skb)->when;
	int want_cookie = 0;

	/* Never answer to SYNs send to broadcast or multicast */
E
Eric Dumazet 已提交
1262
	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
L
Linus Torvalds 已提交
1263 1264 1265 1266 1267 1268
		goto drop;

	/* TW buckets are converted to open requests without
	 * limitations, they conserve resources and peer is
	 * evidently real one.
	 */
1269
	if (inet_csk_reqsk_queue_is_full(sk) && !isn) {
1270 1271 1272
		want_cookie = tcp_syn_flood_action(sk, skb, "TCP");
		if (!want_cookie)
			goto drop;
L
Linus Torvalds 已提交
1273 1274 1275 1276 1277 1278 1279
	}

	/* Accept backlog is full. If we have already queued enough
	 * of warm entries in syn queue, drop request. It is better than
	 * clogging syn queue with openreqs with exponentially increasing
	 * timeout.
	 */
1280
	if (sk_acceptq_is_full(sk) && inet_csk_reqsk_queue_young(sk) > 1)
L
Linus Torvalds 已提交
1281 1282
		goto drop;

1283
	req = inet_reqsk_alloc(&tcp_request_sock_ops);
L
Linus Torvalds 已提交
1284 1285 1286
	if (!req)
		goto drop;

1287 1288 1289 1290
#ifdef CONFIG_TCP_MD5SIG
	tcp_rsk(req)->af_specific = &tcp_request_sock_ipv4_ops;
#endif

L
Linus Torvalds 已提交
1291
	tcp_clear_options(&tmp_opt);
1292
	tmp_opt.mss_clamp = TCP_MSS_DEFAULT;
1293
	tmp_opt.user_mss  = tp->rx_opt.user_mss;
1294
	tcp_parse_options(skb, &tmp_opt, &hash_location, 0);
1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309

	if (tmp_opt.cookie_plus > 0 &&
	    tmp_opt.saw_tstamp &&
	    !tp->rx_opt.cookie_out_never &&
	    (sysctl_tcp_cookie_size > 0 ||
	     (tp->cookie_values != NULL &&
	      tp->cookie_values->cookie_desired > 0))) {
		u8 *c;
		u32 *mess = &tmp_ext.cookie_bakery[COOKIE_DIGEST_WORDS];
		int l = tmp_opt.cookie_plus - TCPOLEN_COOKIE_BASE;

		if (tcp_cookie_generator(&tmp_ext.cookie_bakery[0]) != 0)
			goto drop_and_release;

		/* Secret recipe starts with IP addresses */
1310 1311
		*mess++ ^= (__force u32)daddr;
		*mess++ ^= (__force u32)saddr;
L
Linus Torvalds 已提交
1312

1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328
		/* plus variable length Initiator Cookie */
		c = (u8 *)mess;
		while (l-- > 0)
			*c++ ^= *hash_location++;

		want_cookie = 0;	/* not our kind of cookie */
		tmp_ext.cookie_out_never = 0; /* false */
		tmp_ext.cookie_plus = tmp_opt.cookie_plus;
	} else if (!tp->rx_opt.cookie_in_always) {
		/* redundant indications, but ensure initialization. */
		tmp_ext.cookie_out_never = 1; /* true */
		tmp_ext.cookie_plus = 0;
	} else {
		goto drop_and_release;
	}
	tmp_ext.cookie_in_always = tp->rx_opt.cookie_in_always;
L
Linus Torvalds 已提交
1329

1330
	if (want_cookie && !tmp_opt.saw_tstamp)
L
Linus Torvalds 已提交
1331 1332 1333 1334 1335
		tcp_clear_options(&tmp_opt);

	tmp_opt.tstamp_ok = tmp_opt.saw_tstamp;
	tcp_openreq_init(req, &tmp_opt, skb);

1336 1337 1338 1339 1340 1341
	ireq = inet_rsk(req);
	ireq->loc_addr = daddr;
	ireq->rmt_addr = saddr;
	ireq->no_srccheck = inet_sk(sk)->transparent;
	ireq->opt = tcp_v4_save_options(sk, skb);

1342
	if (security_inet_conn_request(sk, skb, req))
1343
		goto drop_and_free;
1344

1345
	if (!want_cookie || tmp_opt.tstamp_ok)
1346
		TCP_ECN_create_request(req, tcp_hdr(skb));
L
Linus Torvalds 已提交
1347 1348 1349

	if (want_cookie) {
		isn = cookie_v4_init_sequence(sk, skb, &req->mss);
1350
		req->cookie_ts = tmp_opt.tstamp_ok;
L
Linus Torvalds 已提交
1351 1352
	} else if (!isn) {
		struct inet_peer *peer = NULL;
1353
		struct flowi4 fl4;
L
Linus Torvalds 已提交
1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364

		/* VJ's idea. We save last timestamp seen
		 * from the destination in peer table, when entering
		 * state TIME-WAIT, and check against it before
		 * accepting new connection request.
		 *
		 * If "isn" is not zero, this request hit alive
		 * timewait bucket, so that all the necessary checks
		 * are made in the function processing timewait state.
		 */
		if (tmp_opt.saw_tstamp &&
1365
		    tcp_death_row.sysctl_tw_recycle &&
1366
		    (dst = inet_csk_route_req(sk, &fl4, req)) != NULL &&
1367 1368
		    fl4.daddr == saddr &&
		    (peer = rt_get_peer((struct rtable *)dst, fl4.daddr)) != NULL) {
1369
			inet_peer_refcheck(peer);
E
Eric Dumazet 已提交
1370
			if ((u32)get_seconds() - peer->tcp_ts_stamp < TCP_PAWS_MSL &&
L
Linus Torvalds 已提交
1371 1372
			    (s32)(peer->tcp_ts - req->ts_recent) >
							TCP_PAWS_WINDOW) {
1373
				NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_PAWSPASSIVEREJECTED);
1374
				goto drop_and_release;
L
Linus Torvalds 已提交
1375 1376 1377 1378
			}
		}
		/* Kill the following clause, if you dislike this way. */
		else if (!sysctl_tcp_syncookies &&
1379
			 (sysctl_max_syn_backlog - inet_csk_reqsk_queue_len(sk) <
L
Linus Torvalds 已提交
1380 1381 1382 1383 1384 1385 1386 1387 1388 1389
			  (sysctl_max_syn_backlog >> 2)) &&
			 (!peer || !peer->tcp_ts_stamp) &&
			 (!dst || !dst_metric(dst, RTAX_RTT))) {
			/* Without syncookies last quarter of
			 * backlog is filled with destinations,
			 * proven to be alive.
			 * It means that we continue to communicate
			 * to destinations, already remembered
			 * to the moment of synflood.
			 */
1390 1391
			LIMIT_NETDEBUG(KERN_DEBUG "TCP: drop open request from %pI4/%u\n",
				       &saddr, ntohs(tcp_hdr(skb)->source));
1392
			goto drop_and_release;
L
Linus Torvalds 已提交
1393 1394
		}

1395
		isn = tcp_v4_init_sequence(skb);
L
Linus Torvalds 已提交
1396
	}
1397
	tcp_rsk(req)->snt_isn = isn;
1398
	tcp_rsk(req)->snt_synack = tcp_time_stamp;
L
Linus Torvalds 已提交
1399

1400 1401
	if (tcp_v4_send_synack(sk, dst, req,
			       (struct request_values *)&tmp_ext) ||
1402
	    want_cookie)
L
Linus Torvalds 已提交
1403 1404
		goto drop_and_free;

1405
	inet_csk_reqsk_queue_hash_add(sk, req, TCP_TIMEOUT_INIT);
L
Linus Torvalds 已提交
1406 1407
	return 0;

1408 1409
drop_and_release:
	dst_release(dst);
L
Linus Torvalds 已提交
1410
drop_and_free:
1411
	reqsk_free(req);
L
Linus Torvalds 已提交
1412 1413 1414
drop:
	return 0;
}
E
Eric Dumazet 已提交
1415
EXPORT_SYMBOL(tcp_v4_conn_request);
L
Linus Torvalds 已提交
1416 1417 1418 1419 1420 1421 1422


/*
 * The three way handshake has completed - we got a valid synack -
 * now create the new socket.
 */
struct sock *tcp_v4_syn_recv_sock(struct sock *sk, struct sk_buff *skb,
1423
				  struct request_sock *req,
L
Linus Torvalds 已提交
1424 1425
				  struct dst_entry *dst)
{
1426
	struct inet_request_sock *ireq;
L
Linus Torvalds 已提交
1427 1428 1429
	struct inet_sock *newinet;
	struct tcp_sock *newtp;
	struct sock *newsk;
1430 1431 1432
#ifdef CONFIG_TCP_MD5SIG
	struct tcp_md5sig_key *key;
#endif
1433
	struct ip_options_rcu *inet_opt;
L
Linus Torvalds 已提交
1434 1435 1436 1437 1438 1439

	if (sk_acceptq_is_full(sk))
		goto exit_overflow;

	newsk = tcp_create_openreq_child(sk, req, skb);
	if (!newsk)
1440
		goto exit_nonewsk;
L
Linus Torvalds 已提交
1441

1442
	newsk->sk_gso_type = SKB_GSO_TCPV4;
L
Linus Torvalds 已提交
1443 1444 1445

	newtp		      = tcp_sk(newsk);
	newinet		      = inet_sk(newsk);
1446
	ireq		      = inet_rsk(req);
E
Eric Dumazet 已提交
1447 1448 1449
	newinet->inet_daddr   = ireq->rmt_addr;
	newinet->inet_rcv_saddr = ireq->loc_addr;
	newinet->inet_saddr	      = ireq->loc_addr;
1450 1451
	inet_opt	      = ireq->opt;
	rcu_assign_pointer(newinet->inet_opt, inet_opt);
1452
	ireq->opt	      = NULL;
1453
	newinet->mc_index     = inet_iif(skb);
1454
	newinet->mc_ttl	      = ip_hdr(skb)->ttl;
1455
	inet_csk(newsk)->icsk_ext_hdr_len = 0;
1456 1457
	if (inet_opt)
		inet_csk(newsk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
E
Eric Dumazet 已提交
1458
	newinet->inet_id = newtp->write_seq ^ jiffies;
L
Linus Torvalds 已提交
1459

1460 1461 1462 1463 1464
	if (!dst && (dst = inet_csk_route_child_sock(sk, newsk, req)) == NULL)
		goto put_and_exit;

	sk_setup_caps(newsk, dst);

J
John Heffner 已提交
1465
	tcp_mtup_init(newsk);
L
Linus Torvalds 已提交
1466
	tcp_sync_mss(newsk, dst_mtu(dst));
1467
	newtp->advmss = dst_metric_advmss(dst);
1468 1469 1470 1471
	if (tcp_sk(sk)->rx_opt.user_mss &&
	    tcp_sk(sk)->rx_opt.user_mss < newtp->advmss)
		newtp->advmss = tcp_sk(sk)->rx_opt.user_mss;

L
Linus Torvalds 已提交
1472
	tcp_initialize_rcv_mss(newsk);
1473 1474 1475 1476
	if (tcp_rsk(req)->snt_synack)
		tcp_valid_rtt_meas(newsk,
		    tcp_time_stamp - tcp_rsk(req)->snt_synack);
	newtp->total_retrans = req->retrans;
L
Linus Torvalds 已提交
1477

1478 1479
#ifdef CONFIG_TCP_MD5SIG
	/* Copy over the MD5 key from the original socket */
E
Eric Dumazet 已提交
1480 1481
	key = tcp_v4_md5_do_lookup(sk, newinet->inet_daddr);
	if (key != NULL) {
1482 1483 1484 1485 1486 1487
		/*
		 * We're using one, so create a matching key
		 * on the newsk structure. If we fail to get
		 * memory, then we end up not copying the key
		 * across. Shucks.
		 */
1488 1489
		char *newkey = kmemdup(key->key, key->keylen, GFP_ATOMIC);
		if (newkey != NULL)
E
Eric Dumazet 已提交
1490
			tcp_v4_md5_do_add(newsk, newinet->inet_daddr,
1491
					  newkey, key->keylen);
E
Eric Dumazet 已提交
1492
		sk_nocaps_add(newsk, NETIF_F_GSO_MASK);
1493 1494 1495
	}
#endif

1496 1497
	if (__inet_inherit_port(sk, newsk) < 0)
		goto put_and_exit;
1498
	__inet_hash_nolisten(newsk, NULL);
L
Linus Torvalds 已提交
1499 1500 1501 1502

	return newsk;

exit_overflow:
1503
	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS);
1504 1505
exit_nonewsk:
	dst_release(dst);
L
Linus Torvalds 已提交
1506
exit:
1507
	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
L
Linus Torvalds 已提交
1508
	return NULL;
1509 1510 1511
put_and_exit:
	sock_put(newsk);
	goto exit;
L
Linus Torvalds 已提交
1512
}
E
Eric Dumazet 已提交
1513
EXPORT_SYMBOL(tcp_v4_syn_recv_sock);
L
Linus Torvalds 已提交
1514 1515 1516

static struct sock *tcp_v4_hnd_req(struct sock *sk, struct sk_buff *skb)
{
1517
	struct tcphdr *th = tcp_hdr(skb);
1518
	const struct iphdr *iph = ip_hdr(skb);
L
Linus Torvalds 已提交
1519
	struct sock *nsk;
1520
	struct request_sock **prev;
L
Linus Torvalds 已提交
1521
	/* Find possible connection requests. */
1522 1523
	struct request_sock *req = inet_csk_search_req(sk, &prev, th->source,
						       iph->saddr, iph->daddr);
L
Linus Torvalds 已提交
1524 1525 1526
	if (req)
		return tcp_check_req(sk, skb, req, prev);

1527
	nsk = inet_lookup_established(sock_net(sk), &tcp_hashinfo, iph->saddr,
1528
			th->source, iph->daddr, th->dest, inet_iif(skb));
L
Linus Torvalds 已提交
1529 1530 1531 1532 1533 1534

	if (nsk) {
		if (nsk->sk_state != TCP_TIME_WAIT) {
			bh_lock_sock(nsk);
			return nsk;
		}
1535
		inet_twsk_put(inet_twsk(nsk));
L
Linus Torvalds 已提交
1536 1537 1538 1539
		return NULL;
	}

#ifdef CONFIG_SYN_COOKIES
1540
	if (!th->syn)
L
Linus Torvalds 已提交
1541 1542 1543 1544 1545
		sk = cookie_v4_check(sk, skb, &(IPCB(skb)->opt));
#endif
	return sk;
}

1546
static __sum16 tcp_v4_checksum_init(struct sk_buff *skb)
L
Linus Torvalds 已提交
1547
{
1548 1549
	const struct iphdr *iph = ip_hdr(skb);

1550
	if (skb->ip_summed == CHECKSUM_COMPLETE) {
1551 1552
		if (!tcp_v4_check(skb->len, iph->saddr,
				  iph->daddr, skb->csum)) {
1553
			skb->ip_summed = CHECKSUM_UNNECESSARY;
L
Linus Torvalds 已提交
1554
			return 0;
1555
		}
L
Linus Torvalds 已提交
1556
	}
1557

1558
	skb->csum = csum_tcpudp_nofold(iph->saddr, iph->daddr,
1559 1560
				       skb->len, IPPROTO_TCP, 0);

L
Linus Torvalds 已提交
1561
	if (skb->len <= 76) {
1562
		return __skb_checksum_complete(skb);
L
Linus Torvalds 已提交
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
	}
	return 0;
}


/* The socket must have it's spinlock held when we get
 * here.
 *
 * We have a potential double-lock case here, so even when
 * doing backlog processing we use the BH locking scheme.
 * This is because we cannot sleep with the original spinlock
 * held.
 */
int tcp_v4_do_rcv(struct sock *sk, struct sk_buff *skb)
{
1578 1579 1580 1581 1582 1583 1584 1585
	struct sock *rsk;
#ifdef CONFIG_TCP_MD5SIG
	/*
	 * We really want to reject the packet as early as possible
	 * if:
	 *  o We're expecting an MD5'd packet and this is no MD5 tcp option
	 *  o There is an MD5 option and we're not expecting one
	 */
1586
	if (tcp_v4_inbound_md5_hash(sk, skb))
1587 1588 1589
		goto discard;
#endif

L
Linus Torvalds 已提交
1590
	if (sk->sk_state == TCP_ESTABLISHED) { /* Fast path */
1591
		sock_rps_save_rxhash(sk, skb);
1592
		if (tcp_rcv_established(sk, skb, tcp_hdr(skb), skb->len)) {
1593
			rsk = sk;
L
Linus Torvalds 已提交
1594
			goto reset;
1595
		}
L
Linus Torvalds 已提交
1596 1597 1598
		return 0;
	}

1599
	if (skb->len < tcp_hdrlen(skb) || tcp_checksum_complete(skb))
L
Linus Torvalds 已提交
1600 1601 1602 1603 1604 1605 1606 1607
		goto csum_err;

	if (sk->sk_state == TCP_LISTEN) {
		struct sock *nsk = tcp_v4_hnd_req(sk, skb);
		if (!nsk)
			goto discard;

		if (nsk != sk) {
1608
			sock_rps_save_rxhash(nsk, skb);
1609 1610
			if (tcp_child_process(sk, nsk, skb)) {
				rsk = nsk;
L
Linus Torvalds 已提交
1611
				goto reset;
1612
			}
L
Linus Torvalds 已提交
1613 1614
			return 0;
		}
1615
	} else
1616
		sock_rps_save_rxhash(sk, skb);
1617

1618
	if (tcp_rcv_state_process(sk, skb, tcp_hdr(skb), skb->len)) {
1619
		rsk = sk;
L
Linus Torvalds 已提交
1620
		goto reset;
1621
	}
L
Linus Torvalds 已提交
1622 1623 1624
	return 0;

reset:
1625
	tcp_v4_send_reset(rsk, skb);
L
Linus Torvalds 已提交
1626 1627 1628 1629 1630 1631 1632 1633 1634 1635
discard:
	kfree_skb(skb);
	/* Be careful here. If this function gets more complicated and
	 * gcc suffers from register pressure on the x86, sk (in %ebx)
	 * might be destroyed here. This current version compiles correctly,
	 * but you have been warned.
	 */
	return 0;

csum_err:
1636
	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_INERRS);
L
Linus Torvalds 已提交
1637 1638
	goto discard;
}
E
Eric Dumazet 已提交
1639
EXPORT_SYMBOL(tcp_v4_do_rcv);
L
Linus Torvalds 已提交
1640 1641 1642 1643 1644 1645 1646

/*
 *	From tcp_input.c
 */

int tcp_v4_rcv(struct sk_buff *skb)
{
1647
	const struct iphdr *iph;
1648
	const struct tcphdr *th;
L
Linus Torvalds 已提交
1649 1650
	struct sock *sk;
	int ret;
1651
	struct net *net = dev_net(skb->dev);
L
Linus Torvalds 已提交
1652 1653 1654 1655 1656

	if (skb->pkt_type != PACKET_HOST)
		goto discard_it;

	/* Count it even if it's bad */
1657
	TCP_INC_STATS_BH(net, TCP_MIB_INSEGS);
L
Linus Torvalds 已提交
1658 1659 1660 1661

	if (!pskb_may_pull(skb, sizeof(struct tcphdr)))
		goto discard_it;

1662
	th = tcp_hdr(skb);
L
Linus Torvalds 已提交
1663 1664 1665 1666 1667 1668 1669 1670

	if (th->doff < sizeof(struct tcphdr) / 4)
		goto bad_packet;
	if (!pskb_may_pull(skb, th->doff * 4))
		goto discard_it;

	/* An explanation is required here, I think.
	 * Packet length and doff are validated by header prediction,
S
Stephen Hemminger 已提交
1671
	 * provided case of th->doff==0 is eliminated.
L
Linus Torvalds 已提交
1672
	 * So, we defer the checks. */
1673
	if (!skb_csum_unnecessary(skb) && tcp_v4_checksum_init(skb))
L
Linus Torvalds 已提交
1674 1675
		goto bad_packet;

1676
	th = tcp_hdr(skb);
1677
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
1678 1679 1680 1681 1682
	TCP_SKB_CB(skb)->seq = ntohl(th->seq);
	TCP_SKB_CB(skb)->end_seq = (TCP_SKB_CB(skb)->seq + th->syn + th->fin +
				    skb->len - th->doff * 4);
	TCP_SKB_CB(skb)->ack_seq = ntohl(th->ack_seq);
	TCP_SKB_CB(skb)->when	 = 0;
1683
	TCP_SKB_CB(skb)->ip_dsfield = ipv4_get_dsfield(iph);
L
Linus Torvalds 已提交
1684 1685
	TCP_SKB_CB(skb)->sacked	 = 0;

1686
	sk = __inet_lookup_skb(&tcp_hashinfo, skb, th->source, th->dest);
L
Linus Torvalds 已提交
1687 1688 1689
	if (!sk)
		goto no_tcp_socket;

E
Eric Dumazet 已提交
1690 1691 1692 1693
process:
	if (sk->sk_state == TCP_TIME_WAIT)
		goto do_time_wait;

1694 1695
	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
1696
		goto discard_and_relse;
1697
	}
1698

L
Linus Torvalds 已提交
1699 1700
	if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb))
		goto discard_and_relse;
1701
	nf_reset(skb);
L
Linus Torvalds 已提交
1702

1703
	if (sk_filter(sk, skb))
L
Linus Torvalds 已提交
1704 1705 1706 1707
		goto discard_and_relse;

	skb->dev = NULL;

1708
	bh_lock_sock_nested(sk);
L
Linus Torvalds 已提交
1709 1710
	ret = 0;
	if (!sock_owned_by_user(sk)) {
C
Chris Leech 已提交
1711 1712 1713
#ifdef CONFIG_NET_DMA
		struct tcp_sock *tp = tcp_sk(sk);
		if (!tp->ucopy.dma_chan && tp->ucopy.pinned_list)
1714
			tp->ucopy.dma_chan = dma_find_channel(DMA_MEMCPY);
C
Chris Leech 已提交
1715
		if (tp->ucopy.dma_chan)
L
Linus Torvalds 已提交
1716
			ret = tcp_v4_do_rcv(sk, skb);
C
Chris Leech 已提交
1717 1718 1719 1720
		else
#endif
		{
			if (!tcp_prequeue(sk, skb))
S
Shan Wei 已提交
1721
				ret = tcp_v4_do_rcv(sk, skb);
C
Chris Leech 已提交
1722
		}
1723
	} else if (unlikely(sk_add_backlog(sk, skb))) {
Z
Zhu Yi 已提交
1724
		bh_unlock_sock(sk);
1725
		NET_INC_STATS_BH(net, LINUX_MIB_TCPBACKLOGDROP);
Z
Zhu Yi 已提交
1726 1727
		goto discard_and_relse;
	}
L
Linus Torvalds 已提交
1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739
	bh_unlock_sock(sk);

	sock_put(sk);

	return ret;

no_tcp_socket:
	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb))
		goto discard_it;

	if (skb->len < (th->doff << 2) || tcp_checksum_complete(skb)) {
bad_packet:
1740
		TCP_INC_STATS_BH(net, TCP_MIB_INERRS);
L
Linus Torvalds 已提交
1741
	} else {
1742
		tcp_v4_send_reset(NULL, skb);
L
Linus Torvalds 已提交
1743 1744 1745 1746 1747
	}

discard_it:
	/* Discard frame. */
	kfree_skb(skb);
1748
	return 0;
L
Linus Torvalds 已提交
1749 1750 1751 1752 1753 1754 1755

discard_and_relse:
	sock_put(sk);
	goto discard_it;

do_time_wait:
	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) {
1756
		inet_twsk_put(inet_twsk(sk));
L
Linus Torvalds 已提交
1757 1758 1759 1760
		goto discard_it;
	}

	if (skb->len < (th->doff << 2) || tcp_checksum_complete(skb)) {
1761
		TCP_INC_STATS_BH(net, TCP_MIB_INERRS);
1762
		inet_twsk_put(inet_twsk(sk));
L
Linus Torvalds 已提交
1763 1764
		goto discard_it;
	}
1765
	switch (tcp_timewait_state_process(inet_twsk(sk), skb, th)) {
L
Linus Torvalds 已提交
1766
	case TCP_TW_SYN: {
1767
		struct sock *sk2 = inet_lookup_listener(dev_net(skb->dev),
1768
							&tcp_hashinfo,
1769
							iph->daddr, th->dest,
1770
							inet_iif(skb));
L
Linus Torvalds 已提交
1771
		if (sk2) {
1772 1773
			inet_twsk_deschedule(inet_twsk(sk), &tcp_death_row);
			inet_twsk_put(inet_twsk(sk));
L
Linus Torvalds 已提交
1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788
			sk = sk2;
			goto process;
		}
		/* Fall through to ACK */
	}
	case TCP_TW_ACK:
		tcp_v4_timewait_ack(sk, skb);
		break;
	case TCP_TW_RST:
		goto no_tcp_socket;
	case TCP_TW_SUCCESS:;
	}
	goto discard_it;
}

1789
struct inet_peer *tcp_v4_get_peer(struct sock *sk, bool *release_it)
L
Linus Torvalds 已提交
1790
{
1791
	struct rtable *rt = (struct rtable *) __sk_dst_get(sk);
L
Linus Torvalds 已提交
1792
	struct inet_sock *inet = inet_sk(sk);
1793
	struct inet_peer *peer;
L
Linus Torvalds 已提交
1794

1795 1796
	if (!rt ||
	    inet->cork.fl.u.ip4.daddr != inet->inet_daddr) {
1797
		peer = inet_getpeer_v4(inet->inet_daddr, 1);
1798
		*release_it = true;
L
Linus Torvalds 已提交
1799 1800
	} else {
		if (!rt->peer)
1801
			rt_bind_peer(rt, inet->inet_daddr, 1);
L
Linus Torvalds 已提交
1802
		peer = rt->peer;
1803
		*release_it = false;
L
Linus Torvalds 已提交
1804 1805
	}

1806
	return peer;
L
Linus Torvalds 已提交
1807
}
1808
EXPORT_SYMBOL(tcp_v4_get_peer);
L
Linus Torvalds 已提交
1809

1810
void *tcp_v4_tw_get_peer(struct sock *sk)
L
Linus Torvalds 已提交
1811
{
1812
	const struct inet_timewait_sock *tw = inet_twsk(sk);
L
Linus Torvalds 已提交
1813

1814
	return inet_getpeer_v4(tw->tw_daddr, 1);
L
Linus Torvalds 已提交
1815
}
1816 1817 1818 1819 1820 1821 1822 1823
EXPORT_SYMBOL(tcp_v4_tw_get_peer);

static struct timewait_sock_ops tcp_timewait_sock_ops = {
	.twsk_obj_size	= sizeof(struct tcp_timewait_sock),
	.twsk_unique	= tcp_twsk_unique,
	.twsk_destructor= tcp_twsk_destructor,
	.twsk_getpeer	= tcp_v4_tw_get_peer,
};
L
Linus Torvalds 已提交
1824

1825
const struct inet_connection_sock_af_ops ipv4_specific = {
1826 1827 1828 1829 1830
	.queue_xmit	   = ip_queue_xmit,
	.send_check	   = tcp_v4_send_check,
	.rebuild_header	   = inet_sk_rebuild_header,
	.conn_request	   = tcp_v4_conn_request,
	.syn_recv_sock	   = tcp_v4_syn_recv_sock,
1831
	.get_peer	   = tcp_v4_get_peer,
1832 1833 1834 1835 1836
	.net_header_len	   = sizeof(struct iphdr),
	.setsockopt	   = ip_setsockopt,
	.getsockopt	   = ip_getsockopt,
	.addr2sockaddr	   = inet_csk_addr2sockaddr,
	.sockaddr_len	   = sizeof(struct sockaddr_in),
1837
	.bind_conflict	   = inet_csk_bind_conflict,
1838
#ifdef CONFIG_COMPAT
1839 1840
	.compat_setsockopt = compat_ip_setsockopt,
	.compat_getsockopt = compat_ip_getsockopt,
1841
#endif
L
Linus Torvalds 已提交
1842
};
E
Eric Dumazet 已提交
1843
EXPORT_SYMBOL(ipv4_specific);
L
Linus Torvalds 已提交
1844

1845
#ifdef CONFIG_TCP_MD5SIG
1846
static const struct tcp_sock_af_ops tcp_sock_ipv4_specific = {
1847
	.md5_lookup		= tcp_v4_md5_lookup,
1848
	.calc_md5_hash		= tcp_v4_md5_hash_skb,
1849 1850 1851
	.md5_add		= tcp_v4_md5_add_func,
	.md5_parse		= tcp_v4_parse_md5_keys,
};
1852
#endif
1853

L
Linus Torvalds 已提交
1854 1855 1856 1857 1858
/* NOTE: A lot of things set to zero explicitly by call to
 *       sk_alloc() so need not be done here.
 */
static int tcp_v4_init_sock(struct sock *sk)
{
1859
	struct inet_connection_sock *icsk = inet_csk(sk);
L
Linus Torvalds 已提交
1860 1861 1862 1863 1864 1865
	struct tcp_sock *tp = tcp_sk(sk);

	skb_queue_head_init(&tp->out_of_order_queue);
	tcp_init_xmit_timers(sk);
	tcp_prequeue_init(tp);

1866
	icsk->icsk_rto = TCP_TIMEOUT_INIT;
L
Linus Torvalds 已提交
1867 1868 1869 1870 1871 1872 1873
	tp->mdev = TCP_TIMEOUT_INIT;

	/* So many TCP implementations out there (incorrectly) count the
	 * initial SYN frame in their delayed-ACK and congestion control
	 * algorithms that we must have the following bandaid to talk
	 * efficiently to them.  -DaveM
	 */
1874
	tp->snd_cwnd = TCP_INIT_CWND;
L
Linus Torvalds 已提交
1875 1876 1877 1878

	/* See draft-stevens-tcpca-spec-01 for discussion of the
	 * initialization of these values.
	 */
I
Ilpo Järvinen 已提交
1879
	tp->snd_ssthresh = TCP_INFINITE_SSTHRESH;
L
Linus Torvalds 已提交
1880
	tp->snd_cwnd_clamp = ~0;
1881
	tp->mss_cache = TCP_MSS_DEFAULT;
L
Linus Torvalds 已提交
1882 1883

	tp->reordering = sysctl_tcp_reordering;
1884
	icsk->icsk_ca_ops = &tcp_init_congestion_ops;
L
Linus Torvalds 已提交
1885 1886 1887 1888 1889 1890

	sk->sk_state = TCP_CLOSE;

	sk->sk_write_space = sk_stream_write_space;
	sock_set_flag(sk, SOCK_USE_WRITE_QUEUE);

1891
	icsk->icsk_af_ops = &ipv4_specific;
1892
	icsk->icsk_sync_mss = tcp_sync_mss;
1893 1894 1895
#ifdef CONFIG_TCP_MD5SIG
	tp->af_specific = &tcp_sock_ipv4_specific;
#endif
L
Linus Torvalds 已提交
1896

1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909
	/* TCP Cookie Transactions */
	if (sysctl_tcp_cookie_size > 0) {
		/* Default, cookies without s_data_payload. */
		tp->cookie_values =
			kzalloc(sizeof(*tp->cookie_values),
				sk->sk_allocation);
		if (tp->cookie_values != NULL)
			kref_init(&tp->cookie_values->kref);
	}
	/* Presumed zeroed, in order of appearance:
	 *	cookie_in_always, cookie_out_never,
	 *	s_data_constant, s_data_in, s_data_out
	 */
L
Linus Torvalds 已提交
1910 1911 1912
	sk->sk_sndbuf = sysctl_tcp_wmem[1];
	sk->sk_rcvbuf = sysctl_tcp_rmem[1];

H
Herbert Xu 已提交
1913
	local_bh_disable();
1914
	percpu_counter_inc(&tcp_sockets_allocated);
H
Herbert Xu 已提交
1915
	local_bh_enable();
L
Linus Torvalds 已提交
1916 1917 1918 1919

	return 0;
}

1920
void tcp_v4_destroy_sock(struct sock *sk)
L
Linus Torvalds 已提交
1921 1922 1923 1924 1925
{
	struct tcp_sock *tp = tcp_sk(sk);

	tcp_clear_xmit_timers(sk);

1926
	tcp_cleanup_congestion_control(sk);
1927

L
Linus Torvalds 已提交
1928
	/* Cleanup up the write buffer. */
1929
	tcp_write_queue_purge(sk);
L
Linus Torvalds 已提交
1930 1931

	/* Cleans up our, hopefully empty, out_of_order_queue. */
1932
	__skb_queue_purge(&tp->out_of_order_queue);
L
Linus Torvalds 已提交
1933

1934 1935 1936 1937 1938 1939 1940 1941 1942
#ifdef CONFIG_TCP_MD5SIG
	/* Clean up the MD5 key list, if any */
	if (tp->md5sig_info) {
		tcp_v4_clear_md5_list(sk);
		kfree(tp->md5sig_info);
		tp->md5sig_info = NULL;
	}
#endif

C
Chris Leech 已提交
1943 1944
#ifdef CONFIG_NET_DMA
	/* Cleans up our sk_async_wait_queue */
1945
	__skb_queue_purge(&sk->sk_async_wait_queue);
C
Chris Leech 已提交
1946 1947
#endif

L
Linus Torvalds 已提交
1948 1949 1950 1951
	/* Clean prequeue, it must be empty really */
	__skb_queue_purge(&tp->ucopy.prequeue);

	/* Clean up a referenced TCP bind bucket. */
1952
	if (inet_csk(sk)->icsk_bind_hash)
1953
		inet_put_port(sk);
L
Linus Torvalds 已提交
1954 1955 1956 1957 1958 1959 1960 1961 1962

	/*
	 * If sendmsg cached page exists, toss it.
	 */
	if (sk->sk_sndmsg_page) {
		__free_page(sk->sk_sndmsg_page);
		sk->sk_sndmsg_page = NULL;
	}

1963 1964 1965 1966 1967 1968 1969
	/* TCP Cookie Transactions */
	if (tp->cookie_values != NULL) {
		kref_put(&tp->cookie_values->kref,
			 tcp_cookie_values_release);
		tp->cookie_values = NULL;
	}

1970
	percpu_counter_dec(&tcp_sockets_allocated);
L
Linus Torvalds 已提交
1971 1972 1973 1974 1975 1976
}
EXPORT_SYMBOL(tcp_v4_destroy_sock);

#ifdef CONFIG_PROC_FS
/* Proc filesystem TCP sock list dumping. */

1977
static inline struct inet_timewait_sock *tw_head(struct hlist_nulls_head *head)
L
Linus Torvalds 已提交
1978
{
1979
	return hlist_nulls_empty(head) ? NULL :
1980
		list_entry(head->first, struct inet_timewait_sock, tw_node);
L
Linus Torvalds 已提交
1981 1982
}

1983
static inline struct inet_timewait_sock *tw_next(struct inet_timewait_sock *tw)
L
Linus Torvalds 已提交
1984
{
1985 1986
	return !is_a_nulls(tw->tw_node.next) ?
		hlist_nulls_entry(tw->tw_node.next, typeof(*tw), tw_node) : NULL;
L
Linus Torvalds 已提交
1987 1988
}

1989 1990 1991 1992 1993
/*
 * Get next listener socket follow cur.  If cur is NULL, get first socket
 * starting from bucket given in st->bucket; when st->bucket is zero the
 * very first socket in the hash table is returned.
 */
L
Linus Torvalds 已提交
1994 1995
static void *listening_get_next(struct seq_file *seq, void *cur)
{
1996
	struct inet_connection_sock *icsk;
1997
	struct hlist_nulls_node *node;
L
Linus Torvalds 已提交
1998
	struct sock *sk = cur;
1999
	struct inet_listen_hashbucket *ilb;
J
Jianjun Kong 已提交
2000
	struct tcp_iter_state *st = seq->private;
2001
	struct net *net = seq_file_net(seq);
L
Linus Torvalds 已提交
2002 2003

	if (!sk) {
2004
		ilb = &tcp_hashinfo.listening_hash[st->bucket];
2005
		spin_lock_bh(&ilb->lock);
2006
		sk = sk_nulls_head(&ilb->head);
2007
		st->offset = 0;
L
Linus Torvalds 已提交
2008 2009
		goto get_sk;
	}
2010
	ilb = &tcp_hashinfo.listening_hash[st->bucket];
L
Linus Torvalds 已提交
2011
	++st->num;
2012
	++st->offset;
L
Linus Torvalds 已提交
2013 2014

	if (st->state == TCP_SEQ_STATE_OPENREQ) {
2015
		struct request_sock *req = cur;
L
Linus Torvalds 已提交
2016

2017
		icsk = inet_csk(st->syn_wait_sk);
L
Linus Torvalds 已提交
2018 2019 2020
		req = req->dl_next;
		while (1) {
			while (req) {
2021
				if (req->rsk_ops->family == st->family) {
L
Linus Torvalds 已提交
2022 2023 2024 2025 2026
					cur = req;
					goto out;
				}
				req = req->dl_next;
			}
2027
			if (++st->sbucket >= icsk->icsk_accept_queue.listen_opt->nr_table_entries)
L
Linus Torvalds 已提交
2028 2029
				break;
get_req:
2030
			req = icsk->icsk_accept_queue.listen_opt->syn_table[st->sbucket];
L
Linus Torvalds 已提交
2031
		}
E
Eric Dumazet 已提交
2032
		sk	  = sk_nulls_next(st->syn_wait_sk);
L
Linus Torvalds 已提交
2033
		st->state = TCP_SEQ_STATE_LISTENING;
2034
		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
L
Linus Torvalds 已提交
2035
	} else {
2036
		icsk = inet_csk(sk);
2037 2038
		read_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
		if (reqsk_queue_len(&icsk->icsk_accept_queue))
L
Linus Torvalds 已提交
2039
			goto start_req;
2040
		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
E
Eric Dumazet 已提交
2041
		sk = sk_nulls_next(sk);
L
Linus Torvalds 已提交
2042 2043
	}
get_sk:
2044
	sk_nulls_for_each_from(sk, node) {
2045 2046 2047
		if (!net_eq(sock_net(sk), net))
			continue;
		if (sk->sk_family == st->family) {
L
Linus Torvalds 已提交
2048 2049 2050
			cur = sk;
			goto out;
		}
2051
		icsk = inet_csk(sk);
2052 2053
		read_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
		if (reqsk_queue_len(&icsk->icsk_accept_queue)) {
L
Linus Torvalds 已提交
2054 2055 2056 2057 2058 2059 2060
start_req:
			st->uid		= sock_i_uid(sk);
			st->syn_wait_sk = sk;
			st->state	= TCP_SEQ_STATE_OPENREQ;
			st->sbucket	= 0;
			goto get_req;
		}
2061
		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
L
Linus Torvalds 已提交
2062
	}
2063
	spin_unlock_bh(&ilb->lock);
2064
	st->offset = 0;
2065
	if (++st->bucket < INET_LHTABLE_SIZE) {
2066 2067
		ilb = &tcp_hashinfo.listening_hash[st->bucket];
		spin_lock_bh(&ilb->lock);
2068
		sk = sk_nulls_head(&ilb->head);
L
Linus Torvalds 已提交
2069 2070 2071 2072 2073 2074 2075 2076 2077
		goto get_sk;
	}
	cur = NULL;
out:
	return cur;
}

static void *listening_get_idx(struct seq_file *seq, loff_t *pos)
{
2078 2079 2080 2081 2082 2083
	struct tcp_iter_state *st = seq->private;
	void *rc;

	st->bucket = 0;
	st->offset = 0;
	rc = listening_get_next(seq, NULL);
L
Linus Torvalds 已提交
2084 2085 2086 2087 2088 2089 2090 2091

	while (rc && *pos) {
		rc = listening_get_next(seq, rc);
		--*pos;
	}
	return rc;
}

2092 2093
static inline int empty_bucket(struct tcp_iter_state *st)
{
2094 2095
	return hlist_nulls_empty(&tcp_hashinfo.ehash[st->bucket].chain) &&
		hlist_nulls_empty(&tcp_hashinfo.ehash[st->bucket].twchain);
2096 2097
}

2098 2099 2100 2101
/*
 * Get first established socket starting from bucket given in st->bucket.
 * If st->bucket is zero, the very first socket in the hash is returned.
 */
L
Linus Torvalds 已提交
2102 2103
static void *established_get_first(struct seq_file *seq)
{
J
Jianjun Kong 已提交
2104
	struct tcp_iter_state *st = seq->private;
2105
	struct net *net = seq_file_net(seq);
L
Linus Torvalds 已提交
2106 2107
	void *rc = NULL;

2108 2109
	st->offset = 0;
	for (; st->bucket <= tcp_hashinfo.ehash_mask; ++st->bucket) {
L
Linus Torvalds 已提交
2110
		struct sock *sk;
2111
		struct hlist_nulls_node *node;
2112
		struct inet_timewait_sock *tw;
2113
		spinlock_t *lock = inet_ehash_lockp(&tcp_hashinfo, st->bucket);
L
Linus Torvalds 已提交
2114

2115 2116 2117 2118
		/* Lockless fast path for the common case of empty buckets */
		if (empty_bucket(st))
			continue;

2119
		spin_lock_bh(lock);
2120
		sk_nulls_for_each(sk, node, &tcp_hashinfo.ehash[st->bucket].chain) {
2121
			if (sk->sk_family != st->family ||
2122
			    !net_eq(sock_net(sk), net)) {
L
Linus Torvalds 已提交
2123 2124 2125 2126 2127 2128
				continue;
			}
			rc = sk;
			goto out;
		}
		st->state = TCP_SEQ_STATE_TIME_WAIT;
2129
		inet_twsk_for_each(tw, node,
2130
				   &tcp_hashinfo.ehash[st->bucket].twchain) {
2131
			if (tw->tw_family != st->family ||
2132
			    !net_eq(twsk_net(tw), net)) {
L
Linus Torvalds 已提交
2133 2134 2135 2136 2137
				continue;
			}
			rc = tw;
			goto out;
		}
2138
		spin_unlock_bh(lock);
L
Linus Torvalds 已提交
2139 2140 2141 2142 2143 2144 2145 2146 2147
		st->state = TCP_SEQ_STATE_ESTABLISHED;
	}
out:
	return rc;
}

static void *established_get_next(struct seq_file *seq, void *cur)
{
	struct sock *sk = cur;
2148
	struct inet_timewait_sock *tw;
2149
	struct hlist_nulls_node *node;
J
Jianjun Kong 已提交
2150
	struct tcp_iter_state *st = seq->private;
2151
	struct net *net = seq_file_net(seq);
L
Linus Torvalds 已提交
2152 2153

	++st->num;
2154
	++st->offset;
L
Linus Torvalds 已提交
2155 2156 2157 2158 2159

	if (st->state == TCP_SEQ_STATE_TIME_WAIT) {
		tw = cur;
		tw = tw_next(tw);
get_tw:
2160
		while (tw && (tw->tw_family != st->family || !net_eq(twsk_net(tw), net))) {
L
Linus Torvalds 已提交
2161 2162 2163 2164 2165 2166
			tw = tw_next(tw);
		}
		if (tw) {
			cur = tw;
			goto out;
		}
2167
		spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
L
Linus Torvalds 已提交
2168 2169
		st->state = TCP_SEQ_STATE_ESTABLISHED;

2170
		/* Look for next non empty bucket */
2171
		st->offset = 0;
2172
		while (++st->bucket <= tcp_hashinfo.ehash_mask &&
2173 2174
				empty_bucket(st))
			;
2175
		if (st->bucket > tcp_hashinfo.ehash_mask)
2176 2177
			return NULL;

2178
		spin_lock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
2179
		sk = sk_nulls_head(&tcp_hashinfo.ehash[st->bucket].chain);
L
Linus Torvalds 已提交
2180
	} else
2181
		sk = sk_nulls_next(sk);
L
Linus Torvalds 已提交
2182

2183
	sk_nulls_for_each_from(sk, node) {
2184
		if (sk->sk_family == st->family && net_eq(sock_net(sk), net))
L
Linus Torvalds 已提交
2185 2186 2187 2188
			goto found;
	}

	st->state = TCP_SEQ_STATE_TIME_WAIT;
2189
	tw = tw_head(&tcp_hashinfo.ehash[st->bucket].twchain);
L
Linus Torvalds 已提交
2190 2191 2192 2193 2194 2195 2196 2197 2198
	goto get_tw;
found:
	cur = sk;
out:
	return cur;
}

static void *established_get_idx(struct seq_file *seq, loff_t pos)
{
2199 2200 2201 2202 2203
	struct tcp_iter_state *st = seq->private;
	void *rc;

	st->bucket = 0;
	rc = established_get_first(seq);
L
Linus Torvalds 已提交
2204 2205 2206 2207

	while (rc && pos) {
		rc = established_get_next(seq, rc);
		--pos;
2208
	}
L
Linus Torvalds 已提交
2209 2210 2211 2212 2213 2214
	return rc;
}

static void *tcp_get_idx(struct seq_file *seq, loff_t pos)
{
	void *rc;
J
Jianjun Kong 已提交
2215
	struct tcp_iter_state *st = seq->private;
L
Linus Torvalds 已提交
2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227

	st->state = TCP_SEQ_STATE_LISTENING;
	rc	  = listening_get_idx(seq, &pos);

	if (!rc) {
		st->state = TCP_SEQ_STATE_ESTABLISHED;
		rc	  = established_get_idx(seq, pos);
	}

	return rc;
}

2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262
static void *tcp_seek_last_pos(struct seq_file *seq)
{
	struct tcp_iter_state *st = seq->private;
	int offset = st->offset;
	int orig_num = st->num;
	void *rc = NULL;

	switch (st->state) {
	case TCP_SEQ_STATE_OPENREQ:
	case TCP_SEQ_STATE_LISTENING:
		if (st->bucket >= INET_LHTABLE_SIZE)
			break;
		st->state = TCP_SEQ_STATE_LISTENING;
		rc = listening_get_next(seq, NULL);
		while (offset-- && rc)
			rc = listening_get_next(seq, rc);
		if (rc)
			break;
		st->bucket = 0;
		/* Fallthrough */
	case TCP_SEQ_STATE_ESTABLISHED:
	case TCP_SEQ_STATE_TIME_WAIT:
		st->state = TCP_SEQ_STATE_ESTABLISHED;
		if (st->bucket > tcp_hashinfo.ehash_mask)
			break;
		rc = established_get_first(seq);
		while (offset-- && rc)
			rc = established_get_next(seq, rc);
	}

	st->num = orig_num;

	return rc;
}

L
Linus Torvalds 已提交
2263 2264
static void *tcp_seq_start(struct seq_file *seq, loff_t *pos)
{
J
Jianjun Kong 已提交
2265
	struct tcp_iter_state *st = seq->private;
2266 2267 2268 2269 2270 2271 2272 2273
	void *rc;

	if (*pos && *pos == st->last_pos) {
		rc = tcp_seek_last_pos(seq);
		if (rc)
			goto out;
	}

L
Linus Torvalds 已提交
2274 2275
	st->state = TCP_SEQ_STATE_LISTENING;
	st->num = 0;
2276 2277 2278 2279 2280 2281 2282
	st->bucket = 0;
	st->offset = 0;
	rc = *pos ? tcp_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;

out:
	st->last_pos = *pos;
	return rc;
L
Linus Torvalds 已提交
2283 2284 2285 2286
}

static void *tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
2287
	struct tcp_iter_state *st = seq->private;
L
Linus Torvalds 已提交
2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300
	void *rc = NULL;

	if (v == SEQ_START_TOKEN) {
		rc = tcp_get_idx(seq, 0);
		goto out;
	}

	switch (st->state) {
	case TCP_SEQ_STATE_OPENREQ:
	case TCP_SEQ_STATE_LISTENING:
		rc = listening_get_next(seq, v);
		if (!rc) {
			st->state = TCP_SEQ_STATE_ESTABLISHED;
2301 2302
			st->bucket = 0;
			st->offset = 0;
L
Linus Torvalds 已提交
2303 2304 2305 2306 2307 2308 2309 2310 2311 2312
			rc	  = established_get_first(seq);
		}
		break;
	case TCP_SEQ_STATE_ESTABLISHED:
	case TCP_SEQ_STATE_TIME_WAIT:
		rc = established_get_next(seq, v);
		break;
	}
out:
	++*pos;
2313
	st->last_pos = *pos;
L
Linus Torvalds 已提交
2314 2315 2316 2317 2318
	return rc;
}

static void tcp_seq_stop(struct seq_file *seq, void *v)
{
J
Jianjun Kong 已提交
2319
	struct tcp_iter_state *st = seq->private;
L
Linus Torvalds 已提交
2320 2321 2322 2323

	switch (st->state) {
	case TCP_SEQ_STATE_OPENREQ:
		if (v) {
2324 2325
			struct inet_connection_sock *icsk = inet_csk(st->syn_wait_sk);
			read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
L
Linus Torvalds 已提交
2326 2327 2328
		}
	case TCP_SEQ_STATE_LISTENING:
		if (v != SEQ_START_TOKEN)
2329
			spin_unlock_bh(&tcp_hashinfo.listening_hash[st->bucket].lock);
L
Linus Torvalds 已提交
2330 2331 2332 2333
		break;
	case TCP_SEQ_STATE_TIME_WAIT:
	case TCP_SEQ_STATE_ESTABLISHED:
		if (v)
2334
			spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
L
Linus Torvalds 已提交
2335 2336 2337 2338 2339 2340 2341 2342
		break;
	}
}

static int tcp_seq_open(struct inode *inode, struct file *file)
{
	struct tcp_seq_afinfo *afinfo = PDE(inode)->data;
	struct tcp_iter_state *s;
2343
	int err;
L
Linus Torvalds 已提交
2344

2345 2346 2347 2348
	err = seq_open_net(inode, file, &afinfo->seq_ops,
			  sizeof(struct tcp_iter_state));
	if (err < 0)
		return err;
2349

2350
	s = ((struct seq_file *)file->private_data)->private;
L
Linus Torvalds 已提交
2351
	s->family		= afinfo->family;
2352
	s->last_pos 		= 0;
2353 2354 2355
	return 0;
}

2356
int tcp_proc_register(struct net *net, struct tcp_seq_afinfo *afinfo)
L
Linus Torvalds 已提交
2357 2358 2359 2360
{
	int rc = 0;
	struct proc_dir_entry *p;

2361 2362 2363 2364
	afinfo->seq_fops.open		= tcp_seq_open;
	afinfo->seq_fops.read		= seq_read;
	afinfo->seq_fops.llseek		= seq_lseek;
	afinfo->seq_fops.release	= seq_release_net;
2365

2366 2367 2368 2369
	afinfo->seq_ops.start		= tcp_seq_start;
	afinfo->seq_ops.next		= tcp_seq_next;
	afinfo->seq_ops.stop		= tcp_seq_stop;

2370 2371 2372
	p = proc_create_data(afinfo->name, S_IRUGO, net->proc_net,
			     &afinfo->seq_fops, afinfo);
	if (!p)
L
Linus Torvalds 已提交
2373 2374 2375
		rc = -ENOMEM;
	return rc;
}
E
Eric Dumazet 已提交
2376
EXPORT_SYMBOL(tcp_proc_register);
L
Linus Torvalds 已提交
2377

2378
void tcp_proc_unregister(struct net *net, struct tcp_seq_afinfo *afinfo)
L
Linus Torvalds 已提交
2379
{
2380
	proc_net_remove(net, afinfo->name);
L
Linus Torvalds 已提交
2381
}
E
Eric Dumazet 已提交
2382
EXPORT_SYMBOL(tcp_proc_unregister);
L
Linus Torvalds 已提交
2383

2384
static void get_openreq4(const struct sock *sk, const struct request_sock *req,
2385
			 struct seq_file *f, int i, int uid, int *len)
L
Linus Torvalds 已提交
2386
{
2387
	const struct inet_request_sock *ireq = inet_rsk(req);
L
Linus Torvalds 已提交
2388 2389
	int ttd = req->expires - jiffies;

2390
	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
D
Dan Rosenberg 已提交
2391
		" %02X %08X:%08X %02X:%08lX %08X %5d %8d %u %d %pK%n",
L
Linus Torvalds 已提交
2392
		i,
2393
		ireq->loc_addr,
E
Eric Dumazet 已提交
2394
		ntohs(inet_sk(sk)->inet_sport),
2395 2396
		ireq->rmt_addr,
		ntohs(ireq->rmt_port),
L
Linus Torvalds 已提交
2397 2398 2399 2400 2401 2402 2403 2404 2405
		TCP_SYN_RECV,
		0, 0, /* could print option size, but that is af dependent. */
		1,    /* timers active (only the expire timer) */
		jiffies_to_clock_t(ttd),
		req->retrans,
		uid,
		0,  /* non standard timer */
		0, /* open_requests have no inode */
		atomic_read(&sk->sk_refcnt),
2406 2407
		req,
		len);
L
Linus Torvalds 已提交
2408 2409
}

2410
static void get_tcp4_sock(struct sock *sk, struct seq_file *f, int i, int *len)
L
Linus Torvalds 已提交
2411 2412 2413
{
	int timer_active;
	unsigned long timer_expires;
2414
	const struct tcp_sock *tp = tcp_sk(sk);
2415
	const struct inet_connection_sock *icsk = inet_csk(sk);
2416
	const struct inet_sock *inet = inet_sk(sk);
E
Eric Dumazet 已提交
2417 2418 2419 2420
	__be32 dest = inet->inet_daddr;
	__be32 src = inet->inet_rcv_saddr;
	__u16 destp = ntohs(inet->inet_dport);
	__u16 srcp = ntohs(inet->inet_sport);
2421
	int rx_queue;
L
Linus Torvalds 已提交
2422

2423
	if (icsk->icsk_pending == ICSK_TIME_RETRANS) {
L
Linus Torvalds 已提交
2424
		timer_active	= 1;
2425 2426
		timer_expires	= icsk->icsk_timeout;
	} else if (icsk->icsk_pending == ICSK_TIME_PROBE0) {
L
Linus Torvalds 已提交
2427
		timer_active	= 4;
2428
		timer_expires	= icsk->icsk_timeout;
2429
	} else if (timer_pending(&sk->sk_timer)) {
L
Linus Torvalds 已提交
2430
		timer_active	= 2;
2431
		timer_expires	= sk->sk_timer.expires;
L
Linus Torvalds 已提交
2432 2433 2434 2435 2436
	} else {
		timer_active	= 0;
		timer_expires = jiffies;
	}

2437 2438 2439 2440 2441 2442 2443 2444
	if (sk->sk_state == TCP_LISTEN)
		rx_queue = sk->sk_ack_backlog;
	else
		/*
		 * because we dont lock socket, we might find a transient negative value
		 */
		rx_queue = max_t(int, tp->rcv_nxt - tp->copied_seq, 0);

2445
	seq_printf(f, "%4d: %08X:%04X %08X:%04X %02X %08X:%08X %02X:%08lX "
D
Dan Rosenberg 已提交
2446
			"%08X %5d %8d %lu %d %pK %lu %lu %u %u %d%n",
2447
		i, src, srcp, dest, destp, sk->sk_state,
2448
		tp->write_seq - tp->snd_una,
2449
		rx_queue,
L
Linus Torvalds 已提交
2450 2451
		timer_active,
		jiffies_to_clock_t(timer_expires - jiffies),
2452
		icsk->icsk_retransmits,
2453
		sock_i_uid(sk),
2454
		icsk->icsk_probes_out,
2455 2456
		sock_i_ino(sk),
		atomic_read(&sk->sk_refcnt), sk,
2457 2458
		jiffies_to_clock_t(icsk->icsk_rto),
		jiffies_to_clock_t(icsk->icsk_ack.ato),
2459
		(icsk->icsk_ack.quick << 1) | icsk->icsk_ack.pingpong,
L
Linus Torvalds 已提交
2460
		tp->snd_cwnd,
I
Ilpo Järvinen 已提交
2461
		tcp_in_initial_slowstart(tp) ? -1 : tp->snd_ssthresh,
2462
		len);
L
Linus Torvalds 已提交
2463 2464
}

2465
static void get_timewait4_sock(const struct inet_timewait_sock *tw,
2466
			       struct seq_file *f, int i, int *len)
L
Linus Torvalds 已提交
2467
{
2468
	__be32 dest, src;
L
Linus Torvalds 已提交
2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479
	__u16 destp, srcp;
	int ttd = tw->tw_ttd - jiffies;

	if (ttd < 0)
		ttd = 0;

	dest  = tw->tw_daddr;
	src   = tw->tw_rcv_saddr;
	destp = ntohs(tw->tw_dport);
	srcp  = ntohs(tw->tw_sport);

2480
	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
D
Dan Rosenberg 已提交
2481
		" %02X %08X:%08X %02X:%08lX %08X %5d %8d %d %d %pK%n",
L
Linus Torvalds 已提交
2482 2483
		i, src, srcp, dest, destp, tw->tw_substate, 0, 0,
		3, jiffies_to_clock_t(ttd), 0, 0, 0, 0,
2484
		atomic_read(&tw->tw_refcnt), tw, len);
L
Linus Torvalds 已提交
2485 2486 2487 2488 2489 2490
}

#define TMPSZ 150

static int tcp4_seq_show(struct seq_file *seq, void *v)
{
J
Jianjun Kong 已提交
2491
	struct tcp_iter_state *st;
2492
	int len;
L
Linus Torvalds 已提交
2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505

	if (v == SEQ_START_TOKEN) {
		seq_printf(seq, "%-*s\n", TMPSZ - 1,
			   "  sl  local_address rem_address   st tx_queue "
			   "rx_queue tr tm->when retrnsmt   uid  timeout "
			   "inode");
		goto out;
	}
	st = seq->private;

	switch (st->state) {
	case TCP_SEQ_STATE_LISTENING:
	case TCP_SEQ_STATE_ESTABLISHED:
2506
		get_tcp4_sock(v, seq, st->num, &len);
L
Linus Torvalds 已提交
2507 2508
		break;
	case TCP_SEQ_STATE_OPENREQ:
2509
		get_openreq4(st->syn_wait_sk, v, seq, st->num, st->uid, &len);
L
Linus Torvalds 已提交
2510 2511
		break;
	case TCP_SEQ_STATE_TIME_WAIT:
2512
		get_timewait4_sock(v, seq, st->num, &len);
L
Linus Torvalds 已提交
2513 2514
		break;
	}
2515
	seq_printf(seq, "%*s\n", TMPSZ - 1 - len, "");
L
Linus Torvalds 已提交
2516 2517 2518 2519 2520 2521 2522
out:
	return 0;
}

static struct tcp_seq_afinfo tcp4_seq_afinfo = {
	.name		= "tcp",
	.family		= AF_INET,
2523 2524 2525
	.seq_fops	= {
		.owner		= THIS_MODULE,
	},
2526 2527 2528
	.seq_ops	= {
		.show		= tcp4_seq_show,
	},
L
Linus Torvalds 已提交
2529 2530
};

2531
static int __net_init tcp4_proc_init_net(struct net *net)
2532 2533 2534 2535
{
	return tcp_proc_register(net, &tcp4_seq_afinfo);
}

2536
static void __net_exit tcp4_proc_exit_net(struct net *net)
2537 2538 2539 2540 2541 2542 2543 2544 2545
{
	tcp_proc_unregister(net, &tcp4_seq_afinfo);
}

static struct pernet_operations tcp4_net_ops = {
	.init = tcp4_proc_init_net,
	.exit = tcp4_proc_exit_net,
};

L
Linus Torvalds 已提交
2546 2547
int __init tcp4_proc_init(void)
{
2548
	return register_pernet_subsys(&tcp4_net_ops);
L
Linus Torvalds 已提交
2549 2550 2551 2552
}

void tcp4_proc_exit(void)
{
2553
	unregister_pernet_subsys(&tcp4_net_ops);
L
Linus Torvalds 已提交
2554 2555 2556
}
#endif /* CONFIG_PROC_FS */

H
Herbert Xu 已提交
2557 2558
struct sk_buff **tcp4_gro_receive(struct sk_buff **head, struct sk_buff *skb)
{
2559
	const struct iphdr *iph = skb_gro_network_header(skb);
H
Herbert Xu 已提交
2560 2561 2562

	switch (skb->ip_summed) {
	case CHECKSUM_COMPLETE:
2563
		if (!tcp_v4_check(skb_gro_len(skb), iph->saddr, iph->daddr,
H
Herbert Xu 已提交
2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579
				  skb->csum)) {
			skb->ip_summed = CHECKSUM_UNNECESSARY;
			break;
		}

		/* fall through */
	case CHECKSUM_NONE:
		NAPI_GRO_CB(skb)->flush = 1;
		return NULL;
	}

	return tcp_gro_receive(head, skb);
}

int tcp4_gro_complete(struct sk_buff *skb)
{
2580
	const struct iphdr *iph = ip_hdr(skb);
H
Herbert Xu 已提交
2581 2582 2583 2584 2585 2586 2587 2588 2589
	struct tcphdr *th = tcp_hdr(skb);

	th->check = ~tcp_v4_check(skb->len - skb_transport_offset(skb),
				  iph->saddr, iph->daddr, 0);
	skb_shinfo(skb)->gso_type = SKB_GSO_TCPV4;

	return tcp_gro_complete(skb);
}

L
Linus Torvalds 已提交
2590 2591 2592 2593 2594 2595
struct proto tcp_prot = {
	.name			= "TCP",
	.owner			= THIS_MODULE,
	.close			= tcp_close,
	.connect		= tcp_v4_connect,
	.disconnect		= tcp_disconnect,
2596
	.accept			= inet_csk_accept,
L
Linus Torvalds 已提交
2597 2598 2599 2600 2601 2602 2603
	.ioctl			= tcp_ioctl,
	.init			= tcp_v4_init_sock,
	.destroy		= tcp_v4_destroy_sock,
	.shutdown		= tcp_shutdown,
	.setsockopt		= tcp_setsockopt,
	.getsockopt		= tcp_getsockopt,
	.recvmsg		= tcp_recvmsg,
2604 2605
	.sendmsg		= tcp_sendmsg,
	.sendpage		= tcp_sendpage,
L
Linus Torvalds 已提交
2606
	.backlog_rcv		= tcp_v4_do_rcv,
2607 2608 2609
	.hash			= inet_hash,
	.unhash			= inet_unhash,
	.get_port		= inet_csk_get_port,
L
Linus Torvalds 已提交
2610 2611
	.enter_memory_pressure	= tcp_enter_memory_pressure,
	.sockets_allocated	= &tcp_sockets_allocated,
2612
	.orphan_count		= &tcp_orphan_count,
L
Linus Torvalds 已提交
2613 2614 2615 2616 2617 2618 2619
	.memory_allocated	= &tcp_memory_allocated,
	.memory_pressure	= &tcp_memory_pressure,
	.sysctl_mem		= sysctl_tcp_mem,
	.sysctl_wmem		= sysctl_tcp_wmem,
	.sysctl_rmem		= sysctl_tcp_rmem,
	.max_header		= MAX_TCP_HEADER,
	.obj_size		= sizeof(struct tcp_sock),
2620
	.slab_flags		= SLAB_DESTROY_BY_RCU,
2621
	.twsk_prot		= &tcp_timewait_sock_ops,
2622
	.rsk_prot		= &tcp_request_sock_ops,
2623
	.h.hashinfo		= &tcp_hashinfo,
2624
	.no_autobind		= true,
2625 2626 2627 2628
#ifdef CONFIG_COMPAT
	.compat_setsockopt	= compat_tcp_setsockopt,
	.compat_getsockopt	= compat_tcp_getsockopt,
#endif
L
Linus Torvalds 已提交
2629
};
E
Eric Dumazet 已提交
2630
EXPORT_SYMBOL(tcp_prot);
L
Linus Torvalds 已提交
2631

2632 2633 2634 2635 2636 2637 2638 2639 2640 2641

static int __net_init tcp_sk_init(struct net *net)
{
	return inet_ctl_sock_create(&net->ipv4.tcp_sock,
				    PF_INET, SOCK_RAW, IPPROTO_TCP, net);
}

static void __net_exit tcp_sk_exit(struct net *net)
{
	inet_ctl_sock_destroy(net->ipv4.tcp_sock);
E
Eric W. Biederman 已提交
2642 2643 2644 2645 2646
}

static void __net_exit tcp_sk_exit_batch(struct list_head *net_exit_list)
{
	inet_twsk_purge(&tcp_hashinfo, &tcp_death_row, AF_INET);
2647 2648 2649
}

static struct pernet_operations __net_initdata tcp_sk_ops = {
E
Eric W. Biederman 已提交
2650 2651 2652
       .init	   = tcp_sk_init,
       .exit	   = tcp_sk_exit,
       .exit_batch = tcp_sk_exit_batch,
2653 2654
};

2655
void __init tcp_v4_init(void)
L
Linus Torvalds 已提交
2656
{
2657
	inet_hashinfo_init(&tcp_hashinfo);
2658
	if (register_pernet_subsys(&tcp_sk_ops))
L
Linus Torvalds 已提交
2659 2660
		panic("Failed to create the TCP control socket.\n");
}