1 // SPDX-License-Identifier: GPL-2.0-or-later 2 /* 3 * INET An implementation of the TCP/IP protocol suite for the LINUX 4 * operating system. INET is implemented using the BSD Socket 5 * interface as the means of communication with the user level. 6 * 7 * Implementation of the Transmission Control Protocol(TCP). 8 * 9 * IPv4 specific functions 10 * 11 * code split from: 12 * linux/ipv4/tcp.c 13 * linux/ipv4/tcp_input.c 14 * linux/ipv4/tcp_output.c 15 * 16 * See tcp.c for author information 17 */ 18 19 /* 20 * Changes: 21 * David S. Miller : New socket lookup architecture. 22 * This code is dedicated to John Dyson. 23 * David S. Miller : Change semantics of established hash, 24 * half is devoted to TIME_WAIT sockets 25 * and the rest go in the other half. 26 * Andi Kleen : Add support for syncookies and fixed 27 * some bugs: ip options weren't passed to 28 * the TCP layer, missed a check for an 29 * ACK bit. 30 * Andi Kleen : Implemented fast path mtu discovery. 31 * Fixed many serious bugs in the 32 * request_sock handling and moved 33 * most of it into the af independent code. 34 * Added tail drop and some other bugfixes. 35 * Added new listen semantics. 36 * Mike McLagan : Routing by source 37 * Juan Jose Ciarlante: ip_dynaddr bits 38 * Andi Kleen: various fixes. 39 * Vitaly E. Lavrov : Transparent proxy revived after year 40 * coma. 41 * Andi Kleen : Fix new listen. 42 * Andi Kleen : Fix accept error reporting. 43 * YOSHIFUJI Hideaki @USAGI and: Support IPV6_V6ONLY socket option, which 44 * Alexey Kuznetsov allow both IPv4 and IPv6 sockets to bind 45 * a single port at the same time. 46 */ 47 48 #define pr_fmt(fmt) "TCP: " fmt 49 50 #include <linux/bottom_half.h> 51 #include <linux/types.h> 52 #include <linux/fcntl.h> 53 #include <linux/module.h> 54 #include <linux/random.h> 55 #include <linux/cache.h> 56 #include <linux/jhash.h> 57 #include <linux/init.h> 58 #include <linux/times.h> 59 #include <linux/slab.h> 60 #include <linux/sched.h> 61 62 #include <net/net_namespace.h> 63 #include <net/icmp.h> 64 #include <net/inet_hashtables.h> 65 #include <net/tcp.h> 66 #include <net/transp_v6.h> 67 #include <net/ipv6.h> 68 #include <net/inet_common.h> 69 #include <net/timewait_sock.h> 70 #include <net/xfrm.h> 71 #include <net/secure_seq.h> 72 #include <net/busy_poll.h> 73 74 #include <linux/inet.h> 75 #include <linux/ipv6.h> 76 #include <linux/stddef.h> 77 #include <linux/proc_fs.h> 78 #include <linux/seq_file.h> 79 #include <linux/inetdevice.h> 80 #include <linux/btf_ids.h> 81 82 #include <crypto/hash.h> 83 #include <linux/scatterlist.h> 84 85 #include <trace/events/tcp.h> 86 87 #ifdef CONFIG_TCP_MD5SIG 88 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key, 89 __be32 daddr, __be32 saddr, const struct tcphdr *th); 90 #endif 91 92 struct inet_hashinfo tcp_hashinfo; 93 EXPORT_SYMBOL(tcp_hashinfo); 94 95 static DEFINE_PER_CPU(struct sock *, ipv4_tcp_sk); 96 97 static u32 tcp_v4_init_seq(const struct sk_buff *skb) 98 { 99 return secure_tcp_seq(ip_hdr(skb)->daddr, 100 ip_hdr(skb)->saddr, 101 tcp_hdr(skb)->dest, 102 tcp_hdr(skb)->source); 103 } 104 105 static u32 tcp_v4_init_ts_off(const struct net *net, const struct sk_buff *skb) 106 { 107 return secure_tcp_ts_off(net, ip_hdr(skb)->daddr, ip_hdr(skb)->saddr); 108 } 109 110 int tcp_twsk_unique(struct sock *sk, struct sock *sktw, void *twp) 111 { 112 int reuse = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_tw_reuse); 113 const struct inet_timewait_sock *tw = inet_twsk(sktw); 114 const struct tcp_timewait_sock *tcptw = tcp_twsk(sktw); 115 struct tcp_sock *tp = tcp_sk(sk); 116 117 if (reuse == 2) { 118 /* Still does not detect *everything* that goes through 119 * lo, since we require a loopback src or dst address 120 * or direct binding to 'lo' interface. 121 */ 122 bool loopback = false; 123 if (tw->tw_bound_dev_if == LOOPBACK_IFINDEX) 124 loopback = true; 125 #if IS_ENABLED(CONFIG_IPV6) 126 if (tw->tw_family == AF_INET6) { 127 if (ipv6_addr_loopback(&tw->tw_v6_daddr) || 128 ipv6_addr_v4mapped_loopback(&tw->tw_v6_daddr) || 129 ipv6_addr_loopback(&tw->tw_v6_rcv_saddr) || 130 ipv6_addr_v4mapped_loopback(&tw->tw_v6_rcv_saddr)) 131 loopback = true; 132 } else 133 #endif 134 { 135 if (ipv4_is_loopback(tw->tw_daddr) || 136 ipv4_is_loopback(tw->tw_rcv_saddr)) 137 loopback = true; 138 } 139 if (!loopback) 140 reuse = 0; 141 } 142 143 /* With PAWS, it is safe from the viewpoint 144 of data integrity. Even without PAWS it is safe provided sequence 145 spaces do not overlap i.e. at data rates <= 80Mbit/sec. 146 147 Actually, the idea is close to VJ's one, only timestamp cache is 148 held not per host, but per port pair and TW bucket is used as state 149 holder. 150 151 If TW bucket has been already destroyed we fall back to VJ's scheme 152 and use initial timestamp retrieved from peer table. 153 */ 154 if (tcptw->tw_ts_recent_stamp && 155 (!twp || (reuse && time_after32(ktime_get_seconds(), 156 tcptw->tw_ts_recent_stamp)))) { 157 /* In case of repair and re-using TIME-WAIT sockets we still 158 * want to be sure that it is safe as above but honor the 159 * sequence numbers and time stamps set as part of the repair 160 * process. 161 * 162 * Without this check re-using a TIME-WAIT socket with TCP 163 * repair would accumulate a -1 on the repair assigned 164 * sequence number. The first time it is reused the sequence 165 * is -1, the second time -2, etc. This fixes that issue 166 * without appearing to create any others. 167 */ 168 if (likely(!tp->repair)) { 169 u32 seq = tcptw->tw_snd_nxt + 65535 + 2; 170 171 if (!seq) 172 seq = 1; 173 WRITE_ONCE(tp->write_seq, seq); 174 tp->rx_opt.ts_recent = tcptw->tw_ts_recent; 175 tp->rx_opt.ts_recent_stamp = tcptw->tw_ts_recent_stamp; 176 } 177 sock_hold(sktw); 178 return 1; 179 } 180 181 return 0; 182 } 183 EXPORT_SYMBOL_GPL(tcp_twsk_unique); 184 185 static int tcp_v4_pre_connect(struct sock *sk, struct sockaddr *uaddr, 186 int addr_len) 187 { 188 /* This check is replicated from tcp_v4_connect() and intended to 189 * prevent BPF program called below from accessing bytes that are out 190 * of the bound specified by user in addr_len. 191 */ 192 if (addr_len < sizeof(struct sockaddr_in)) 193 return -EINVAL; 194 195 sock_owned_by_me(sk); 196 197 return BPF_CGROUP_RUN_PROG_INET4_CONNECT(sk, uaddr); 198 } 199 200 /* This will initiate an outgoing connection. */ 201 int tcp_v4_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len) 202 { 203 struct sockaddr_in *usin = (struct sockaddr_in *)uaddr; 204 struct inet_timewait_death_row *tcp_death_row; 205 struct inet_sock *inet = inet_sk(sk); 206 struct tcp_sock *tp = tcp_sk(sk); 207 struct ip_options_rcu *inet_opt; 208 struct net *net = sock_net(sk); 209 __be16 orig_sport, orig_dport; 210 __be32 daddr, nexthop; 211 struct flowi4 *fl4; 212 struct rtable *rt; 213 int err; 214 215 if (addr_len < sizeof(struct sockaddr_in)) 216 return -EINVAL; 217 218 if (usin->sin_family != AF_INET) 219 return -EAFNOSUPPORT; 220 221 nexthop = daddr = usin->sin_addr.s_addr; 222 inet_opt = rcu_dereference_protected(inet->inet_opt, 223 lockdep_sock_is_held(sk)); 224 if (inet_opt && inet_opt->opt.srr) { 225 if (!daddr) 226 return -EINVAL; 227 nexthop = inet_opt->opt.faddr; 228 } 229 230 orig_sport = inet->inet_sport; 231 orig_dport = usin->sin_port; 232 fl4 = &inet->cork.fl.u.ip4; 233 rt = ip_route_connect(fl4, nexthop, inet->inet_saddr, 234 sk->sk_bound_dev_if, IPPROTO_TCP, orig_sport, 235 orig_dport, sk); 236 if (IS_ERR(rt)) { 237 err = PTR_ERR(rt); 238 if (err == -ENETUNREACH) 239 IP_INC_STATS(net, IPSTATS_MIB_OUTNOROUTES); 240 return err; 241 } 242 243 if (rt->rt_flags & (RTCF_MULTICAST | RTCF_BROADCAST)) { 244 ip_rt_put(rt); 245 return -ENETUNREACH; 246 } 247 248 if (!inet_opt || !inet_opt->opt.srr) 249 daddr = fl4->daddr; 250 251 tcp_death_row = &sock_net(sk)->ipv4.tcp_death_row; 252 253 if (!inet->inet_saddr) { 254 err = inet_bhash2_update_saddr(sk, &fl4->saddr, AF_INET); 255 if (err) { 256 ip_rt_put(rt); 257 return err; 258 } 259 } else { 260 sk_rcv_saddr_set(sk, inet->inet_saddr); 261 } 262 263 if (tp->rx_opt.ts_recent_stamp && inet->inet_daddr != daddr) { 264 /* Reset inherited state */ 265 tp->rx_opt.ts_recent = 0; 266 tp->rx_opt.ts_recent_stamp = 0; 267 if (likely(!tp->repair)) 268 WRITE_ONCE(tp->write_seq, 0); 269 } 270 271 inet->inet_dport = usin->sin_port; 272 sk_daddr_set(sk, daddr); 273 274 inet_csk(sk)->icsk_ext_hdr_len = 0; 275 if (inet_opt) 276 inet_csk(sk)->icsk_ext_hdr_len = inet_opt->opt.optlen; 277 278 tp->rx_opt.mss_clamp = TCP_MSS_DEFAULT; 279 280 /* Socket identity is still unknown (sport may be zero). 281 * However we set state to SYN-SENT and not releasing socket 282 * lock select source port, enter ourselves into the hash tables and 283 * complete initialization after this. 284 */ 285 tcp_set_state(sk, TCP_SYN_SENT); 286 err = inet_hash_connect(tcp_death_row, sk); 287 if (err) 288 goto failure; 289 290 sk_set_txhash(sk); 291 292 rt = ip_route_newports(fl4, rt, orig_sport, orig_dport, 293 inet->inet_sport, inet->inet_dport, sk); 294 if (IS_ERR(rt)) { 295 err = PTR_ERR(rt); 296 rt = NULL; 297 goto failure; 298 } 299 /* OK, now commit destination to socket. */ 300 sk->sk_gso_type = SKB_GSO_TCPV4; 301 sk_setup_caps(sk, &rt->dst); 302 rt = NULL; 303 304 if (likely(!tp->repair)) { 305 if (!tp->write_seq) 306 WRITE_ONCE(tp->write_seq, 307 secure_tcp_seq(inet->inet_saddr, 308 inet->inet_daddr, 309 inet->inet_sport, 310 usin->sin_port)); 311 WRITE_ONCE(tp->tsoffset, 312 secure_tcp_ts_off(net, inet->inet_saddr, 313 inet->inet_daddr)); 314 } 315 316 inet->inet_id = get_random_u16(); 317 318 if (tcp_fastopen_defer_connect(sk, &err)) 319 return err; 320 if (err) 321 goto failure; 322 323 err = tcp_connect(sk); 324 325 if (err) 326 goto failure; 327 328 return 0; 329 330 failure: 331 /* 332 * This unhashes the socket and releases the local port, 333 * if necessary. 334 */ 335 tcp_set_state(sk, TCP_CLOSE); 336 inet_bhash2_reset_saddr(sk); 337 ip_rt_put(rt); 338 sk->sk_route_caps = 0; 339 inet->inet_dport = 0; 340 return err; 341 } 342 EXPORT_SYMBOL(tcp_v4_connect); 343 344 /* 345 * This routine reacts to ICMP_FRAG_NEEDED mtu indications as defined in RFC1191. 346 * It can be called through tcp_release_cb() if socket was owned by user 347 * at the time tcp_v4_err() was called to handle ICMP message. 348 */ 349 void tcp_v4_mtu_reduced(struct sock *sk) 350 { 351 struct inet_sock *inet = inet_sk(sk); 352 struct dst_entry *dst; 353 u32 mtu; 354 355 if ((1 << sk->sk_state) & (TCPF_LISTEN | TCPF_CLOSE)) 356 return; 357 mtu = READ_ONCE(tcp_sk(sk)->mtu_info); 358 dst = inet_csk_update_pmtu(sk, mtu); 359 if (!dst) 360 return; 361 362 /* Something is about to be wrong... Remember soft error 363 * for the case, if this connection will not able to recover. 364 */ 365 if (mtu < dst_mtu(dst) && ip_dont_fragment(sk, dst)) 366 WRITE_ONCE(sk->sk_err_soft, EMSGSIZE); 367 368 mtu = dst_mtu(dst); 369 370 if (inet->pmtudisc != IP_PMTUDISC_DONT && 371 ip_sk_accept_pmtu(sk) && 372 inet_csk(sk)->icsk_pmtu_cookie > mtu) { 373 tcp_sync_mss(sk, mtu); 374 375 /* Resend the TCP packet because it's 376 * clear that the old packet has been 377 * dropped. This is the new "fast" path mtu 378 * discovery. 379 */ 380 tcp_simple_retransmit(sk); 381 } /* else let the usual retransmit timer handle it */ 382 } 383 EXPORT_SYMBOL(tcp_v4_mtu_reduced); 384 385 static void do_redirect(struct sk_buff *skb, struct sock *sk) 386 { 387 struct dst_entry *dst = __sk_dst_check(sk, 0); 388 389 if (dst) 390 dst->ops->redirect(dst, sk, skb); 391 } 392 393 394 /* handle ICMP messages on TCP_NEW_SYN_RECV request sockets */ 395 void tcp_req_err(struct sock *sk, u32 seq, bool abort) 396 { 397 struct request_sock *req = inet_reqsk(sk); 398 struct net *net = sock_net(sk); 399 400 /* ICMPs are not backlogged, hence we cannot get 401 * an established socket here. 402 */ 403 if (seq != tcp_rsk(req)->snt_isn) { 404 __NET_INC_STATS(net, LINUX_MIB_OUTOFWINDOWICMPS); 405 } else if (abort) { 406 /* 407 * Still in SYN_RECV, just remove it silently. 408 * There is no good way to pass the error to the newly 409 * created socket, and POSIX does not want network 410 * errors returned from accept(). 411 */ 412 inet_csk_reqsk_queue_drop(req->rsk_listener, req); 413 tcp_listendrop(req->rsk_listener); 414 } 415 reqsk_put(req); 416 } 417 EXPORT_SYMBOL(tcp_req_err); 418 419 /* TCP-LD (RFC 6069) logic */ 420 void tcp_ld_RTO_revert(struct sock *sk, u32 seq) 421 { 422 struct inet_connection_sock *icsk = inet_csk(sk); 423 struct tcp_sock *tp = tcp_sk(sk); 424 struct sk_buff *skb; 425 s32 remaining; 426 u32 delta_us; 427 428 if (sock_owned_by_user(sk)) 429 return; 430 431 if (seq != tp->snd_una || !icsk->icsk_retransmits || 432 !icsk->icsk_backoff) 433 return; 434 435 skb = tcp_rtx_queue_head(sk); 436 if (WARN_ON_ONCE(!skb)) 437 return; 438 439 icsk->icsk_backoff--; 440 icsk->icsk_rto = tp->srtt_us ? __tcp_set_rto(tp) : TCP_TIMEOUT_INIT; 441 icsk->icsk_rto = inet_csk_rto_backoff(icsk, TCP_RTO_MAX); 442 443 tcp_mstamp_refresh(tp); 444 delta_us = (u32)(tp->tcp_mstamp - tcp_skb_timestamp_us(skb)); 445 remaining = icsk->icsk_rto - usecs_to_jiffies(delta_us); 446 447 if (remaining > 0) { 448 inet_csk_reset_xmit_timer(sk, ICSK_TIME_RETRANS, 449 remaining, TCP_RTO_MAX); 450 } else { 451 /* RTO revert clocked out retransmission. 452 * Will retransmit now. 453 */ 454 tcp_retransmit_timer(sk); 455 } 456 } 457 EXPORT_SYMBOL(tcp_ld_RTO_revert); 458 459 /* 460 * This routine is called by the ICMP module when it gets some 461 * sort of error condition. If err < 0 then the socket should 462 * be closed and the error returned to the user. If err > 0 463 * it's just the icmp type << 8 | icmp code. After adjustment 464 * header points to the first 8 bytes of the tcp header. We need 465 * to find the appropriate port. 466 * 467 * The locking strategy used here is very "optimistic". When 468 * someone else accesses the socket the ICMP is just dropped 469 * and for some paths there is no check at all. 470 * A more general error queue to queue errors for later handling 471 * is probably better. 472 * 473 */ 474 475 int tcp_v4_err(struct sk_buff *skb, u32 info) 476 { 477 const struct iphdr *iph = (const struct iphdr *)skb->data; 478 struct tcphdr *th = (struct tcphdr *)(skb->data + (iph->ihl << 2)); 479 struct tcp_sock *tp; 480 struct inet_sock *inet; 481 const int type = icmp_hdr(skb)->type; 482 const int code = icmp_hdr(skb)->code; 483 struct sock *sk; 484 struct request_sock *fastopen; 485 u32 seq, snd_una; 486 int err; 487 struct net *net = dev_net(skb->dev); 488 489 sk = __inet_lookup_established(net, net->ipv4.tcp_death_row.hashinfo, 490 iph->daddr, th->dest, iph->saddr, 491 ntohs(th->source), inet_iif(skb), 0); 492 if (!sk) { 493 __ICMP_INC_STATS(net, ICMP_MIB_INERRORS); 494 return -ENOENT; 495 } 496 if (sk->sk_state == TCP_TIME_WAIT) { 497 inet_twsk_put(inet_twsk(sk)); 498 return 0; 499 } 500 seq = ntohl(th->seq); 501 if (sk->sk_state == TCP_NEW_SYN_RECV) { 502 tcp_req_err(sk, seq, type == ICMP_PARAMETERPROB || 503 type == ICMP_TIME_EXCEEDED || 504 (type == ICMP_DEST_UNREACH && 505 (code == ICMP_NET_UNREACH || 506 code == ICMP_HOST_UNREACH))); 507 return 0; 508 } 509 510 bh_lock_sock(sk); 511 /* If too many ICMPs get dropped on busy 512 * servers this needs to be solved differently. 513 * We do take care of PMTU discovery (RFC1191) special case : 514 * we can receive locally generated ICMP messages while socket is held. 515 */ 516 if (sock_owned_by_user(sk)) { 517 if (!(type == ICMP_DEST_UNREACH && code == ICMP_FRAG_NEEDED)) 518 __NET_INC_STATS(net, LINUX_MIB_LOCKDROPPEDICMPS); 519 } 520 if (sk->sk_state == TCP_CLOSE) 521 goto out; 522 523 if (static_branch_unlikely(&ip4_min_ttl)) { 524 /* min_ttl can be changed concurrently from do_ip_setsockopt() */ 525 if (unlikely(iph->ttl < READ_ONCE(inet_sk(sk)->min_ttl))) { 526 __NET_INC_STATS(net, LINUX_MIB_TCPMINTTLDROP); 527 goto out; 528 } 529 } 530 531 tp = tcp_sk(sk); 532 /* XXX (TFO) - tp->snd_una should be ISN (tcp_create_openreq_child() */ 533 fastopen = rcu_dereference(tp->fastopen_rsk); 534 snd_una = fastopen ? tcp_rsk(fastopen)->snt_isn : tp->snd_una; 535 if (sk->sk_state != TCP_LISTEN && 536 !between(seq, snd_una, tp->snd_nxt)) { 537 __NET_INC_STATS(net, LINUX_MIB_OUTOFWINDOWICMPS); 538 goto out; 539 } 540 541 switch (type) { 542 case ICMP_REDIRECT: 543 if (!sock_owned_by_user(sk)) 544 do_redirect(skb, sk); 545 goto out; 546 case ICMP_SOURCE_QUENCH: 547 /* Just silently ignore these. */ 548 goto out; 549 case ICMP_PARAMETERPROB: 550 err = EPROTO; 551 break; 552 case ICMP_DEST_UNREACH: 553 if (code > NR_ICMP_UNREACH) 554 goto out; 555 556 if (code == ICMP_FRAG_NEEDED) { /* PMTU discovery (RFC1191) */ 557 /* We are not interested in TCP_LISTEN and open_requests 558 * (SYN-ACKs send out by Linux are always <576bytes so 559 * they should go through unfragmented). 560 */ 561 if (sk->sk_state == TCP_LISTEN) 562 goto out; 563 564 WRITE_ONCE(tp->mtu_info, info); 565 if (!sock_owned_by_user(sk)) { 566 tcp_v4_mtu_reduced(sk); 567 } else { 568 if (!test_and_set_bit(TCP_MTU_REDUCED_DEFERRED, &sk->sk_tsq_flags)) 569 sock_hold(sk); 570 } 571 goto out; 572 } 573 574 err = icmp_err_convert[code].errno; 575 /* check if this ICMP message allows revert of backoff. 576 * (see RFC 6069) 577 */ 578 if (!fastopen && 579 (code == ICMP_NET_UNREACH || code == ICMP_HOST_UNREACH)) 580 tcp_ld_RTO_revert(sk, seq); 581 break; 582 case ICMP_TIME_EXCEEDED: 583 err = EHOSTUNREACH; 584 break; 585 default: 586 goto out; 587 } 588 589 switch (sk->sk_state) { 590 case TCP_SYN_SENT: 591 case TCP_SYN_RECV: 592 /* Only in fast or simultaneous open. If a fast open socket is 593 * already accepted it is treated as a connected one below. 594 */ 595 if (fastopen && !fastopen->sk) 596 break; 597 598 ip_icmp_error(sk, skb, err, th->dest, info, (u8 *)th); 599 600 if (!sock_owned_by_user(sk)) { 601 WRITE_ONCE(sk->sk_err, err); 602 603 sk_error_report(sk); 604 605 tcp_done(sk); 606 } else { 607 WRITE_ONCE(sk->sk_err_soft, err); 608 } 609 goto out; 610 } 611 612 /* If we've already connected we will keep trying 613 * until we time out, or the user gives up. 614 * 615 * rfc1122 4.2.3.9 allows to consider as hard errors 616 * only PROTO_UNREACH and PORT_UNREACH (well, FRAG_FAILED too, 617 * but it is obsoleted by pmtu discovery). 618 * 619 * Note, that in modern internet, where routing is unreliable 620 * and in each dark corner broken firewalls sit, sending random 621 * errors ordered by their masters even this two messages finally lose 622 * their original sense (even Linux sends invalid PORT_UNREACHs) 623 * 624 * Now we are in compliance with RFCs. 625 * --ANK (980905) 626 */ 627 628 inet = inet_sk(sk); 629 if (!sock_owned_by_user(sk) && inet->recverr) { 630 WRITE_ONCE(sk->sk_err, err); 631 sk_error_report(sk); 632 } else { /* Only an error on timeout */ 633 WRITE_ONCE(sk->sk_err_soft, err); 634 } 635 636 out: 637 bh_unlock_sock(sk); 638 sock_put(sk); 639 return 0; 640 } 641 642 void __tcp_v4_send_check(struct sk_buff *skb, __be32 saddr, __be32 daddr) 643 { 644 struct tcphdr *th = tcp_hdr(skb); 645 646 th->check = ~tcp_v4_check(skb->len, saddr, daddr, 0); 647 skb->csum_start = skb_transport_header(skb) - skb->head; 648 skb->csum_offset = offsetof(struct tcphdr, check); 649 } 650 651 /* This routine computes an IPv4 TCP checksum. */ 652 void tcp_v4_send_check(struct sock *sk, struct sk_buff *skb) 653 { 654 const struct inet_sock *inet = inet_sk(sk); 655 656 __tcp_v4_send_check(skb, inet->inet_saddr, inet->inet_daddr); 657 } 658 EXPORT_SYMBOL(tcp_v4_send_check); 659 660 /* 661 * This routine will send an RST to the other tcp. 662 * 663 * Someone asks: why I NEVER use socket parameters (TOS, TTL etc.) 664 * for reset. 665 * Answer: if a packet caused RST, it is not for a socket 666 * existing in our system, if it is matched to a socket, 667 * it is just duplicate segment or bug in other side's TCP. 668 * So that we build reply only basing on parameters 669 * arrived with segment. 670 * Exception: precedence violation. We do not implement it in any case. 671 */ 672 673 #ifdef CONFIG_TCP_MD5SIG 674 #define OPTION_BYTES TCPOLEN_MD5SIG_ALIGNED 675 #else 676 #define OPTION_BYTES sizeof(__be32) 677 #endif 678 679 static void tcp_v4_send_reset(const struct sock *sk, struct sk_buff *skb) 680 { 681 const struct tcphdr *th = tcp_hdr(skb); 682 struct { 683 struct tcphdr th; 684 __be32 opt[OPTION_BYTES / sizeof(__be32)]; 685 } rep; 686 struct ip_reply_arg arg; 687 #ifdef CONFIG_TCP_MD5SIG 688 struct tcp_md5sig_key *key = NULL; 689 const __u8 *hash_location = NULL; 690 unsigned char newhash[16]; 691 int genhash; 692 struct sock *sk1 = NULL; 693 #endif 694 u64 transmit_time = 0; 695 struct sock *ctl_sk; 696 struct net *net; 697 u32 txhash = 0; 698 699 /* Never send a reset in response to a reset. */ 700 if (th->rst) 701 return; 702 703 /* If sk not NULL, it means we did a successful lookup and incoming 704 * route had to be correct. prequeue might have dropped our dst. 705 */ 706 if (!sk && skb_rtable(skb)->rt_type != RTN_LOCAL) 707 return; 708 709 /* Swap the send and the receive. */ 710 memset(&rep, 0, sizeof(rep)); 711 rep.th.dest = th->source; 712 rep.th.source = th->dest; 713 rep.th.doff = sizeof(struct tcphdr) / 4; 714 rep.th.rst = 1; 715 716 if (th->ack) { 717 rep.th.seq = th->ack_seq; 718 } else { 719 rep.th.ack = 1; 720 rep.th.ack_seq = htonl(ntohl(th->seq) + th->syn + th->fin + 721 skb->len - (th->doff << 2)); 722 } 723 724 memset(&arg, 0, sizeof(arg)); 725 arg.iov[0].iov_base = (unsigned char *)&rep; 726 arg.iov[0].iov_len = sizeof(rep.th); 727 728 net = sk ? sock_net(sk) : dev_net(skb_dst(skb)->dev); 729 #ifdef CONFIG_TCP_MD5SIG 730 rcu_read_lock(); 731 hash_location = tcp_parse_md5sig_option(th); 732 if (sk && sk_fullsock(sk)) { 733 const union tcp_md5_addr *addr; 734 int l3index; 735 736 /* sdif set, means packet ingressed via a device 737 * in an L3 domain and inet_iif is set to it. 738 */ 739 l3index = tcp_v4_sdif(skb) ? inet_iif(skb) : 0; 740 addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr; 741 key = tcp_md5_do_lookup(sk, l3index, addr, AF_INET); 742 } else if (hash_location) { 743 const union tcp_md5_addr *addr; 744 int sdif = tcp_v4_sdif(skb); 745 int dif = inet_iif(skb); 746 int l3index; 747 748 /* 749 * active side is lost. Try to find listening socket through 750 * source port, and then find md5 key through listening socket. 751 * we are not loose security here: 752 * Incoming packet is checked with md5 hash with finding key, 753 * no RST generated if md5 hash doesn't match. 754 */ 755 sk1 = __inet_lookup_listener(net, net->ipv4.tcp_death_row.hashinfo, 756 NULL, 0, ip_hdr(skb)->saddr, 757 th->source, ip_hdr(skb)->daddr, 758 ntohs(th->source), dif, sdif); 759 /* don't send rst if it can't find key */ 760 if (!sk1) 761 goto out; 762 763 /* sdif set, means packet ingressed via a device 764 * in an L3 domain and dif is set to it. 765 */ 766 l3index = sdif ? dif : 0; 767 addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr; 768 key = tcp_md5_do_lookup(sk1, l3index, addr, AF_INET); 769 if (!key) 770 goto out; 771 772 773 genhash = tcp_v4_md5_hash_skb(newhash, key, NULL, skb); 774 if (genhash || memcmp(hash_location, newhash, 16) != 0) 775 goto out; 776 777 } 778 779 if (key) { 780 rep.opt[0] = htonl((TCPOPT_NOP << 24) | 781 (TCPOPT_NOP << 16) | 782 (TCPOPT_MD5SIG << 8) | 783 TCPOLEN_MD5SIG); 784 /* Update length and the length the header thinks exists */ 785 arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED; 786 rep.th.doff = arg.iov[0].iov_len / 4; 787 788 tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[1], 789 key, ip_hdr(skb)->saddr, 790 ip_hdr(skb)->daddr, &rep.th); 791 } 792 #endif 793 /* Can't co-exist with TCPMD5, hence check rep.opt[0] */ 794 if (rep.opt[0] == 0) { 795 __be32 mrst = mptcp_reset_option(skb); 796 797 if (mrst) { 798 rep.opt[0] = mrst; 799 arg.iov[0].iov_len += sizeof(mrst); 800 rep.th.doff = arg.iov[0].iov_len / 4; 801 } 802 } 803 804 arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr, 805 ip_hdr(skb)->saddr, /* XXX */ 806 arg.iov[0].iov_len, IPPROTO_TCP, 0); 807 arg.csumoffset = offsetof(struct tcphdr, check) / 2; 808 arg.flags = (sk && inet_sk_transparent(sk)) ? IP_REPLY_ARG_NOSRCCHECK : 0; 809 810 /* When socket is gone, all binding information is lost. 811 * routing might fail in this case. No choice here, if we choose to force 812 * input interface, we will misroute in case of asymmetric route. 813 */ 814 if (sk) { 815 arg.bound_dev_if = sk->sk_bound_dev_if; 816 if (sk_fullsock(sk)) 817 trace_tcp_send_reset(sk, skb); 818 } 819 820 BUILD_BUG_ON(offsetof(struct sock, sk_bound_dev_if) != 821 offsetof(struct inet_timewait_sock, tw_bound_dev_if)); 822 823 arg.tos = ip_hdr(skb)->tos; 824 arg.uid = sock_net_uid(net, sk && sk_fullsock(sk) ? sk : NULL); 825 local_bh_disable(); 826 ctl_sk = this_cpu_read(ipv4_tcp_sk); 827 sock_net_set(ctl_sk, net); 828 if (sk) { 829 ctl_sk->sk_mark = (sk->sk_state == TCP_TIME_WAIT) ? 830 inet_twsk(sk)->tw_mark : sk->sk_mark; 831 ctl_sk->sk_priority = (sk->sk_state == TCP_TIME_WAIT) ? 832 inet_twsk(sk)->tw_priority : sk->sk_priority; 833 transmit_time = tcp_transmit_time(sk); 834 xfrm_sk_clone_policy(ctl_sk, sk); 835 txhash = (sk->sk_state == TCP_TIME_WAIT) ? 836 inet_twsk(sk)->tw_txhash : sk->sk_txhash; 837 } else { 838 ctl_sk->sk_mark = 0; 839 ctl_sk->sk_priority = 0; 840 } 841 ip_send_unicast_reply(ctl_sk, 842 skb, &TCP_SKB_CB(skb)->header.h4.opt, 843 ip_hdr(skb)->saddr, ip_hdr(skb)->daddr, 844 &arg, arg.iov[0].iov_len, 845 transmit_time, txhash); 846 847 xfrm_sk_free_policy(ctl_sk); 848 sock_net_set(ctl_sk, &init_net); 849 __TCP_INC_STATS(net, TCP_MIB_OUTSEGS); 850 __TCP_INC_STATS(net, TCP_MIB_OUTRSTS); 851 local_bh_enable(); 852 853 #ifdef CONFIG_TCP_MD5SIG 854 out: 855 rcu_read_unlock(); 856 #endif 857 } 858 859 /* The code following below sending ACKs in SYN-RECV and TIME-WAIT states 860 outside socket context is ugly, certainly. What can I do? 861 */ 862 863 static void tcp_v4_send_ack(const struct sock *sk, 864 struct sk_buff *skb, u32 seq, u32 ack, 865 u32 win, u32 tsval, u32 tsecr, int oif, 866 struct tcp_md5sig_key *key, 867 int reply_flags, u8 tos, u32 txhash) 868 { 869 const struct tcphdr *th = tcp_hdr(skb); 870 struct { 871 struct tcphdr th; 872 __be32 opt[(TCPOLEN_TSTAMP_ALIGNED >> 2) 873 #ifdef CONFIG_TCP_MD5SIG 874 + (TCPOLEN_MD5SIG_ALIGNED >> 2) 875 #endif 876 ]; 877 } rep; 878 struct net *net = sock_net(sk); 879 struct ip_reply_arg arg; 880 struct sock *ctl_sk; 881 u64 transmit_time; 882 883 memset(&rep.th, 0, sizeof(struct tcphdr)); 884 memset(&arg, 0, sizeof(arg)); 885 886 arg.iov[0].iov_base = (unsigned char *)&rep; 887 arg.iov[0].iov_len = sizeof(rep.th); 888 if (tsecr) { 889 rep.opt[0] = htonl((TCPOPT_NOP << 24) | (TCPOPT_NOP << 16) | 890 (TCPOPT_TIMESTAMP << 8) | 891 TCPOLEN_TIMESTAMP); 892 rep.opt[1] = htonl(tsval); 893 rep.opt[2] = htonl(tsecr); 894 arg.iov[0].iov_len += TCPOLEN_TSTAMP_ALIGNED; 895 } 896 897 /* Swap the send and the receive. */ 898 rep.th.dest = th->source; 899 rep.th.source = th->dest; 900 rep.th.doff = arg.iov[0].iov_len / 4; 901 rep.th.seq = htonl(seq); 902 rep.th.ack_seq = htonl(ack); 903 rep.th.ack = 1; 904 rep.th.window = htons(win); 905 906 #ifdef CONFIG_TCP_MD5SIG 907 if (key) { 908 int offset = (tsecr) ? 3 : 0; 909 910 rep.opt[offset++] = htonl((TCPOPT_NOP << 24) | 911 (TCPOPT_NOP << 16) | 912 (TCPOPT_MD5SIG << 8) | 913 TCPOLEN_MD5SIG); 914 arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED; 915 rep.th.doff = arg.iov[0].iov_len/4; 916 917 tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[offset], 918 key, ip_hdr(skb)->saddr, 919 ip_hdr(skb)->daddr, &rep.th); 920 } 921 #endif 922 arg.flags = reply_flags; 923 arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr, 924 ip_hdr(skb)->saddr, /* XXX */ 925 arg.iov[0].iov_len, IPPROTO_TCP, 0); 926 arg.csumoffset = offsetof(struct tcphdr, check) / 2; 927 if (oif) 928 arg.bound_dev_if = oif; 929 arg.tos = tos; 930 arg.uid = sock_net_uid(net, sk_fullsock(sk) ? sk : NULL); 931 local_bh_disable(); 932 ctl_sk = this_cpu_read(ipv4_tcp_sk); 933 sock_net_set(ctl_sk, net); 934 ctl_sk->sk_mark = (sk->sk_state == TCP_TIME_WAIT) ? 935 inet_twsk(sk)->tw_mark : sk->sk_mark; 936 ctl_sk->sk_priority = (sk->sk_state == TCP_TIME_WAIT) ? 937 inet_twsk(sk)->tw_priority : sk->sk_priority; 938 transmit_time = tcp_transmit_time(sk); 939 ip_send_unicast_reply(ctl_sk, 940 skb, &TCP_SKB_CB(skb)->header.h4.opt, 941 ip_hdr(skb)->saddr, ip_hdr(skb)->daddr, 942 &arg, arg.iov[0].iov_len, 943 transmit_time, txhash); 944 945 sock_net_set(ctl_sk, &init_net); 946 __TCP_INC_STATS(net, TCP_MIB_OUTSEGS); 947 local_bh_enable(); 948 } 949 950 static void tcp_v4_timewait_ack(struct sock *sk, struct sk_buff *skb) 951 { 952 struct inet_timewait_sock *tw = inet_twsk(sk); 953 struct tcp_timewait_sock *tcptw = tcp_twsk(sk); 954 955 tcp_v4_send_ack(sk, skb, 956 tcptw->tw_snd_nxt, tcptw->tw_rcv_nxt, 957 tcptw->tw_rcv_wnd >> tw->tw_rcv_wscale, 958 tcp_time_stamp_raw() + tcptw->tw_ts_offset, 959 tcptw->tw_ts_recent, 960 tw->tw_bound_dev_if, 961 tcp_twsk_md5_key(tcptw), 962 tw->tw_transparent ? IP_REPLY_ARG_NOSRCCHECK : 0, 963 tw->tw_tos, 964 tw->tw_txhash 965 ); 966 967 inet_twsk_put(tw); 968 } 969 970 static void tcp_v4_reqsk_send_ack(const struct sock *sk, struct sk_buff *skb, 971 struct request_sock *req) 972 { 973 const union tcp_md5_addr *addr; 974 int l3index; 975 976 /* sk->sk_state == TCP_LISTEN -> for regular TCP_SYN_RECV 977 * sk->sk_state == TCP_SYN_RECV -> for Fast Open. 978 */ 979 u32 seq = (sk->sk_state == TCP_LISTEN) ? tcp_rsk(req)->snt_isn + 1 : 980 tcp_sk(sk)->snd_nxt; 981 982 /* RFC 7323 2.3 983 * The window field (SEG.WND) of every outgoing segment, with the 984 * exception of <SYN> segments, MUST be right-shifted by 985 * Rcv.Wind.Shift bits: 986 */ 987 addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr; 988 l3index = tcp_v4_sdif(skb) ? inet_iif(skb) : 0; 989 tcp_v4_send_ack(sk, skb, seq, 990 tcp_rsk(req)->rcv_nxt, 991 req->rsk_rcv_wnd >> inet_rsk(req)->rcv_wscale, 992 tcp_time_stamp_raw() + tcp_rsk(req)->ts_off, 993 READ_ONCE(req->ts_recent), 994 0, 995 tcp_md5_do_lookup(sk, l3index, addr, AF_INET), 996 inet_rsk(req)->no_srccheck ? IP_REPLY_ARG_NOSRCCHECK : 0, 997 ip_hdr(skb)->tos, 998 READ_ONCE(tcp_rsk(req)->txhash)); 999 } 1000 1001 /* 1002 * Send a SYN-ACK after having received a SYN. 1003 * This still operates on a request_sock only, not on a big 1004 * socket. 1005 */ 1006 static int tcp_v4_send_synack(const struct sock *sk, struct dst_entry *dst, 1007 struct flowi *fl, 1008 struct request_sock *req, 1009 struct tcp_fastopen_cookie *foc, 1010 enum tcp_synack_type synack_type, 1011 struct sk_buff *syn_skb) 1012 { 1013 const struct inet_request_sock *ireq = inet_rsk(req); 1014 struct flowi4 fl4; 1015 int err = -1; 1016 struct sk_buff *skb; 1017 u8 tos; 1018 1019 /* First, grab a route. */ 1020 if (!dst && (dst = inet_csk_route_req(sk, &fl4, req)) == NULL) 1021 return -1; 1022 1023 skb = tcp_make_synack(sk, dst, req, foc, synack_type, syn_skb); 1024 1025 if (skb) { 1026 __tcp_v4_send_check(skb, ireq->ir_loc_addr, ireq->ir_rmt_addr); 1027 1028 tos = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_reflect_tos) ? 1029 (tcp_rsk(req)->syn_tos & ~INET_ECN_MASK) | 1030 (inet_sk(sk)->tos & INET_ECN_MASK) : 1031 inet_sk(sk)->tos; 1032 1033 if (!INET_ECN_is_capable(tos) && 1034 tcp_bpf_ca_needs_ecn((struct sock *)req)) 1035 tos |= INET_ECN_ECT_0; 1036 1037 rcu_read_lock(); 1038 err = ip_build_and_send_pkt(skb, sk, ireq->ir_loc_addr, 1039 ireq->ir_rmt_addr, 1040 rcu_dereference(ireq->ireq_opt), 1041 tos); 1042 rcu_read_unlock(); 1043 err = net_xmit_eval(err); 1044 } 1045 1046 return err; 1047 } 1048 1049 /* 1050 * IPv4 request_sock destructor. 1051 */ 1052 static void tcp_v4_reqsk_destructor(struct request_sock *req) 1053 { 1054 kfree(rcu_dereference_protected(inet_rsk(req)->ireq_opt, 1)); 1055 } 1056 1057 #ifdef CONFIG_TCP_MD5SIG 1058 /* 1059 * RFC2385 MD5 checksumming requires a mapping of 1060 * IP address->MD5 Key. 1061 * We need to maintain these in the sk structure. 1062 */ 1063 1064 DEFINE_STATIC_KEY_DEFERRED_FALSE(tcp_md5_needed, HZ); 1065 EXPORT_SYMBOL(tcp_md5_needed); 1066 1067 static bool better_md5_match(struct tcp_md5sig_key *old, struct tcp_md5sig_key *new) 1068 { 1069 if (!old) 1070 return true; 1071 1072 /* l3index always overrides non-l3index */ 1073 if (old->l3index && new->l3index == 0) 1074 return false; 1075 if (old->l3index == 0 && new->l3index) 1076 return true; 1077 1078 return old->prefixlen < new->prefixlen; 1079 } 1080 1081 /* Find the Key structure for an address. */ 1082 struct tcp_md5sig_key *__tcp_md5_do_lookup(const struct sock *sk, int l3index, 1083 const union tcp_md5_addr *addr, 1084 int family) 1085 { 1086 const struct tcp_sock *tp = tcp_sk(sk); 1087 struct tcp_md5sig_key *key; 1088 const struct tcp_md5sig_info *md5sig; 1089 __be32 mask; 1090 struct tcp_md5sig_key *best_match = NULL; 1091 bool match; 1092 1093 /* caller either holds rcu_read_lock() or socket lock */ 1094 md5sig = rcu_dereference_check(tp->md5sig_info, 1095 lockdep_sock_is_held(sk)); 1096 if (!md5sig) 1097 return NULL; 1098 1099 hlist_for_each_entry_rcu(key, &md5sig->head, node, 1100 lockdep_sock_is_held(sk)) { 1101 if (key->family != family) 1102 continue; 1103 if (key->flags & TCP_MD5SIG_FLAG_IFINDEX && key->l3index != l3index) 1104 continue; 1105 if (family == AF_INET) { 1106 mask = inet_make_mask(key->prefixlen); 1107 match = (key->addr.a4.s_addr & mask) == 1108 (addr->a4.s_addr & mask); 1109 #if IS_ENABLED(CONFIG_IPV6) 1110 } else if (family == AF_INET6) { 1111 match = ipv6_prefix_equal(&key->addr.a6, &addr->a6, 1112 key->prefixlen); 1113 #endif 1114 } else { 1115 match = false; 1116 } 1117 1118 if (match && better_md5_match(best_match, key)) 1119 best_match = key; 1120 } 1121 return best_match; 1122 } 1123 EXPORT_SYMBOL(__tcp_md5_do_lookup); 1124 1125 static struct tcp_md5sig_key *tcp_md5_do_lookup_exact(const struct sock *sk, 1126 const union tcp_md5_addr *addr, 1127 int family, u8 prefixlen, 1128 int l3index, u8 flags) 1129 { 1130 const struct tcp_sock *tp = tcp_sk(sk); 1131 struct tcp_md5sig_key *key; 1132 unsigned int size = sizeof(struct in_addr); 1133 const struct tcp_md5sig_info *md5sig; 1134 1135 /* caller either holds rcu_read_lock() or socket lock */ 1136 md5sig = rcu_dereference_check(tp->md5sig_info, 1137 lockdep_sock_is_held(sk)); 1138 if (!md5sig) 1139 return NULL; 1140 #if IS_ENABLED(CONFIG_IPV6) 1141 if (family == AF_INET6) 1142 size = sizeof(struct in6_addr); 1143 #endif 1144 hlist_for_each_entry_rcu(key, &md5sig->head, node, 1145 lockdep_sock_is_held(sk)) { 1146 if (key->family != family) 1147 continue; 1148 if ((key->flags & TCP_MD5SIG_FLAG_IFINDEX) != (flags & TCP_MD5SIG_FLAG_IFINDEX)) 1149 continue; 1150 if (key->l3index != l3index) 1151 continue; 1152 if (!memcmp(&key->addr, addr, size) && 1153 key->prefixlen == prefixlen) 1154 return key; 1155 } 1156 return NULL; 1157 } 1158 1159 struct tcp_md5sig_key *tcp_v4_md5_lookup(const struct sock *sk, 1160 const struct sock *addr_sk) 1161 { 1162 const union tcp_md5_addr *addr; 1163 int l3index; 1164 1165 l3index = l3mdev_master_ifindex_by_index(sock_net(sk), 1166 addr_sk->sk_bound_dev_if); 1167 addr = (const union tcp_md5_addr *)&addr_sk->sk_daddr; 1168 return tcp_md5_do_lookup(sk, l3index, addr, AF_INET); 1169 } 1170 EXPORT_SYMBOL(tcp_v4_md5_lookup); 1171 1172 static int tcp_md5sig_info_add(struct sock *sk, gfp_t gfp) 1173 { 1174 struct tcp_sock *tp = tcp_sk(sk); 1175 struct tcp_md5sig_info *md5sig; 1176 1177 md5sig = kmalloc(sizeof(*md5sig), gfp); 1178 if (!md5sig) 1179 return -ENOMEM; 1180 1181 sk_gso_disable(sk); 1182 INIT_HLIST_HEAD(&md5sig->head); 1183 rcu_assign_pointer(tp->md5sig_info, md5sig); 1184 return 0; 1185 } 1186 1187 /* This can be called on a newly created socket, from other files */ 1188 static int __tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr, 1189 int family, u8 prefixlen, int l3index, u8 flags, 1190 const u8 *newkey, u8 newkeylen, gfp_t gfp) 1191 { 1192 /* Add Key to the list */ 1193 struct tcp_md5sig_key *key; 1194 struct tcp_sock *tp = tcp_sk(sk); 1195 struct tcp_md5sig_info *md5sig; 1196 1197 key = tcp_md5_do_lookup_exact(sk, addr, family, prefixlen, l3index, flags); 1198 if (key) { 1199 /* Pre-existing entry - just update that one. 1200 * Note that the key might be used concurrently. 1201 * data_race() is telling kcsan that we do not care of 1202 * key mismatches, since changing MD5 key on live flows 1203 * can lead to packet drops. 1204 */ 1205 data_race(memcpy(key->key, newkey, newkeylen)); 1206 1207 /* Pairs with READ_ONCE() in tcp_md5_hash_key(). 1208 * Also note that a reader could catch new key->keylen value 1209 * but old key->key[], this is the reason we use __GFP_ZERO 1210 * at sock_kmalloc() time below these lines. 1211 */ 1212 WRITE_ONCE(key->keylen, newkeylen); 1213 1214 return 0; 1215 } 1216 1217 md5sig = rcu_dereference_protected(tp->md5sig_info, 1218 lockdep_sock_is_held(sk)); 1219 1220 key = sock_kmalloc(sk, sizeof(*key), gfp | __GFP_ZERO); 1221 if (!key) 1222 return -ENOMEM; 1223 if (!tcp_alloc_md5sig_pool()) { 1224 sock_kfree_s(sk, key, sizeof(*key)); 1225 return -ENOMEM; 1226 } 1227 1228 memcpy(key->key, newkey, newkeylen); 1229 key->keylen = newkeylen; 1230 key->family = family; 1231 key->prefixlen = prefixlen; 1232 key->l3index = l3index; 1233 key->flags = flags; 1234 memcpy(&key->addr, addr, 1235 (IS_ENABLED(CONFIG_IPV6) && family == AF_INET6) ? sizeof(struct in6_addr) : 1236 sizeof(struct in_addr)); 1237 hlist_add_head_rcu(&key->node, &md5sig->head); 1238 return 0; 1239 } 1240 1241 int tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr, 1242 int family, u8 prefixlen, int l3index, u8 flags, 1243 const u8 *newkey, u8 newkeylen) 1244 { 1245 struct tcp_sock *tp = tcp_sk(sk); 1246 1247 if (!rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk))) { 1248 if (tcp_md5sig_info_add(sk, GFP_KERNEL)) 1249 return -ENOMEM; 1250 1251 if (!static_branch_inc(&tcp_md5_needed.key)) { 1252 struct tcp_md5sig_info *md5sig; 1253 1254 md5sig = rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk)); 1255 rcu_assign_pointer(tp->md5sig_info, NULL); 1256 kfree_rcu(md5sig, rcu); 1257 return -EUSERS; 1258 } 1259 } 1260 1261 return __tcp_md5_do_add(sk, addr, family, prefixlen, l3index, flags, 1262 newkey, newkeylen, GFP_KERNEL); 1263 } 1264 EXPORT_SYMBOL(tcp_md5_do_add); 1265 1266 int tcp_md5_key_copy(struct sock *sk, const union tcp_md5_addr *addr, 1267 int family, u8 prefixlen, int l3index, 1268 struct tcp_md5sig_key *key) 1269 { 1270 struct tcp_sock *tp = tcp_sk(sk); 1271 1272 if (!rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk))) { 1273 if (tcp_md5sig_info_add(sk, sk_gfp_mask(sk, GFP_ATOMIC))) 1274 return -ENOMEM; 1275 1276 if (!static_key_fast_inc_not_disabled(&tcp_md5_needed.key.key)) { 1277 struct tcp_md5sig_info *md5sig; 1278 1279 md5sig = rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk)); 1280 net_warn_ratelimited("Too many TCP-MD5 keys in the system\n"); 1281 rcu_assign_pointer(tp->md5sig_info, NULL); 1282 kfree_rcu(md5sig, rcu); 1283 return -EUSERS; 1284 } 1285 } 1286 1287 return __tcp_md5_do_add(sk, addr, family, prefixlen, l3index, 1288 key->flags, key->key, key->keylen, 1289 sk_gfp_mask(sk, GFP_ATOMIC)); 1290 } 1291 EXPORT_SYMBOL(tcp_md5_key_copy); 1292 1293 int tcp_md5_do_del(struct sock *sk, const union tcp_md5_addr *addr, int family, 1294 u8 prefixlen, int l3index, u8 flags) 1295 { 1296 struct tcp_md5sig_key *key; 1297 1298 key = tcp_md5_do_lookup_exact(sk, addr, family, prefixlen, l3index, flags); 1299 if (!key) 1300 return -ENOENT; 1301 hlist_del_rcu(&key->node); 1302 atomic_sub(sizeof(*key), &sk->sk_omem_alloc); 1303 kfree_rcu(key, rcu); 1304 return 0; 1305 } 1306 EXPORT_SYMBOL(tcp_md5_do_del); 1307 1308 static void tcp_clear_md5_list(struct sock *sk) 1309 { 1310 struct tcp_sock *tp = tcp_sk(sk); 1311 struct tcp_md5sig_key *key; 1312 struct hlist_node *n; 1313 struct tcp_md5sig_info *md5sig; 1314 1315 md5sig = rcu_dereference_protected(tp->md5sig_info, 1); 1316 1317 hlist_for_each_entry_safe(key, n, &md5sig->head, node) { 1318 hlist_del_rcu(&key->node); 1319 atomic_sub(sizeof(*key), &sk->sk_omem_alloc); 1320 kfree_rcu(key, rcu); 1321 } 1322 } 1323 1324 static int tcp_v4_parse_md5_keys(struct sock *sk, int optname, 1325 sockptr_t optval, int optlen) 1326 { 1327 struct tcp_md5sig cmd; 1328 struct sockaddr_in *sin = (struct sockaddr_in *)&cmd.tcpm_addr; 1329 const union tcp_md5_addr *addr; 1330 u8 prefixlen = 32; 1331 int l3index = 0; 1332 u8 flags; 1333 1334 if (optlen < sizeof(cmd)) 1335 return -EINVAL; 1336 1337 if (copy_from_sockptr(&cmd, optval, sizeof(cmd))) 1338 return -EFAULT; 1339 1340 if (sin->sin_family != AF_INET) 1341 return -EINVAL; 1342 1343 flags = cmd.tcpm_flags & TCP_MD5SIG_FLAG_IFINDEX; 1344 1345 if (optname == TCP_MD5SIG_EXT && 1346 cmd.tcpm_flags & TCP_MD5SIG_FLAG_PREFIX) { 1347 prefixlen = cmd.tcpm_prefixlen; 1348 if (prefixlen > 32) 1349 return -EINVAL; 1350 } 1351 1352 if (optname == TCP_MD5SIG_EXT && cmd.tcpm_ifindex && 1353 cmd.tcpm_flags & TCP_MD5SIG_FLAG_IFINDEX) { 1354 struct net_device *dev; 1355 1356 rcu_read_lock(); 1357 dev = dev_get_by_index_rcu(sock_net(sk), cmd.tcpm_ifindex); 1358 if (dev && netif_is_l3_master(dev)) 1359 l3index = dev->ifindex; 1360 1361 rcu_read_unlock(); 1362 1363 /* ok to reference set/not set outside of rcu; 1364 * right now device MUST be an L3 master 1365 */ 1366 if (!dev || !l3index) 1367 return -EINVAL; 1368 } 1369 1370 addr = (union tcp_md5_addr *)&sin->sin_addr.s_addr; 1371 1372 if (!cmd.tcpm_keylen) 1373 return tcp_md5_do_del(sk, addr, AF_INET, prefixlen, l3index, flags); 1374 1375 if (cmd.tcpm_keylen > TCP_MD5SIG_MAXKEYLEN) 1376 return -EINVAL; 1377 1378 return tcp_md5_do_add(sk, addr, AF_INET, prefixlen, l3index, flags, 1379 cmd.tcpm_key, cmd.tcpm_keylen); 1380 } 1381 1382 static int tcp_v4_md5_hash_headers(struct tcp_md5sig_pool *hp, 1383 __be32 daddr, __be32 saddr, 1384 const struct tcphdr *th, int nbytes) 1385 { 1386 struct tcp4_pseudohdr *bp; 1387 struct scatterlist sg; 1388 struct tcphdr *_th; 1389 1390 bp = hp->scratch; 1391 bp->saddr = saddr; 1392 bp->daddr = daddr; 1393 bp->pad = 0; 1394 bp->protocol = IPPROTO_TCP; 1395 bp->len = cpu_to_be16(nbytes); 1396 1397 _th = (struct tcphdr *)(bp + 1); 1398 memcpy(_th, th, sizeof(*th)); 1399 _th->check = 0; 1400 1401 sg_init_one(&sg, bp, sizeof(*bp) + sizeof(*th)); 1402 ahash_request_set_crypt(hp->md5_req, &sg, NULL, 1403 sizeof(*bp) + sizeof(*th)); 1404 return crypto_ahash_update(hp->md5_req); 1405 } 1406 1407 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key, 1408 __be32 daddr, __be32 saddr, const struct tcphdr *th) 1409 { 1410 struct tcp_md5sig_pool *hp; 1411 struct ahash_request *req; 1412 1413 hp = tcp_get_md5sig_pool(); 1414 if (!hp) 1415 goto clear_hash_noput; 1416 req = hp->md5_req; 1417 1418 if (crypto_ahash_init(req)) 1419 goto clear_hash; 1420 if (tcp_v4_md5_hash_headers(hp, daddr, saddr, th, th->doff << 2)) 1421 goto clear_hash; 1422 if (tcp_md5_hash_key(hp, key)) 1423 goto clear_hash; 1424 ahash_request_set_crypt(req, NULL, md5_hash, 0); 1425 if (crypto_ahash_final(req)) 1426 goto clear_hash; 1427 1428 tcp_put_md5sig_pool(); 1429 return 0; 1430 1431 clear_hash: 1432 tcp_put_md5sig_pool(); 1433 clear_hash_noput: 1434 memset(md5_hash, 0, 16); 1435 return 1; 1436 } 1437 1438 int tcp_v4_md5_hash_skb(char *md5_hash, const struct tcp_md5sig_key *key, 1439 const struct sock *sk, 1440 const struct sk_buff *skb) 1441 { 1442 struct tcp_md5sig_pool *hp; 1443 struct ahash_request *req; 1444 const struct tcphdr *th = tcp_hdr(skb); 1445 __be32 saddr, daddr; 1446 1447 if (sk) { /* valid for establish/request sockets */ 1448 saddr = sk->sk_rcv_saddr; 1449 daddr = sk->sk_daddr; 1450 } else { 1451 const struct iphdr *iph = ip_hdr(skb); 1452 saddr = iph->saddr; 1453 daddr = iph->daddr; 1454 } 1455 1456 hp = tcp_get_md5sig_pool(); 1457 if (!hp) 1458 goto clear_hash_noput; 1459 req = hp->md5_req; 1460 1461 if (crypto_ahash_init(req)) 1462 goto clear_hash; 1463 1464 if (tcp_v4_md5_hash_headers(hp, daddr, saddr, th, skb->len)) 1465 goto clear_hash; 1466 if (tcp_md5_hash_skb_data(hp, skb, th->doff << 2)) 1467 goto clear_hash; 1468 if (tcp_md5_hash_key(hp, key)) 1469 goto clear_hash; 1470 ahash_request_set_crypt(req, NULL, md5_hash, 0); 1471 if (crypto_ahash_final(req)) 1472 goto clear_hash; 1473 1474 tcp_put_md5sig_pool(); 1475 return 0; 1476 1477 clear_hash: 1478 tcp_put_md5sig_pool(); 1479 clear_hash_noput: 1480 memset(md5_hash, 0, 16); 1481 return 1; 1482 } 1483 EXPORT_SYMBOL(tcp_v4_md5_hash_skb); 1484 1485 #endif 1486 1487 static void tcp_v4_init_req(struct request_sock *req, 1488 const struct sock *sk_listener, 1489 struct sk_buff *skb) 1490 { 1491 struct inet_request_sock *ireq = inet_rsk(req); 1492 struct net *net = sock_net(sk_listener); 1493 1494 sk_rcv_saddr_set(req_to_sk(req), ip_hdr(skb)->daddr); 1495 sk_daddr_set(req_to_sk(req), ip_hdr(skb)->saddr); 1496 RCU_INIT_POINTER(ireq->ireq_opt, tcp_v4_save_options(net, skb)); 1497 } 1498 1499 static struct dst_entry *tcp_v4_route_req(const struct sock *sk, 1500 struct sk_buff *skb, 1501 struct flowi *fl, 1502 struct request_sock *req) 1503 { 1504 tcp_v4_init_req(req, sk, skb); 1505 1506 if (security_inet_conn_request(sk, skb, req)) 1507 return NULL; 1508 1509 return inet_csk_route_req(sk, &fl->u.ip4, req); 1510 } 1511 1512 struct request_sock_ops tcp_request_sock_ops __read_mostly = { 1513 .family = PF_INET, 1514 .obj_size = sizeof(struct tcp_request_sock), 1515 .rtx_syn_ack = tcp_rtx_synack, 1516 .send_ack = tcp_v4_reqsk_send_ack, 1517 .destructor = tcp_v4_reqsk_destructor, 1518 .send_reset = tcp_v4_send_reset, 1519 .syn_ack_timeout = tcp_syn_ack_timeout, 1520 }; 1521 1522 const struct tcp_request_sock_ops tcp_request_sock_ipv4_ops = { 1523 .mss_clamp = TCP_MSS_DEFAULT, 1524 #ifdef CONFIG_TCP_MD5SIG 1525 .req_md5_lookup = tcp_v4_md5_lookup, 1526 .calc_md5_hash = tcp_v4_md5_hash_skb, 1527 #endif 1528 #ifdef CONFIG_SYN_COOKIES 1529 .cookie_init_seq = cookie_v4_init_sequence, 1530 #endif 1531 .route_req = tcp_v4_route_req, 1532 .init_seq = tcp_v4_init_seq, 1533 .init_ts_off = tcp_v4_init_ts_off, 1534 .send_synack = tcp_v4_send_synack, 1535 }; 1536 1537 int tcp_v4_conn_request(struct sock *sk, struct sk_buff *skb) 1538 { 1539 /* Never answer to SYNs send to broadcast or multicast */ 1540 if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST)) 1541 goto drop; 1542 1543 return tcp_conn_request(&tcp_request_sock_ops, 1544 &tcp_request_sock_ipv4_ops, sk, skb); 1545 1546 drop: 1547 tcp_listendrop(sk); 1548 return 0; 1549 } 1550 EXPORT_SYMBOL(tcp_v4_conn_request); 1551 1552 1553 /* 1554 * The three way handshake has completed - we got a valid synack - 1555 * now create the new socket. 1556 */ 1557 struct sock *tcp_v4_syn_recv_sock(const struct sock *sk, struct sk_buff *skb, 1558 struct request_sock *req, 1559 struct dst_entry *dst, 1560 struct request_sock *req_unhash, 1561 bool *own_req) 1562 { 1563 struct inet_request_sock *ireq; 1564 bool found_dup_sk = false; 1565 struct inet_sock *newinet; 1566 struct tcp_sock *newtp; 1567 struct sock *newsk; 1568 #ifdef CONFIG_TCP_MD5SIG 1569 const union tcp_md5_addr *addr; 1570 struct tcp_md5sig_key *key; 1571 int l3index; 1572 #endif 1573 struct ip_options_rcu *inet_opt; 1574 1575 if (sk_acceptq_is_full(sk)) 1576 goto exit_overflow; 1577 1578 newsk = tcp_create_openreq_child(sk, req, skb); 1579 if (!newsk) 1580 goto exit_nonewsk; 1581 1582 newsk->sk_gso_type = SKB_GSO_TCPV4; 1583 inet_sk_rx_dst_set(newsk, skb); 1584 1585 newtp = tcp_sk(newsk); 1586 newinet = inet_sk(newsk); 1587 ireq = inet_rsk(req); 1588 sk_daddr_set(newsk, ireq->ir_rmt_addr); 1589 sk_rcv_saddr_set(newsk, ireq->ir_loc_addr); 1590 newsk->sk_bound_dev_if = ireq->ir_iif; 1591 newinet->inet_saddr = ireq->ir_loc_addr; 1592 inet_opt = rcu_dereference(ireq->ireq_opt); 1593 RCU_INIT_POINTER(newinet->inet_opt, inet_opt); 1594 newinet->mc_index = inet_iif(skb); 1595 newinet->mc_ttl = ip_hdr(skb)->ttl; 1596 newinet->rcv_tos = ip_hdr(skb)->tos; 1597 inet_csk(newsk)->icsk_ext_hdr_len = 0; 1598 if (inet_opt) 1599 inet_csk(newsk)->icsk_ext_hdr_len = inet_opt->opt.optlen; 1600 newinet->inet_id = get_random_u16(); 1601 1602 /* Set ToS of the new socket based upon the value of incoming SYN. 1603 * ECT bits are set later in tcp_init_transfer(). 1604 */ 1605 if (READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_reflect_tos)) 1606 newinet->tos = tcp_rsk(req)->syn_tos & ~INET_ECN_MASK; 1607 1608 if (!dst) { 1609 dst = inet_csk_route_child_sock(sk, newsk, req); 1610 if (!dst) 1611 goto put_and_exit; 1612 } else { 1613 /* syncookie case : see end of cookie_v4_check() */ 1614 } 1615 sk_setup_caps(newsk, dst); 1616 1617 tcp_ca_openreq_child(newsk, dst); 1618 1619 tcp_sync_mss(newsk, dst_mtu(dst)); 1620 newtp->advmss = tcp_mss_clamp(tcp_sk(sk), dst_metric_advmss(dst)); 1621 1622 tcp_initialize_rcv_mss(newsk); 1623 1624 #ifdef CONFIG_TCP_MD5SIG 1625 l3index = l3mdev_master_ifindex_by_index(sock_net(sk), ireq->ir_iif); 1626 /* Copy over the MD5 key from the original socket */ 1627 addr = (union tcp_md5_addr *)&newinet->inet_daddr; 1628 key = tcp_md5_do_lookup(sk, l3index, addr, AF_INET); 1629 if (key) { 1630 if (tcp_md5_key_copy(newsk, addr, AF_INET, 32, l3index, key)) 1631 goto put_and_exit; 1632 sk_gso_disable(newsk); 1633 } 1634 #endif 1635 1636 if (__inet_inherit_port(sk, newsk) < 0) 1637 goto put_and_exit; 1638 *own_req = inet_ehash_nolisten(newsk, req_to_sk(req_unhash), 1639 &found_dup_sk); 1640 if (likely(*own_req)) { 1641 tcp_move_syn(newtp, req); 1642 ireq->ireq_opt = NULL; 1643 } else { 1644 newinet->inet_opt = NULL; 1645 1646 if (!req_unhash && found_dup_sk) { 1647 /* This code path should only be executed in the 1648 * syncookie case only 1649 */ 1650 bh_unlock_sock(newsk); 1651 sock_put(newsk); 1652 newsk = NULL; 1653 } 1654 } 1655 return newsk; 1656 1657 exit_overflow: 1658 NET_INC_STATS(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS); 1659 exit_nonewsk: 1660 dst_release(dst); 1661 exit: 1662 tcp_listendrop(sk); 1663 return NULL; 1664 put_and_exit: 1665 newinet->inet_opt = NULL; 1666 inet_csk_prepare_forced_close(newsk); 1667 tcp_done(newsk); 1668 goto exit; 1669 } 1670 EXPORT_SYMBOL(tcp_v4_syn_recv_sock); 1671 1672 static struct sock *tcp_v4_cookie_check(struct sock *sk, struct sk_buff *skb) 1673 { 1674 #ifdef CONFIG_SYN_COOKIES 1675 const struct tcphdr *th = tcp_hdr(skb); 1676 1677 if (!th->syn) 1678 sk = cookie_v4_check(sk, skb); 1679 #endif 1680 return sk; 1681 } 1682 1683 u16 tcp_v4_get_syncookie(struct sock *sk, struct iphdr *iph, 1684 struct tcphdr *th, u32 *cookie) 1685 { 1686 u16 mss = 0; 1687 #ifdef CONFIG_SYN_COOKIES 1688 mss = tcp_get_syncookie_mss(&tcp_request_sock_ops, 1689 &tcp_request_sock_ipv4_ops, sk, th); 1690 if (mss) { 1691 *cookie = __cookie_v4_init_sequence(iph, th, &mss); 1692 tcp_synq_overflow(sk); 1693 } 1694 #endif 1695 return mss; 1696 } 1697 1698 INDIRECT_CALLABLE_DECLARE(struct dst_entry *ipv4_dst_check(struct dst_entry *, 1699 u32)); 1700 /* The socket must have it's spinlock held when we get 1701 * here, unless it is a TCP_LISTEN socket. 1702 * 1703 * We have a potential double-lock case here, so even when 1704 * doing backlog processing we use the BH locking scheme. 1705 * This is because we cannot sleep with the original spinlock 1706 * held. 1707 */ 1708 int tcp_v4_do_rcv(struct sock *sk, struct sk_buff *skb) 1709 { 1710 enum skb_drop_reason reason; 1711 struct sock *rsk; 1712 1713 if (sk->sk_state == TCP_ESTABLISHED) { /* Fast path */ 1714 struct dst_entry *dst; 1715 1716 dst = rcu_dereference_protected(sk->sk_rx_dst, 1717 lockdep_sock_is_held(sk)); 1718 1719 sock_rps_save_rxhash(sk, skb); 1720 sk_mark_napi_id(sk, skb); 1721 if (dst) { 1722 if (sk->sk_rx_dst_ifindex != skb->skb_iif || 1723 !INDIRECT_CALL_1(dst->ops->check, ipv4_dst_check, 1724 dst, 0)) { 1725 RCU_INIT_POINTER(sk->sk_rx_dst, NULL); 1726 dst_release(dst); 1727 } 1728 } 1729 tcp_rcv_established(sk, skb); 1730 return 0; 1731 } 1732 1733 reason = SKB_DROP_REASON_NOT_SPECIFIED; 1734 if (tcp_checksum_complete(skb)) 1735 goto csum_err; 1736 1737 if (sk->sk_state == TCP_LISTEN) { 1738 struct sock *nsk = tcp_v4_cookie_check(sk, skb); 1739 1740 if (!nsk) 1741 goto discard; 1742 if (nsk != sk) { 1743 if (tcp_child_process(sk, nsk, skb)) { 1744 rsk = nsk; 1745 goto reset; 1746 } 1747 return 0; 1748 } 1749 } else 1750 sock_rps_save_rxhash(sk, skb); 1751 1752 if (tcp_rcv_state_process(sk, skb)) { 1753 rsk = sk; 1754 goto reset; 1755 } 1756 return 0; 1757 1758 reset: 1759 tcp_v4_send_reset(rsk, skb); 1760 discard: 1761 kfree_skb_reason(skb, reason); 1762 /* Be careful here. If this function gets more complicated and 1763 * gcc suffers from register pressure on the x86, sk (in %ebx) 1764 * might be destroyed here. This current version compiles correctly, 1765 * but you have been warned. 1766 */ 1767 return 0; 1768 1769 csum_err: 1770 reason = SKB_DROP_REASON_TCP_CSUM; 1771 trace_tcp_bad_csum(skb); 1772 TCP_INC_STATS(sock_net(sk), TCP_MIB_CSUMERRORS); 1773 TCP_INC_STATS(sock_net(sk), TCP_MIB_INERRS); 1774 goto discard; 1775 } 1776 EXPORT_SYMBOL(tcp_v4_do_rcv); 1777 1778 int tcp_v4_early_demux(struct sk_buff *skb) 1779 { 1780 struct net *net = dev_net(skb->dev); 1781 const struct iphdr *iph; 1782 const struct tcphdr *th; 1783 struct sock *sk; 1784 1785 if (skb->pkt_type != PACKET_HOST) 1786 return 0; 1787 1788 if (!pskb_may_pull(skb, skb_transport_offset(skb) + sizeof(struct tcphdr))) 1789 return 0; 1790 1791 iph = ip_hdr(skb); 1792 th = tcp_hdr(skb); 1793 1794 if (th->doff < sizeof(struct tcphdr) / 4) 1795 return 0; 1796 1797 sk = __inet_lookup_established(net, net->ipv4.tcp_death_row.hashinfo, 1798 iph->saddr, th->source, 1799 iph->daddr, ntohs(th->dest), 1800 skb->skb_iif, inet_sdif(skb)); 1801 if (sk) { 1802 skb->sk = sk; 1803 skb->destructor = sock_edemux; 1804 if (sk_fullsock(sk)) { 1805 struct dst_entry *dst = rcu_dereference(sk->sk_rx_dst); 1806 1807 if (dst) 1808 dst = dst_check(dst, 0); 1809 if (dst && 1810 sk->sk_rx_dst_ifindex == skb->skb_iif) 1811 skb_dst_set_noref(skb, dst); 1812 } 1813 } 1814 return 0; 1815 } 1816 1817 bool tcp_add_backlog(struct sock *sk, struct sk_buff *skb, 1818 enum skb_drop_reason *reason) 1819 { 1820 u32 limit, tail_gso_size, tail_gso_segs; 1821 struct skb_shared_info *shinfo; 1822 const struct tcphdr *th; 1823 struct tcphdr *thtail; 1824 struct sk_buff *tail; 1825 unsigned int hdrlen; 1826 bool fragstolen; 1827 u32 gso_segs; 1828 u32 gso_size; 1829 int delta; 1830 1831 /* In case all data was pulled from skb frags (in __pskb_pull_tail()), 1832 * we can fix skb->truesize to its real value to avoid future drops. 1833 * This is valid because skb is not yet charged to the socket. 1834 * It has been noticed pure SACK packets were sometimes dropped 1835 * (if cooked by drivers without copybreak feature). 1836 */ 1837 skb_condense(skb); 1838 1839 skb_dst_drop(skb); 1840 1841 if (unlikely(tcp_checksum_complete(skb))) { 1842 bh_unlock_sock(sk); 1843 trace_tcp_bad_csum(skb); 1844 *reason = SKB_DROP_REASON_TCP_CSUM; 1845 __TCP_INC_STATS(sock_net(sk), TCP_MIB_CSUMERRORS); 1846 __TCP_INC_STATS(sock_net(sk), TCP_MIB_INERRS); 1847 return true; 1848 } 1849 1850 /* Attempt coalescing to last skb in backlog, even if we are 1851 * above the limits. 1852 * This is okay because skb capacity is limited to MAX_SKB_FRAGS. 1853 */ 1854 th = (const struct tcphdr *)skb->data; 1855 hdrlen = th->doff * 4; 1856 1857 tail = sk->sk_backlog.tail; 1858 if (!tail) 1859 goto no_coalesce; 1860 thtail = (struct tcphdr *)tail->data; 1861 1862 if (TCP_SKB_CB(tail)->end_seq != TCP_SKB_CB(skb)->seq || 1863 TCP_SKB_CB(tail)->ip_dsfield != TCP_SKB_CB(skb)->ip_dsfield || 1864 ((TCP_SKB_CB(tail)->tcp_flags | 1865 TCP_SKB_CB(skb)->tcp_flags) & (TCPHDR_SYN | TCPHDR_RST | TCPHDR_URG)) || 1866 !((TCP_SKB_CB(tail)->tcp_flags & 1867 TCP_SKB_CB(skb)->tcp_flags) & TCPHDR_ACK) || 1868 ((TCP_SKB_CB(tail)->tcp_flags ^ 1869 TCP_SKB_CB(skb)->tcp_flags) & (TCPHDR_ECE | TCPHDR_CWR)) || 1870 #ifdef CONFIG_TLS_DEVICE 1871 tail->decrypted != skb->decrypted || 1872 #endif 1873 thtail->doff != th->doff || 1874 memcmp(thtail + 1, th + 1, hdrlen - sizeof(*th))) 1875 goto no_coalesce; 1876 1877 __skb_pull(skb, hdrlen); 1878 1879 shinfo = skb_shinfo(skb); 1880 gso_size = shinfo->gso_size ?: skb->len; 1881 gso_segs = shinfo->gso_segs ?: 1; 1882 1883 shinfo = skb_shinfo(tail); 1884 tail_gso_size = shinfo->gso_size ?: (tail->len - hdrlen); 1885 tail_gso_segs = shinfo->gso_segs ?: 1; 1886 1887 if (skb_try_coalesce(tail, skb, &fragstolen, &delta)) { 1888 TCP_SKB_CB(tail)->end_seq = TCP_SKB_CB(skb)->end_seq; 1889 1890 if (likely(!before(TCP_SKB_CB(skb)->ack_seq, TCP_SKB_CB(tail)->ack_seq))) { 1891 TCP_SKB_CB(tail)->ack_seq = TCP_SKB_CB(skb)->ack_seq; 1892 thtail->window = th->window; 1893 } 1894 1895 /* We have to update both TCP_SKB_CB(tail)->tcp_flags and 1896 * thtail->fin, so that the fast path in tcp_rcv_established() 1897 * is not entered if we append a packet with a FIN. 1898 * SYN, RST, URG are not present. 1899 * ACK is set on both packets. 1900 * PSH : we do not really care in TCP stack, 1901 * at least for 'GRO' packets. 1902 */ 1903 thtail->fin |= th->fin; 1904 TCP_SKB_CB(tail)->tcp_flags |= TCP_SKB_CB(skb)->tcp_flags; 1905 1906 if (TCP_SKB_CB(skb)->has_rxtstamp) { 1907 TCP_SKB_CB(tail)->has_rxtstamp = true; 1908 tail->tstamp = skb->tstamp; 1909 skb_hwtstamps(tail)->hwtstamp = skb_hwtstamps(skb)->hwtstamp; 1910 } 1911 1912 /* Not as strict as GRO. We only need to carry mss max value */ 1913 shinfo->gso_size = max(gso_size, tail_gso_size); 1914 shinfo->gso_segs = min_t(u32, gso_segs + tail_gso_segs, 0xFFFF); 1915 1916 sk->sk_backlog.len += delta; 1917 __NET_INC_STATS(sock_net(sk), 1918 LINUX_MIB_TCPBACKLOGCOALESCE); 1919 kfree_skb_partial(skb, fragstolen); 1920 return false; 1921 } 1922 __skb_push(skb, hdrlen); 1923 1924 no_coalesce: 1925 limit = (u32)READ_ONCE(sk->sk_rcvbuf) + (u32)(READ_ONCE(sk->sk_sndbuf) >> 1); 1926 1927 /* Only socket owner can try to collapse/prune rx queues 1928 * to reduce memory overhead, so add a little headroom here. 1929 * Few sockets backlog are possibly concurrently non empty. 1930 */ 1931 limit += 64 * 1024; 1932 1933 if (unlikely(sk_add_backlog(sk, skb, limit))) { 1934 bh_unlock_sock(sk); 1935 *reason = SKB_DROP_REASON_SOCKET_BACKLOG; 1936 __NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPBACKLOGDROP); 1937 return true; 1938 } 1939 return false; 1940 } 1941 EXPORT_SYMBOL(tcp_add_backlog); 1942 1943 int tcp_filter(struct sock *sk, struct sk_buff *skb) 1944 { 1945 struct tcphdr *th = (struct tcphdr *)skb->data; 1946 1947 return sk_filter_trim_cap(sk, skb, th->doff * 4); 1948 } 1949 EXPORT_SYMBOL(tcp_filter); 1950 1951 static void tcp_v4_restore_cb(struct sk_buff *skb) 1952 { 1953 memmove(IPCB(skb), &TCP_SKB_CB(skb)->header.h4, 1954 sizeof(struct inet_skb_parm)); 1955 } 1956 1957 static void tcp_v4_fill_cb(struct sk_buff *skb, const struct iphdr *iph, 1958 const struct tcphdr *th) 1959 { 1960 /* This is tricky : We move IPCB at its correct location into TCP_SKB_CB() 1961 * barrier() makes sure compiler wont play fool^Waliasing games. 1962 */ 1963 memmove(&TCP_SKB_CB(skb)->header.h4, IPCB(skb), 1964 sizeof(struct inet_skb_parm)); 1965 barrier(); 1966 1967 TCP_SKB_CB(skb)->seq = ntohl(th->seq); 1968 TCP_SKB_CB(skb)->end_seq = (TCP_SKB_CB(skb)->seq + th->syn + th->fin + 1969 skb->len - th->doff * 4); 1970 TCP_SKB_CB(skb)->ack_seq = ntohl(th->ack_seq); 1971 TCP_SKB_CB(skb)->tcp_flags = tcp_flag_byte(th); 1972 TCP_SKB_CB(skb)->tcp_tw_isn = 0; 1973 TCP_SKB_CB(skb)->ip_dsfield = ipv4_get_dsfield(iph); 1974 TCP_SKB_CB(skb)->sacked = 0; 1975 TCP_SKB_CB(skb)->has_rxtstamp = 1976 skb->tstamp || skb_hwtstamps(skb)->hwtstamp; 1977 } 1978 1979 /* 1980 * From tcp_input.c 1981 */ 1982 1983 int tcp_v4_rcv(struct sk_buff *skb) 1984 { 1985 struct net *net = dev_net(skb->dev); 1986 enum skb_drop_reason drop_reason; 1987 int sdif = inet_sdif(skb); 1988 int dif = inet_iif(skb); 1989 const struct iphdr *iph; 1990 const struct tcphdr *th; 1991 bool refcounted; 1992 struct sock *sk; 1993 int ret; 1994 1995 drop_reason = SKB_DROP_REASON_NOT_SPECIFIED; 1996 if (skb->pkt_type != PACKET_HOST) 1997 goto discard_it; 1998 1999 /* Count it even if it's bad */ 2000 __TCP_INC_STATS(net, TCP_MIB_INSEGS); 2001 2002 if (!pskb_may_pull(skb, sizeof(struct tcphdr))) 2003 goto discard_it; 2004 2005 th = (const struct tcphdr *)skb->data; 2006 2007 if (unlikely(th->doff < sizeof(struct tcphdr) / 4)) { 2008 drop_reason = SKB_DROP_REASON_PKT_TOO_SMALL; 2009 goto bad_packet; 2010 } 2011 if (!pskb_may_pull(skb, th->doff * 4)) 2012 goto discard_it; 2013 2014 /* An explanation is required here, I think. 2015 * Packet length and doff are validated by header prediction, 2016 * provided case of th->doff==0 is eliminated. 2017 * So, we defer the checks. */ 2018 2019 if (skb_checksum_init(skb, IPPROTO_TCP, inet_compute_pseudo)) 2020 goto csum_error; 2021 2022 th = (const struct tcphdr *)skb->data; 2023 iph = ip_hdr(skb); 2024 lookup: 2025 sk = __inet_lookup_skb(net->ipv4.tcp_death_row.hashinfo, 2026 skb, __tcp_hdrlen(th), th->source, 2027 th->dest, sdif, &refcounted); 2028 if (!sk) 2029 goto no_tcp_socket; 2030 2031 process: 2032 if (sk->sk_state == TCP_TIME_WAIT) 2033 goto do_time_wait; 2034 2035 if (sk->sk_state == TCP_NEW_SYN_RECV) { 2036 struct request_sock *req = inet_reqsk(sk); 2037 bool req_stolen = false; 2038 struct sock *nsk; 2039 2040 sk = req->rsk_listener; 2041 if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb)) 2042 drop_reason = SKB_DROP_REASON_XFRM_POLICY; 2043 else 2044 drop_reason = tcp_inbound_md5_hash(sk, skb, 2045 &iph->saddr, &iph->daddr, 2046 AF_INET, dif, sdif); 2047 if (unlikely(drop_reason)) { 2048 sk_drops_add(sk, skb); 2049 reqsk_put(req); 2050 goto discard_it; 2051 } 2052 if (tcp_checksum_complete(skb)) { 2053 reqsk_put(req); 2054 goto csum_error; 2055 } 2056 if (unlikely(sk->sk_state != TCP_LISTEN)) { 2057 nsk = reuseport_migrate_sock(sk, req_to_sk(req), skb); 2058 if (!nsk) { 2059 inet_csk_reqsk_queue_drop_and_put(sk, req); 2060 goto lookup; 2061 } 2062 sk = nsk; 2063 /* reuseport_migrate_sock() has already held one sk_refcnt 2064 * before returning. 2065 */ 2066 } else { 2067 /* We own a reference on the listener, increase it again 2068 * as we might lose it too soon. 2069 */ 2070 sock_hold(sk); 2071 } 2072 refcounted = true; 2073 nsk = NULL; 2074 if (!tcp_filter(sk, skb)) { 2075 th = (const struct tcphdr *)skb->data; 2076 iph = ip_hdr(skb); 2077 tcp_v4_fill_cb(skb, iph, th); 2078 nsk = tcp_check_req(sk, skb, req, false, &req_stolen); 2079 } else { 2080 drop_reason = SKB_DROP_REASON_SOCKET_FILTER; 2081 } 2082 if (!nsk) { 2083 reqsk_put(req); 2084 if (req_stolen) { 2085 /* Another cpu got exclusive access to req 2086 * and created a full blown socket. 2087 * Try to feed this packet to this socket 2088 * instead of discarding it. 2089 */ 2090 tcp_v4_restore_cb(skb); 2091 sock_put(sk); 2092 goto lookup; 2093 } 2094 goto discard_and_relse; 2095 } 2096 nf_reset_ct(skb); 2097 if (nsk == sk) { 2098 reqsk_put(req); 2099 tcp_v4_restore_cb(skb); 2100 } else if (tcp_child_process(sk, nsk, skb)) { 2101 tcp_v4_send_reset(nsk, skb); 2102 goto discard_and_relse; 2103 } else { 2104 sock_put(sk); 2105 return 0; 2106 } 2107 } 2108 2109 if (static_branch_unlikely(&ip4_min_ttl)) { 2110 /* min_ttl can be changed concurrently from do_ip_setsockopt() */ 2111 if (unlikely(iph->ttl < READ_ONCE(inet_sk(sk)->min_ttl))) { 2112 __NET_INC_STATS(net, LINUX_MIB_TCPMINTTLDROP); 2113 drop_reason = SKB_DROP_REASON_TCP_MINTTL; 2114 goto discard_and_relse; 2115 } 2116 } 2117 2118 if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb)) { 2119 drop_reason = SKB_DROP_REASON_XFRM_POLICY; 2120 goto discard_and_relse; 2121 } 2122 2123 drop_reason = tcp_inbound_md5_hash(sk, skb, &iph->saddr, 2124 &iph->daddr, AF_INET, dif, sdif); 2125 if (drop_reason) 2126 goto discard_and_relse; 2127 2128 nf_reset_ct(skb); 2129 2130 if (tcp_filter(sk, skb)) { 2131 drop_reason = SKB_DROP_REASON_SOCKET_FILTER; 2132 goto discard_and_relse; 2133 } 2134 th = (const struct tcphdr *)skb->data; 2135 iph = ip_hdr(skb); 2136 tcp_v4_fill_cb(skb, iph, th); 2137 2138 skb->dev = NULL; 2139 2140 if (sk->sk_state == TCP_LISTEN) { 2141 ret = tcp_v4_do_rcv(sk, skb); 2142 goto put_and_return; 2143 } 2144 2145 sk_incoming_cpu_update(sk); 2146 2147 bh_lock_sock_nested(sk); 2148 tcp_segs_in(tcp_sk(sk), skb); 2149 ret = 0; 2150 if (!sock_owned_by_user(sk)) { 2151 ret = tcp_v4_do_rcv(sk, skb); 2152 } else { 2153 if (tcp_add_backlog(sk, skb, &drop_reason)) 2154 goto discard_and_relse; 2155 } 2156 bh_unlock_sock(sk); 2157 2158 put_and_return: 2159 if (refcounted) 2160 sock_put(sk); 2161 2162 return ret; 2163 2164 no_tcp_socket: 2165 drop_reason = SKB_DROP_REASON_NO_SOCKET; 2166 if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) 2167 goto discard_it; 2168 2169 tcp_v4_fill_cb(skb, iph, th); 2170 2171 if (tcp_checksum_complete(skb)) { 2172 csum_error: 2173 drop_reason = SKB_DROP_REASON_TCP_CSUM; 2174 trace_tcp_bad_csum(skb); 2175 __TCP_INC_STATS(net, TCP_MIB_CSUMERRORS); 2176 bad_packet: 2177 __TCP_INC_STATS(net, TCP_MIB_INERRS); 2178 } else { 2179 tcp_v4_send_reset(NULL, skb); 2180 } 2181 2182 discard_it: 2183 SKB_DR_OR(drop_reason, NOT_SPECIFIED); 2184 /* Discard frame. */ 2185 kfree_skb_reason(skb, drop_reason); 2186 return 0; 2187 2188 discard_and_relse: 2189 sk_drops_add(sk, skb); 2190 if (refcounted) 2191 sock_put(sk); 2192 goto discard_it; 2193 2194 do_time_wait: 2195 if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) { 2196 drop_reason = SKB_DROP_REASON_XFRM_POLICY; 2197 inet_twsk_put(inet_twsk(sk)); 2198 goto discard_it; 2199 } 2200 2201 tcp_v4_fill_cb(skb, iph, th); 2202 2203 if (tcp_checksum_complete(skb)) { 2204 inet_twsk_put(inet_twsk(sk)); 2205 goto csum_error; 2206 } 2207 switch (tcp_timewait_state_process(inet_twsk(sk), skb, th)) { 2208 case TCP_TW_SYN: { 2209 struct sock *sk2 = inet_lookup_listener(net, 2210 net->ipv4.tcp_death_row.hashinfo, 2211 skb, __tcp_hdrlen(th), 2212 iph->saddr, th->source, 2213 iph->daddr, th->dest, 2214 inet_iif(skb), 2215 sdif); 2216 if (sk2) { 2217 inet_twsk_deschedule_put(inet_twsk(sk)); 2218 sk = sk2; 2219 tcp_v4_restore_cb(skb); 2220 refcounted = false; 2221 goto process; 2222 } 2223 } 2224 /* to ACK */ 2225 fallthrough; 2226 case TCP_TW_ACK: 2227 tcp_v4_timewait_ack(sk, skb); 2228 break; 2229 case TCP_TW_RST: 2230 tcp_v4_send_reset(sk, skb); 2231 inet_twsk_deschedule_put(inet_twsk(sk)); 2232 goto discard_it; 2233 case TCP_TW_SUCCESS:; 2234 } 2235 goto discard_it; 2236 } 2237 2238 static struct timewait_sock_ops tcp_timewait_sock_ops = { 2239 .twsk_obj_size = sizeof(struct tcp_timewait_sock), 2240 .twsk_unique = tcp_twsk_unique, 2241 .twsk_destructor= tcp_twsk_destructor, 2242 }; 2243 2244 void inet_sk_rx_dst_set(struct sock *sk, const struct sk_buff *skb) 2245 { 2246 struct dst_entry *dst = skb_dst(skb); 2247 2248 if (dst && dst_hold_safe(dst)) { 2249 rcu_assign_pointer(sk->sk_rx_dst, dst); 2250 sk->sk_rx_dst_ifindex = skb->skb_iif; 2251 } 2252 } 2253 EXPORT_SYMBOL(inet_sk_rx_dst_set); 2254 2255 const struct inet_connection_sock_af_ops ipv4_specific = { 2256 .queue_xmit = ip_queue_xmit, 2257 .send_check = tcp_v4_send_check, 2258 .rebuild_header = inet_sk_rebuild_header, 2259 .sk_rx_dst_set = inet_sk_rx_dst_set, 2260 .conn_request = tcp_v4_conn_request, 2261 .syn_recv_sock = tcp_v4_syn_recv_sock, 2262 .net_header_len = sizeof(struct iphdr), 2263 .setsockopt = ip_setsockopt, 2264 .getsockopt = ip_getsockopt, 2265 .addr2sockaddr = inet_csk_addr2sockaddr, 2266 .sockaddr_len = sizeof(struct sockaddr_in), 2267 .mtu_reduced = tcp_v4_mtu_reduced, 2268 }; 2269 EXPORT_SYMBOL(ipv4_specific); 2270 2271 #ifdef CONFIG_TCP_MD5SIG 2272 static const struct tcp_sock_af_ops tcp_sock_ipv4_specific = { 2273 .md5_lookup = tcp_v4_md5_lookup, 2274 .calc_md5_hash = tcp_v4_md5_hash_skb, 2275 .md5_parse = tcp_v4_parse_md5_keys, 2276 }; 2277 #endif 2278 2279 /* NOTE: A lot of things set to zero explicitly by call to 2280 * sk_alloc() so need not be done here. 2281 */ 2282 static int tcp_v4_init_sock(struct sock *sk) 2283 { 2284 struct inet_connection_sock *icsk = inet_csk(sk); 2285 2286 tcp_init_sock(sk); 2287 2288 icsk->icsk_af_ops = &ipv4_specific; 2289 2290 #ifdef CONFIG_TCP_MD5SIG 2291 tcp_sk(sk)->af_specific = &tcp_sock_ipv4_specific; 2292 #endif 2293 2294 return 0; 2295 } 2296 2297 void tcp_v4_destroy_sock(struct sock *sk) 2298 { 2299 struct tcp_sock *tp = tcp_sk(sk); 2300 2301 trace_tcp_destroy_sock(sk); 2302 2303 tcp_clear_xmit_timers(sk); 2304 2305 tcp_cleanup_congestion_control(sk); 2306 2307 tcp_cleanup_ulp(sk); 2308 2309 /* Cleanup up the write buffer. */ 2310 tcp_write_queue_purge(sk); 2311 2312 /* Check if we want to disable active TFO */ 2313 tcp_fastopen_active_disable_ofo_check(sk); 2314 2315 /* Cleans up our, hopefully empty, out_of_order_queue. */ 2316 skb_rbtree_purge(&tp->out_of_order_queue); 2317 2318 #ifdef CONFIG_TCP_MD5SIG 2319 /* Clean up the MD5 key list, if any */ 2320 if (tp->md5sig_info) { 2321 tcp_clear_md5_list(sk); 2322 kfree_rcu(rcu_dereference_protected(tp->md5sig_info, 1), rcu); 2323 tp->md5sig_info = NULL; 2324 static_branch_slow_dec_deferred(&tcp_md5_needed); 2325 } 2326 #endif 2327 2328 /* Clean up a referenced TCP bind bucket. */ 2329 if (inet_csk(sk)->icsk_bind_hash) 2330 inet_put_port(sk); 2331 2332 BUG_ON(rcu_access_pointer(tp->fastopen_rsk)); 2333 2334 /* If socket is aborted during connect operation */ 2335 tcp_free_fastopen_req(tp); 2336 tcp_fastopen_destroy_cipher(sk); 2337 tcp_saved_syn_free(tp); 2338 2339 sk_sockets_allocated_dec(sk); 2340 } 2341 EXPORT_SYMBOL(tcp_v4_destroy_sock); 2342 2343 #ifdef CONFIG_PROC_FS 2344 /* Proc filesystem TCP sock list dumping. */ 2345 2346 static unsigned short seq_file_family(const struct seq_file *seq); 2347 2348 static bool seq_sk_match(struct seq_file *seq, const struct sock *sk) 2349 { 2350 unsigned short family = seq_file_family(seq); 2351 2352 /* AF_UNSPEC is used as a match all */ 2353 return ((family == AF_UNSPEC || family == sk->sk_family) && 2354 net_eq(sock_net(sk), seq_file_net(seq))); 2355 } 2356 2357 /* Find a non empty bucket (starting from st->bucket) 2358 * and return the first sk from it. 2359 */ 2360 static void *listening_get_first(struct seq_file *seq) 2361 { 2362 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2363 struct tcp_iter_state *st = seq->private; 2364 2365 st->offset = 0; 2366 for (; st->bucket <= hinfo->lhash2_mask; st->bucket++) { 2367 struct inet_listen_hashbucket *ilb2; 2368 struct hlist_nulls_node *node; 2369 struct sock *sk; 2370 2371 ilb2 = &hinfo->lhash2[st->bucket]; 2372 if (hlist_nulls_empty(&ilb2->nulls_head)) 2373 continue; 2374 2375 spin_lock(&ilb2->lock); 2376 sk_nulls_for_each(sk, node, &ilb2->nulls_head) { 2377 if (seq_sk_match(seq, sk)) 2378 return sk; 2379 } 2380 spin_unlock(&ilb2->lock); 2381 } 2382 2383 return NULL; 2384 } 2385 2386 /* Find the next sk of "cur" within the same bucket (i.e. st->bucket). 2387 * If "cur" is the last one in the st->bucket, 2388 * call listening_get_first() to return the first sk of the next 2389 * non empty bucket. 2390 */ 2391 static void *listening_get_next(struct seq_file *seq, void *cur) 2392 { 2393 struct tcp_iter_state *st = seq->private; 2394 struct inet_listen_hashbucket *ilb2; 2395 struct hlist_nulls_node *node; 2396 struct inet_hashinfo *hinfo; 2397 struct sock *sk = cur; 2398 2399 ++st->num; 2400 ++st->offset; 2401 2402 sk = sk_nulls_next(sk); 2403 sk_nulls_for_each_from(sk, node) { 2404 if (seq_sk_match(seq, sk)) 2405 return sk; 2406 } 2407 2408 hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2409 ilb2 = &hinfo->lhash2[st->bucket]; 2410 spin_unlock(&ilb2->lock); 2411 ++st->bucket; 2412 return listening_get_first(seq); 2413 } 2414 2415 static void *listening_get_idx(struct seq_file *seq, loff_t *pos) 2416 { 2417 struct tcp_iter_state *st = seq->private; 2418 void *rc; 2419 2420 st->bucket = 0; 2421 st->offset = 0; 2422 rc = listening_get_first(seq); 2423 2424 while (rc && *pos) { 2425 rc = listening_get_next(seq, rc); 2426 --*pos; 2427 } 2428 return rc; 2429 } 2430 2431 static inline bool empty_bucket(struct inet_hashinfo *hinfo, 2432 const struct tcp_iter_state *st) 2433 { 2434 return hlist_nulls_empty(&hinfo->ehash[st->bucket].chain); 2435 } 2436 2437 /* 2438 * Get first established socket starting from bucket given in st->bucket. 2439 * If st->bucket is zero, the very first socket in the hash is returned. 2440 */ 2441 static void *established_get_first(struct seq_file *seq) 2442 { 2443 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2444 struct tcp_iter_state *st = seq->private; 2445 2446 st->offset = 0; 2447 for (; st->bucket <= hinfo->ehash_mask; ++st->bucket) { 2448 struct sock *sk; 2449 struct hlist_nulls_node *node; 2450 spinlock_t *lock = inet_ehash_lockp(hinfo, st->bucket); 2451 2452 cond_resched(); 2453 2454 /* Lockless fast path for the common case of empty buckets */ 2455 if (empty_bucket(hinfo, st)) 2456 continue; 2457 2458 spin_lock_bh(lock); 2459 sk_nulls_for_each(sk, node, &hinfo->ehash[st->bucket].chain) { 2460 if (seq_sk_match(seq, sk)) 2461 return sk; 2462 } 2463 spin_unlock_bh(lock); 2464 } 2465 2466 return NULL; 2467 } 2468 2469 static void *established_get_next(struct seq_file *seq, void *cur) 2470 { 2471 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2472 struct tcp_iter_state *st = seq->private; 2473 struct hlist_nulls_node *node; 2474 struct sock *sk = cur; 2475 2476 ++st->num; 2477 ++st->offset; 2478 2479 sk = sk_nulls_next(sk); 2480 2481 sk_nulls_for_each_from(sk, node) { 2482 if (seq_sk_match(seq, sk)) 2483 return sk; 2484 } 2485 2486 spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket)); 2487 ++st->bucket; 2488 return established_get_first(seq); 2489 } 2490 2491 static void *established_get_idx(struct seq_file *seq, loff_t pos) 2492 { 2493 struct tcp_iter_state *st = seq->private; 2494 void *rc; 2495 2496 st->bucket = 0; 2497 rc = established_get_first(seq); 2498 2499 while (rc && pos) { 2500 rc = established_get_next(seq, rc); 2501 --pos; 2502 } 2503 return rc; 2504 } 2505 2506 static void *tcp_get_idx(struct seq_file *seq, loff_t pos) 2507 { 2508 void *rc; 2509 struct tcp_iter_state *st = seq->private; 2510 2511 st->state = TCP_SEQ_STATE_LISTENING; 2512 rc = listening_get_idx(seq, &pos); 2513 2514 if (!rc) { 2515 st->state = TCP_SEQ_STATE_ESTABLISHED; 2516 rc = established_get_idx(seq, pos); 2517 } 2518 2519 return rc; 2520 } 2521 2522 static void *tcp_seek_last_pos(struct seq_file *seq) 2523 { 2524 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2525 struct tcp_iter_state *st = seq->private; 2526 int bucket = st->bucket; 2527 int offset = st->offset; 2528 int orig_num = st->num; 2529 void *rc = NULL; 2530 2531 switch (st->state) { 2532 case TCP_SEQ_STATE_LISTENING: 2533 if (st->bucket > hinfo->lhash2_mask) 2534 break; 2535 rc = listening_get_first(seq); 2536 while (offset-- && rc && bucket == st->bucket) 2537 rc = listening_get_next(seq, rc); 2538 if (rc) 2539 break; 2540 st->bucket = 0; 2541 st->state = TCP_SEQ_STATE_ESTABLISHED; 2542 fallthrough; 2543 case TCP_SEQ_STATE_ESTABLISHED: 2544 if (st->bucket > hinfo->ehash_mask) 2545 break; 2546 rc = established_get_first(seq); 2547 while (offset-- && rc && bucket == st->bucket) 2548 rc = established_get_next(seq, rc); 2549 } 2550 2551 st->num = orig_num; 2552 2553 return rc; 2554 } 2555 2556 void *tcp_seq_start(struct seq_file *seq, loff_t *pos) 2557 { 2558 struct tcp_iter_state *st = seq->private; 2559 void *rc; 2560 2561 if (*pos && *pos == st->last_pos) { 2562 rc = tcp_seek_last_pos(seq); 2563 if (rc) 2564 goto out; 2565 } 2566 2567 st->state = TCP_SEQ_STATE_LISTENING; 2568 st->num = 0; 2569 st->bucket = 0; 2570 st->offset = 0; 2571 rc = *pos ? tcp_get_idx(seq, *pos - 1) : SEQ_START_TOKEN; 2572 2573 out: 2574 st->last_pos = *pos; 2575 return rc; 2576 } 2577 EXPORT_SYMBOL(tcp_seq_start); 2578 2579 void *tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos) 2580 { 2581 struct tcp_iter_state *st = seq->private; 2582 void *rc = NULL; 2583 2584 if (v == SEQ_START_TOKEN) { 2585 rc = tcp_get_idx(seq, 0); 2586 goto out; 2587 } 2588 2589 switch (st->state) { 2590 case TCP_SEQ_STATE_LISTENING: 2591 rc = listening_get_next(seq, v); 2592 if (!rc) { 2593 st->state = TCP_SEQ_STATE_ESTABLISHED; 2594 st->bucket = 0; 2595 st->offset = 0; 2596 rc = established_get_first(seq); 2597 } 2598 break; 2599 case TCP_SEQ_STATE_ESTABLISHED: 2600 rc = established_get_next(seq, v); 2601 break; 2602 } 2603 out: 2604 ++*pos; 2605 st->last_pos = *pos; 2606 return rc; 2607 } 2608 EXPORT_SYMBOL(tcp_seq_next); 2609 2610 void tcp_seq_stop(struct seq_file *seq, void *v) 2611 { 2612 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2613 struct tcp_iter_state *st = seq->private; 2614 2615 switch (st->state) { 2616 case TCP_SEQ_STATE_LISTENING: 2617 if (v != SEQ_START_TOKEN) 2618 spin_unlock(&hinfo->lhash2[st->bucket].lock); 2619 break; 2620 case TCP_SEQ_STATE_ESTABLISHED: 2621 if (v) 2622 spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket)); 2623 break; 2624 } 2625 } 2626 EXPORT_SYMBOL(tcp_seq_stop); 2627 2628 static void get_openreq4(const struct request_sock *req, 2629 struct seq_file *f, int i) 2630 { 2631 const struct inet_request_sock *ireq = inet_rsk(req); 2632 long delta = req->rsk_timer.expires - jiffies; 2633 2634 seq_printf(f, "%4d: %08X:%04X %08X:%04X" 2635 " %02X %08X:%08X %02X:%08lX %08X %5u %8d %u %d %pK", 2636 i, 2637 ireq->ir_loc_addr, 2638 ireq->ir_num, 2639 ireq->ir_rmt_addr, 2640 ntohs(ireq->ir_rmt_port), 2641 TCP_SYN_RECV, 2642 0, 0, /* could print option size, but that is af dependent. */ 2643 1, /* timers active (only the expire timer) */ 2644 jiffies_delta_to_clock_t(delta), 2645 req->num_timeout, 2646 from_kuid_munged(seq_user_ns(f), 2647 sock_i_uid(req->rsk_listener)), 2648 0, /* non standard timer */ 2649 0, /* open_requests have no inode */ 2650 0, 2651 req); 2652 } 2653 2654 static void get_tcp4_sock(struct sock *sk, struct seq_file *f, int i) 2655 { 2656 int timer_active; 2657 unsigned long timer_expires; 2658 const struct tcp_sock *tp = tcp_sk(sk); 2659 const struct inet_connection_sock *icsk = inet_csk(sk); 2660 const struct inet_sock *inet = inet_sk(sk); 2661 const struct fastopen_queue *fastopenq = &icsk->icsk_accept_queue.fastopenq; 2662 __be32 dest = inet->inet_daddr; 2663 __be32 src = inet->inet_rcv_saddr; 2664 __u16 destp = ntohs(inet->inet_dport); 2665 __u16 srcp = ntohs(inet->inet_sport); 2666 int rx_queue; 2667 int state; 2668 2669 if (icsk->icsk_pending == ICSK_TIME_RETRANS || 2670 icsk->icsk_pending == ICSK_TIME_REO_TIMEOUT || 2671 icsk->icsk_pending == ICSK_TIME_LOSS_PROBE) { 2672 timer_active = 1; 2673 timer_expires = icsk->icsk_timeout; 2674 } else if (icsk->icsk_pending == ICSK_TIME_PROBE0) { 2675 timer_active = 4; 2676 timer_expires = icsk->icsk_timeout; 2677 } else if (timer_pending(&sk->sk_timer)) { 2678 timer_active = 2; 2679 timer_expires = sk->sk_timer.expires; 2680 } else { 2681 timer_active = 0; 2682 timer_expires = jiffies; 2683 } 2684 2685 state = inet_sk_state_load(sk); 2686 if (state == TCP_LISTEN) 2687 rx_queue = READ_ONCE(sk->sk_ack_backlog); 2688 else 2689 /* Because we don't lock the socket, 2690 * we might find a transient negative value. 2691 */ 2692 rx_queue = max_t(int, READ_ONCE(tp->rcv_nxt) - 2693 READ_ONCE(tp->copied_seq), 0); 2694 2695 seq_printf(f, "%4d: %08X:%04X %08X:%04X %02X %08X:%08X %02X:%08lX " 2696 "%08X %5u %8d %lu %d %pK %lu %lu %u %u %d", 2697 i, src, srcp, dest, destp, state, 2698 READ_ONCE(tp->write_seq) - tp->snd_una, 2699 rx_queue, 2700 timer_active, 2701 jiffies_delta_to_clock_t(timer_expires - jiffies), 2702 icsk->icsk_retransmits, 2703 from_kuid_munged(seq_user_ns(f), sock_i_uid(sk)), 2704 icsk->icsk_probes_out, 2705 sock_i_ino(sk), 2706 refcount_read(&sk->sk_refcnt), sk, 2707 jiffies_to_clock_t(icsk->icsk_rto), 2708 jiffies_to_clock_t(icsk->icsk_ack.ato), 2709 (icsk->icsk_ack.quick << 1) | inet_csk_in_pingpong_mode(sk), 2710 tcp_snd_cwnd(tp), 2711 state == TCP_LISTEN ? 2712 fastopenq->max_qlen : 2713 (tcp_in_initial_slowstart(tp) ? -1 : tp->snd_ssthresh)); 2714 } 2715 2716 static void get_timewait4_sock(const struct inet_timewait_sock *tw, 2717 struct seq_file *f, int i) 2718 { 2719 long delta = tw->tw_timer.expires - jiffies; 2720 __be32 dest, src; 2721 __u16 destp, srcp; 2722 2723 dest = tw->tw_daddr; 2724 src = tw->tw_rcv_saddr; 2725 destp = ntohs(tw->tw_dport); 2726 srcp = ntohs(tw->tw_sport); 2727 2728 seq_printf(f, "%4d: %08X:%04X %08X:%04X" 2729 " %02X %08X:%08X %02X:%08lX %08X %5d %8d %d %d %pK", 2730 i, src, srcp, dest, destp, tw->tw_substate, 0, 0, 2731 3, jiffies_delta_to_clock_t(delta), 0, 0, 0, 0, 2732 refcount_read(&tw->tw_refcnt), tw); 2733 } 2734 2735 #define TMPSZ 150 2736 2737 static int tcp4_seq_show(struct seq_file *seq, void *v) 2738 { 2739 struct tcp_iter_state *st; 2740 struct sock *sk = v; 2741 2742 seq_setwidth(seq, TMPSZ - 1); 2743 if (v == SEQ_START_TOKEN) { 2744 seq_puts(seq, " sl local_address rem_address st tx_queue " 2745 "rx_queue tr tm->when retrnsmt uid timeout " 2746 "inode"); 2747 goto out; 2748 } 2749 st = seq->private; 2750 2751 if (sk->sk_state == TCP_TIME_WAIT) 2752 get_timewait4_sock(v, seq, st->num); 2753 else if (sk->sk_state == TCP_NEW_SYN_RECV) 2754 get_openreq4(v, seq, st->num); 2755 else 2756 get_tcp4_sock(v, seq, st->num); 2757 out: 2758 seq_pad(seq, '\n'); 2759 return 0; 2760 } 2761 2762 #ifdef CONFIG_BPF_SYSCALL 2763 struct bpf_tcp_iter_state { 2764 struct tcp_iter_state state; 2765 unsigned int cur_sk; 2766 unsigned int end_sk; 2767 unsigned int max_sk; 2768 struct sock **batch; 2769 bool st_bucket_done; 2770 }; 2771 2772 struct bpf_iter__tcp { 2773 __bpf_md_ptr(struct bpf_iter_meta *, meta); 2774 __bpf_md_ptr(struct sock_common *, sk_common); 2775 uid_t uid __aligned(8); 2776 }; 2777 2778 static int tcp_prog_seq_show(struct bpf_prog *prog, struct bpf_iter_meta *meta, 2779 struct sock_common *sk_common, uid_t uid) 2780 { 2781 struct bpf_iter__tcp ctx; 2782 2783 meta->seq_num--; /* skip SEQ_START_TOKEN */ 2784 ctx.meta = meta; 2785 ctx.sk_common = sk_common; 2786 ctx.uid = uid; 2787 return bpf_iter_run_prog(prog, &ctx); 2788 } 2789 2790 static void bpf_iter_tcp_put_batch(struct bpf_tcp_iter_state *iter) 2791 { 2792 while (iter->cur_sk < iter->end_sk) 2793 sock_gen_put(iter->batch[iter->cur_sk++]); 2794 } 2795 2796 static int bpf_iter_tcp_realloc_batch(struct bpf_tcp_iter_state *iter, 2797 unsigned int new_batch_sz) 2798 { 2799 struct sock **new_batch; 2800 2801 new_batch = kvmalloc(sizeof(*new_batch) * new_batch_sz, 2802 GFP_USER | __GFP_NOWARN); 2803 if (!new_batch) 2804 return -ENOMEM; 2805 2806 bpf_iter_tcp_put_batch(iter); 2807 kvfree(iter->batch); 2808 iter->batch = new_batch; 2809 iter->max_sk = new_batch_sz; 2810 2811 return 0; 2812 } 2813 2814 static unsigned int bpf_iter_tcp_listening_batch(struct seq_file *seq, 2815 struct sock *start_sk) 2816 { 2817 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2818 struct bpf_tcp_iter_state *iter = seq->private; 2819 struct tcp_iter_state *st = &iter->state; 2820 struct hlist_nulls_node *node; 2821 unsigned int expected = 1; 2822 struct sock *sk; 2823 2824 sock_hold(start_sk); 2825 iter->batch[iter->end_sk++] = start_sk; 2826 2827 sk = sk_nulls_next(start_sk); 2828 sk_nulls_for_each_from(sk, node) { 2829 if (seq_sk_match(seq, sk)) { 2830 if (iter->end_sk < iter->max_sk) { 2831 sock_hold(sk); 2832 iter->batch[iter->end_sk++] = sk; 2833 } 2834 expected++; 2835 } 2836 } 2837 spin_unlock(&hinfo->lhash2[st->bucket].lock); 2838 2839 return expected; 2840 } 2841 2842 static unsigned int bpf_iter_tcp_established_batch(struct seq_file *seq, 2843 struct sock *start_sk) 2844 { 2845 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2846 struct bpf_tcp_iter_state *iter = seq->private; 2847 struct tcp_iter_state *st = &iter->state; 2848 struct hlist_nulls_node *node; 2849 unsigned int expected = 1; 2850 struct sock *sk; 2851 2852 sock_hold(start_sk); 2853 iter->batch[iter->end_sk++] = start_sk; 2854 2855 sk = sk_nulls_next(start_sk); 2856 sk_nulls_for_each_from(sk, node) { 2857 if (seq_sk_match(seq, sk)) { 2858 if (iter->end_sk < iter->max_sk) { 2859 sock_hold(sk); 2860 iter->batch[iter->end_sk++] = sk; 2861 } 2862 expected++; 2863 } 2864 } 2865 spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket)); 2866 2867 return expected; 2868 } 2869 2870 static struct sock *bpf_iter_tcp_batch(struct seq_file *seq) 2871 { 2872 struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo; 2873 struct bpf_tcp_iter_state *iter = seq->private; 2874 struct tcp_iter_state *st = &iter->state; 2875 unsigned int expected; 2876 bool resized = false; 2877 struct sock *sk; 2878 2879 /* The st->bucket is done. Directly advance to the next 2880 * bucket instead of having the tcp_seek_last_pos() to skip 2881 * one by one in the current bucket and eventually find out 2882 * it has to advance to the next bucket. 2883 */ 2884 if (iter->st_bucket_done) { 2885 st->offset = 0; 2886 st->bucket++; 2887 if (st->state == TCP_SEQ_STATE_LISTENING && 2888 st->bucket > hinfo->lhash2_mask) { 2889 st->state = TCP_SEQ_STATE_ESTABLISHED; 2890 st->bucket = 0; 2891 } 2892 } 2893 2894 again: 2895 /* Get a new batch */ 2896 iter->cur_sk = 0; 2897 iter->end_sk = 0; 2898 iter->st_bucket_done = false; 2899 2900 sk = tcp_seek_last_pos(seq); 2901 if (!sk) 2902 return NULL; /* Done */ 2903 2904 if (st->state == TCP_SEQ_STATE_LISTENING) 2905 expected = bpf_iter_tcp_listening_batch(seq, sk); 2906 else 2907 expected = bpf_iter_tcp_established_batch(seq, sk); 2908 2909 if (iter->end_sk == expected) { 2910 iter->st_bucket_done = true; 2911 return sk; 2912 } 2913 2914 if (!resized && !bpf_iter_tcp_realloc_batch(iter, expected * 3 / 2)) { 2915 resized = true; 2916 goto again; 2917 } 2918 2919 return sk; 2920 } 2921 2922 static void *bpf_iter_tcp_seq_start(struct seq_file *seq, loff_t *pos) 2923 { 2924 /* bpf iter does not support lseek, so it always 2925 * continue from where it was stop()-ped. 2926 */ 2927 if (*pos) 2928 return bpf_iter_tcp_batch(seq); 2929 2930 return SEQ_START_TOKEN; 2931 } 2932 2933 static void *bpf_iter_tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos) 2934 { 2935 struct bpf_tcp_iter_state *iter = seq->private; 2936 struct tcp_iter_state *st = &iter->state; 2937 struct sock *sk; 2938 2939 /* Whenever seq_next() is called, the iter->cur_sk is 2940 * done with seq_show(), so advance to the next sk in 2941 * the batch. 2942 */ 2943 if (iter->cur_sk < iter->end_sk) { 2944 /* Keeping st->num consistent in tcp_iter_state. 2945 * bpf_iter_tcp does not use st->num. 2946 * meta.seq_num is used instead. 2947 */ 2948 st->num++; 2949 /* Move st->offset to the next sk in the bucket such that 2950 * the future start() will resume at st->offset in 2951 * st->bucket. See tcp_seek_last_pos(). 2952 */ 2953 st->offset++; 2954 sock_gen_put(iter->batch[iter->cur_sk++]); 2955 } 2956 2957 if (iter->cur_sk < iter->end_sk) 2958 sk = iter->batch[iter->cur_sk]; 2959 else 2960 sk = bpf_iter_tcp_batch(seq); 2961 2962 ++*pos; 2963 /* Keeping st->last_pos consistent in tcp_iter_state. 2964 * bpf iter does not do lseek, so st->last_pos always equals to *pos. 2965 */ 2966 st->last_pos = *pos; 2967 return sk; 2968 } 2969 2970 static int bpf_iter_tcp_seq_show(struct seq_file *seq, void *v) 2971 { 2972 struct bpf_iter_meta meta; 2973 struct bpf_prog *prog; 2974 struct sock *sk = v; 2975 uid_t uid; 2976 int ret; 2977 2978 if (v == SEQ_START_TOKEN) 2979 return 0; 2980 2981 if (sk_fullsock(sk)) 2982 lock_sock(sk); 2983 2984 if (unlikely(sk_unhashed(sk))) { 2985 ret = SEQ_SKIP; 2986 goto unlock; 2987 } 2988 2989 if (sk->sk_state == TCP_TIME_WAIT) { 2990 uid = 0; 2991 } else if (sk->sk_state == TCP_NEW_SYN_RECV) { 2992 const struct request_sock *req = v; 2993 2994 uid = from_kuid_munged(seq_user_ns(seq), 2995 sock_i_uid(req->rsk_listener)); 2996 } else { 2997 uid = from_kuid_munged(seq_user_ns(seq), sock_i_uid(sk)); 2998 } 2999 3000 meta.seq = seq; 3001 prog = bpf_iter_get_info(&meta, false); 3002 ret = tcp_prog_seq_show(prog, &meta, v, uid); 3003 3004 unlock: 3005 if (sk_fullsock(sk)) 3006 release_sock(sk); 3007 return ret; 3008 3009 } 3010 3011 static void bpf_iter_tcp_seq_stop(struct seq_file *seq, void *v) 3012 { 3013 struct bpf_tcp_iter_state *iter = seq->private; 3014 struct bpf_iter_meta meta; 3015 struct bpf_prog *prog; 3016 3017 if (!v) { 3018 meta.seq = seq; 3019 prog = bpf_iter_get_info(&meta, true); 3020 if (prog) 3021 (void)tcp_prog_seq_show(prog, &meta, v, 0); 3022 } 3023 3024 if (iter->cur_sk < iter->end_sk) { 3025 bpf_iter_tcp_put_batch(iter); 3026 iter->st_bucket_done = false; 3027 } 3028 } 3029 3030 static const struct seq_operations bpf_iter_tcp_seq_ops = { 3031 .show = bpf_iter_tcp_seq_show, 3032 .start = bpf_iter_tcp_seq_start, 3033 .next = bpf_iter_tcp_seq_next, 3034 .stop = bpf_iter_tcp_seq_stop, 3035 }; 3036 #endif 3037 static unsigned short seq_file_family(const struct seq_file *seq) 3038 { 3039 const struct tcp_seq_afinfo *afinfo; 3040 3041 #ifdef CONFIG_BPF_SYSCALL 3042 /* Iterated from bpf_iter. Let the bpf prog to filter instead. */ 3043 if (seq->op == &bpf_iter_tcp_seq_ops) 3044 return AF_UNSPEC; 3045 #endif 3046 3047 /* Iterated from proc fs */ 3048 afinfo = pde_data(file_inode(seq->file)); 3049 return afinfo->family; 3050 } 3051 3052 static const struct seq_operations tcp4_seq_ops = { 3053 .show = tcp4_seq_show, 3054 .start = tcp_seq_start, 3055 .next = tcp_seq_next, 3056 .stop = tcp_seq_stop, 3057 }; 3058 3059 static struct tcp_seq_afinfo tcp4_seq_afinfo = { 3060 .family = AF_INET, 3061 }; 3062 3063 static int __net_init tcp4_proc_init_net(struct net *net) 3064 { 3065 if (!proc_create_net_data("tcp", 0444, net->proc_net, &tcp4_seq_ops, 3066 sizeof(struct tcp_iter_state), &tcp4_seq_afinfo)) 3067 return -ENOMEM; 3068 return 0; 3069 } 3070 3071 static void __net_exit tcp4_proc_exit_net(struct net *net) 3072 { 3073 remove_proc_entry("tcp", net->proc_net); 3074 } 3075 3076 static struct pernet_operations tcp4_net_ops = { 3077 .init = tcp4_proc_init_net, 3078 .exit = tcp4_proc_exit_net, 3079 }; 3080 3081 int __init tcp4_proc_init(void) 3082 { 3083 return register_pernet_subsys(&tcp4_net_ops); 3084 } 3085 3086 void tcp4_proc_exit(void) 3087 { 3088 unregister_pernet_subsys(&tcp4_net_ops); 3089 } 3090 #endif /* CONFIG_PROC_FS */ 3091 3092 /* @wake is one when sk_stream_write_space() calls us. 3093 * This sends EPOLLOUT only if notsent_bytes is half the limit. 3094 * This mimics the strategy used in sock_def_write_space(). 3095 */ 3096 bool tcp_stream_memory_free(const struct sock *sk, int wake) 3097 { 3098 const struct tcp_sock *tp = tcp_sk(sk); 3099 u32 notsent_bytes = READ_ONCE(tp->write_seq) - 3100 READ_ONCE(tp->snd_nxt); 3101 3102 return (notsent_bytes << wake) < tcp_notsent_lowat(tp); 3103 } 3104 EXPORT_SYMBOL(tcp_stream_memory_free); 3105 3106 struct proto tcp_prot = { 3107 .name = "TCP", 3108 .owner = THIS_MODULE, 3109 .close = tcp_close, 3110 .pre_connect = tcp_v4_pre_connect, 3111 .connect = tcp_v4_connect, 3112 .disconnect = tcp_disconnect, 3113 .accept = inet_csk_accept, 3114 .ioctl = tcp_ioctl, 3115 .init = tcp_v4_init_sock, 3116 .destroy = tcp_v4_destroy_sock, 3117 .shutdown = tcp_shutdown, 3118 .setsockopt = tcp_setsockopt, 3119 .getsockopt = tcp_getsockopt, 3120 .bpf_bypass_getsockopt = tcp_bpf_bypass_getsockopt, 3121 .keepalive = tcp_set_keepalive, 3122 .recvmsg = tcp_recvmsg, 3123 .sendmsg = tcp_sendmsg, 3124 .splice_eof = tcp_splice_eof, 3125 .backlog_rcv = tcp_v4_do_rcv, 3126 .release_cb = tcp_release_cb, 3127 .hash = inet_hash, 3128 .unhash = inet_unhash, 3129 .get_port = inet_csk_get_port, 3130 .put_port = inet_put_port, 3131 #ifdef CONFIG_BPF_SYSCALL 3132 .psock_update_sk_prot = tcp_bpf_update_proto, 3133 #endif 3134 .enter_memory_pressure = tcp_enter_memory_pressure, 3135 .leave_memory_pressure = tcp_leave_memory_pressure, 3136 .stream_memory_free = tcp_stream_memory_free, 3137 .sockets_allocated = &tcp_sockets_allocated, 3138 .orphan_count = &tcp_orphan_count, 3139 3140 .memory_allocated = &tcp_memory_allocated, 3141 .per_cpu_fw_alloc = &tcp_memory_per_cpu_fw_alloc, 3142 3143 .memory_pressure = &tcp_memory_pressure, 3144 .sysctl_mem = sysctl_tcp_mem, 3145 .sysctl_wmem_offset = offsetof(struct net, ipv4.sysctl_tcp_wmem), 3146 .sysctl_rmem_offset = offsetof(struct net, ipv4.sysctl_tcp_rmem), 3147 .max_header = MAX_TCP_HEADER, 3148 .obj_size = sizeof(struct tcp_sock), 3149 .slab_flags = SLAB_TYPESAFE_BY_RCU, 3150 .twsk_prot = &tcp_timewait_sock_ops, 3151 .rsk_prot = &tcp_request_sock_ops, 3152 .h.hashinfo = NULL, 3153 .no_autobind = true, 3154 .diag_destroy = tcp_abort, 3155 }; 3156 EXPORT_SYMBOL(tcp_prot); 3157 3158 static void __net_exit tcp_sk_exit(struct net *net) 3159 { 3160 if (net->ipv4.tcp_congestion_control) 3161 bpf_module_put(net->ipv4.tcp_congestion_control, 3162 net->ipv4.tcp_congestion_control->owner); 3163 } 3164 3165 static void __net_init tcp_set_hashinfo(struct net *net) 3166 { 3167 struct inet_hashinfo *hinfo; 3168 unsigned int ehash_entries; 3169 struct net *old_net; 3170 3171 if (net_eq(net, &init_net)) 3172 goto fallback; 3173 3174 old_net = current->nsproxy->net_ns; 3175 ehash_entries = READ_ONCE(old_net->ipv4.sysctl_tcp_child_ehash_entries); 3176 if (!ehash_entries) 3177 goto fallback; 3178 3179 ehash_entries = roundup_pow_of_two(ehash_entries); 3180 hinfo = inet_pernet_hashinfo_alloc(&tcp_hashinfo, ehash_entries); 3181 if (!hinfo) { 3182 pr_warn("Failed to allocate TCP ehash (entries: %u) " 3183 "for a netns, fallback to the global one\n", 3184 ehash_entries); 3185 fallback: 3186 hinfo = &tcp_hashinfo; 3187 ehash_entries = tcp_hashinfo.ehash_mask + 1; 3188 } 3189 3190 net->ipv4.tcp_death_row.hashinfo = hinfo; 3191 net->ipv4.tcp_death_row.sysctl_max_tw_buckets = ehash_entries / 2; 3192 net->ipv4.sysctl_max_syn_backlog = max(128U, ehash_entries / 128); 3193 } 3194 3195 static int __net_init tcp_sk_init(struct net *net) 3196 { 3197 net->ipv4.sysctl_tcp_ecn = 2; 3198 net->ipv4.sysctl_tcp_ecn_fallback = 1; 3199 3200 net->ipv4.sysctl_tcp_base_mss = TCP_BASE_MSS; 3201 net->ipv4.sysctl_tcp_min_snd_mss = TCP_MIN_SND_MSS; 3202 net->ipv4.sysctl_tcp_probe_threshold = TCP_PROBE_THRESHOLD; 3203 net->ipv4.sysctl_tcp_probe_interval = TCP_PROBE_INTERVAL; 3204 net->ipv4.sysctl_tcp_mtu_probe_floor = TCP_MIN_SND_MSS; 3205 3206 net->ipv4.sysctl_tcp_keepalive_time = TCP_KEEPALIVE_TIME; 3207 net->ipv4.sysctl_tcp_keepalive_probes = TCP_KEEPALIVE_PROBES; 3208 net->ipv4.sysctl_tcp_keepalive_intvl = TCP_KEEPALIVE_INTVL; 3209 3210 net->ipv4.sysctl_tcp_syn_retries = TCP_SYN_RETRIES; 3211 net->ipv4.sysctl_tcp_synack_retries = TCP_SYNACK_RETRIES; 3212 net->ipv4.sysctl_tcp_syncookies = 1; 3213 net->ipv4.sysctl_tcp_reordering = TCP_FASTRETRANS_THRESH; 3214 net->ipv4.sysctl_tcp_retries1 = TCP_RETR1; 3215 net->ipv4.sysctl_tcp_retries2 = TCP_RETR2; 3216 net->ipv4.sysctl_tcp_orphan_retries = 0; 3217 net->ipv4.sysctl_tcp_fin_timeout = TCP_FIN_TIMEOUT; 3218 net->ipv4.sysctl_tcp_notsent_lowat = UINT_MAX; 3219 net->ipv4.sysctl_tcp_tw_reuse = 2; 3220 net->ipv4.sysctl_tcp_no_ssthresh_metrics_save = 1; 3221 3222 refcount_set(&net->ipv4.tcp_death_row.tw_refcount, 1); 3223 tcp_set_hashinfo(net); 3224 3225 net->ipv4.sysctl_tcp_sack = 1; 3226 net->ipv4.sysctl_tcp_window_scaling = 1; 3227 net->ipv4.sysctl_tcp_timestamps = 1; 3228 net->ipv4.sysctl_tcp_early_retrans = 3; 3229 net->ipv4.sysctl_tcp_recovery = TCP_RACK_LOSS_DETECTION; 3230 net->ipv4.sysctl_tcp_slow_start_after_idle = 1; /* By default, RFC2861 behavior. */ 3231 net->ipv4.sysctl_tcp_retrans_collapse = 1; 3232 net->ipv4.sysctl_tcp_max_reordering = 300; 3233 net->ipv4.sysctl_tcp_dsack = 1; 3234 net->ipv4.sysctl_tcp_app_win = 31; 3235 net->ipv4.sysctl_tcp_adv_win_scale = 1; 3236 net->ipv4.sysctl_tcp_frto = 2; 3237 net->ipv4.sysctl_tcp_moderate_rcvbuf = 1; 3238 /* This limits the percentage of the congestion window which we 3239 * will allow a single TSO frame to consume. Building TSO frames 3240 * which are too large can cause TCP streams to be bursty. 3241 */ 3242 net->ipv4.sysctl_tcp_tso_win_divisor = 3; 3243 /* Default TSQ limit of 16 TSO segments */ 3244 net->ipv4.sysctl_tcp_limit_output_bytes = 16 * 65536; 3245 3246 /* rfc5961 challenge ack rate limiting, per net-ns, disabled by default. */ 3247 net->ipv4.sysctl_tcp_challenge_ack_limit = INT_MAX; 3248 3249 net->ipv4.sysctl_tcp_min_tso_segs = 2; 3250 net->ipv4.sysctl_tcp_tso_rtt_log = 9; /* 2^9 = 512 usec */ 3251 net->ipv4.sysctl_tcp_min_rtt_wlen = 300; 3252 net->ipv4.sysctl_tcp_autocorking = 1; 3253 net->ipv4.sysctl_tcp_invalid_ratelimit = HZ/2; 3254 net->ipv4.sysctl_tcp_pacing_ss_ratio = 200; 3255 net->ipv4.sysctl_tcp_pacing_ca_ratio = 120; 3256 if (net != &init_net) { 3257 memcpy(net->ipv4.sysctl_tcp_rmem, 3258 init_net.ipv4.sysctl_tcp_rmem, 3259 sizeof(init_net.ipv4.sysctl_tcp_rmem)); 3260 memcpy(net->ipv4.sysctl_tcp_wmem, 3261 init_net.ipv4.sysctl_tcp_wmem, 3262 sizeof(init_net.ipv4.sysctl_tcp_wmem)); 3263 } 3264 net->ipv4.sysctl_tcp_comp_sack_delay_ns = NSEC_PER_MSEC; 3265 net->ipv4.sysctl_tcp_comp_sack_slack_ns = 100 * NSEC_PER_USEC; 3266 net->ipv4.sysctl_tcp_comp_sack_nr = 44; 3267 net->ipv4.sysctl_tcp_fastopen = TFO_CLIENT_ENABLE; 3268 net->ipv4.sysctl_tcp_fastopen_blackhole_timeout = 0; 3269 atomic_set(&net->ipv4.tfo_active_disable_times, 0); 3270 3271 /* Set default values for PLB */ 3272 net->ipv4.sysctl_tcp_plb_enabled = 0; /* Disabled by default */ 3273 net->ipv4.sysctl_tcp_plb_idle_rehash_rounds = 3; 3274 net->ipv4.sysctl_tcp_plb_rehash_rounds = 12; 3275 net->ipv4.sysctl_tcp_plb_suspend_rto_sec = 60; 3276 /* Default congestion threshold for PLB to mark a round is 50% */ 3277 net->ipv4.sysctl_tcp_plb_cong_thresh = (1 << TCP_PLB_SCALE) / 2; 3278 3279 /* Reno is always built in */ 3280 if (!net_eq(net, &init_net) && 3281 bpf_try_module_get(init_net.ipv4.tcp_congestion_control, 3282 init_net.ipv4.tcp_congestion_control->owner)) 3283 net->ipv4.tcp_congestion_control = init_net.ipv4.tcp_congestion_control; 3284 else 3285 net->ipv4.tcp_congestion_control = &tcp_reno; 3286 3287 net->ipv4.sysctl_tcp_syn_linear_timeouts = 4; 3288 net->ipv4.sysctl_tcp_shrink_window = 0; 3289 3290 return 0; 3291 } 3292 3293 static void __net_exit tcp_sk_exit_batch(struct list_head *net_exit_list) 3294 { 3295 struct net *net; 3296 3297 tcp_twsk_purge(net_exit_list, AF_INET); 3298 3299 list_for_each_entry(net, net_exit_list, exit_list) { 3300 inet_pernet_hashinfo_free(net->ipv4.tcp_death_row.hashinfo); 3301 WARN_ON_ONCE(!refcount_dec_and_test(&net->ipv4.tcp_death_row.tw_refcount)); 3302 tcp_fastopen_ctx_destroy(net); 3303 } 3304 } 3305 3306 static struct pernet_operations __net_initdata tcp_sk_ops = { 3307 .init = tcp_sk_init, 3308 .exit = tcp_sk_exit, 3309 .exit_batch = tcp_sk_exit_batch, 3310 }; 3311 3312 #if defined(CONFIG_BPF_SYSCALL) && defined(CONFIG_PROC_FS) 3313 DEFINE_BPF_ITER_FUNC(tcp, struct bpf_iter_meta *meta, 3314 struct sock_common *sk_common, uid_t uid) 3315 3316 #define INIT_BATCH_SZ 16 3317 3318 static int bpf_iter_init_tcp(void *priv_data, struct bpf_iter_aux_info *aux) 3319 { 3320 struct bpf_tcp_iter_state *iter = priv_data; 3321 int err; 3322 3323 err = bpf_iter_init_seq_net(priv_data, aux); 3324 if (err) 3325 return err; 3326 3327 err = bpf_iter_tcp_realloc_batch(iter, INIT_BATCH_SZ); 3328 if (err) { 3329 bpf_iter_fini_seq_net(priv_data); 3330 return err; 3331 } 3332 3333 return 0; 3334 } 3335 3336 static void bpf_iter_fini_tcp(void *priv_data) 3337 { 3338 struct bpf_tcp_iter_state *iter = priv_data; 3339 3340 bpf_iter_fini_seq_net(priv_data); 3341 kvfree(iter->batch); 3342 } 3343 3344 static const struct bpf_iter_seq_info tcp_seq_info = { 3345 .seq_ops = &bpf_iter_tcp_seq_ops, 3346 .init_seq_private = bpf_iter_init_tcp, 3347 .fini_seq_private = bpf_iter_fini_tcp, 3348 .seq_priv_size = sizeof(struct bpf_tcp_iter_state), 3349 }; 3350 3351 static const struct bpf_func_proto * 3352 bpf_iter_tcp_get_func_proto(enum bpf_func_id func_id, 3353 const struct bpf_prog *prog) 3354 { 3355 switch (func_id) { 3356 case BPF_FUNC_setsockopt: 3357 return &bpf_sk_setsockopt_proto; 3358 case BPF_FUNC_getsockopt: 3359 return &bpf_sk_getsockopt_proto; 3360 default: 3361 return NULL; 3362 } 3363 } 3364 3365 static struct bpf_iter_reg tcp_reg_info = { 3366 .target = "tcp", 3367 .ctx_arg_info_size = 1, 3368 .ctx_arg_info = { 3369 { offsetof(struct bpf_iter__tcp, sk_common), 3370 PTR_TO_BTF_ID_OR_NULL | PTR_TRUSTED }, 3371 }, 3372 .get_func_proto = bpf_iter_tcp_get_func_proto, 3373 .seq_info = &tcp_seq_info, 3374 }; 3375 3376 static void __init bpf_iter_register(void) 3377 { 3378 tcp_reg_info.ctx_arg_info[0].btf_id = btf_sock_ids[BTF_SOCK_TYPE_SOCK_COMMON]; 3379 if (bpf_iter_reg_target(&tcp_reg_info)) 3380 pr_warn("Warning: could not register bpf iterator tcp\n"); 3381 } 3382 3383 #endif 3384 3385 void __init tcp_v4_init(void) 3386 { 3387 int cpu, res; 3388 3389 for_each_possible_cpu(cpu) { 3390 struct sock *sk; 3391 3392 res = inet_ctl_sock_create(&sk, PF_INET, SOCK_RAW, 3393 IPPROTO_TCP, &init_net); 3394 if (res) 3395 panic("Failed to create the TCP control socket.\n"); 3396 sock_set_flag(sk, SOCK_USE_WRITE_QUEUE); 3397 3398 /* Please enforce IP_DF and IPID==0 for RST and 3399 * ACK sent in SYN-RECV and TIME-WAIT state. 3400 */ 3401 inet_sk(sk)->pmtudisc = IP_PMTUDISC_DO; 3402 3403 per_cpu(ipv4_tcp_sk, cpu) = sk; 3404 } 3405 if (register_pernet_subsys(&tcp_sk_ops)) 3406 panic("Failed to create the TCP control socket.\n"); 3407 3408 #if defined(CONFIG_BPF_SYSCALL) && defined(CONFIG_PROC_FS) 3409 bpf_iter_register(); 3410 #endif 3411 } 3412