1 // SPDX-License-Identifier: GPL-2.0 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 * The IP fragmentation functionality. 8 * 9 * Authors: Fred N. van Kempen <waltje@uWalt.NL.Mugnet.ORG> 10 * Alan Cox <alan@lxorguk.ukuu.org.uk> 11 * 12 * Fixes: 13 * Alan Cox : Split from ip.c , see ip_input.c for history. 14 * David S. Miller : Begin massive cleanup... 15 * Andi Kleen : Add sysctls. 16 * xxxx : Overlapfrag bug. 17 * Ultima : ip_expire() kernel panic. 18 * Bill Hawes : Frag accounting and evictor fixes. 19 * John McDonald : 0 length frag bug. 20 * Alexey Kuznetsov: SMP races, threading, cleanup. 21 * Patrick McHardy : LRU queue of frag heads for evictor. 22 */ 23 24 #define pr_fmt(fmt) "IPv4: " fmt 25 26 #include <linux/compiler.h> 27 #include <linux/module.h> 28 #include <linux/types.h> 29 #include <linux/mm.h> 30 #include <linux/jiffies.h> 31 #include <linux/skbuff.h> 32 #include <linux/list.h> 33 #include <linux/ip.h> 34 #include <linux/icmp.h> 35 #include <linux/netdevice.h> 36 #include <linux/jhash.h> 37 #include <linux/random.h> 38 #include <linux/slab.h> 39 #include <net/route.h> 40 #include <net/dst.h> 41 #include <net/sock.h> 42 #include <net/ip.h> 43 #include <net/icmp.h> 44 #include <net/checksum.h> 45 #include <net/inetpeer.h> 46 #include <net/inet_frag.h> 47 #include <linux/tcp.h> 48 #include <linux/udp.h> 49 #include <linux/inet.h> 50 #include <linux/netfilter_ipv4.h> 51 #include <net/inet_ecn.h> 52 #include <net/l3mdev.h> 53 54 /* NOTE. Logic of IP defragmentation is parallel to corresponding IPv6 55 * code now. If you change something here, _PLEASE_ update ipv6/reassembly.c 56 * as well. Or notify me, at least. --ANK 57 */ 58 static const char ip_frag_cache_name[] = "ip4-frags"; 59 60 /* Describe an entry in the "incomplete datagrams" queue. */ 61 struct ipq { 62 struct inet_frag_queue q; 63 64 u8 ecn; /* RFC3168 support */ 65 u16 max_df_size; /* largest frag with DF set seen */ 66 int iif; 67 unsigned int rid; 68 struct inet_peer *peer; 69 }; 70 71 static u8 ip4_frag_ecn(u8 tos) 72 { 73 return 1 << (tos & INET_ECN_MASK); 74 } 75 76 static struct inet_frags ip4_frags; 77 78 static int ip_frag_reasm(struct ipq *qp, struct sk_buff *skb, 79 struct sk_buff *prev_tail, struct net_device *dev); 80 81 82 static void ip4_frag_init(struct inet_frag_queue *q, const void *a) 83 { 84 struct ipq *qp = container_of(q, struct ipq, q); 85 struct net *net = q->fqdir->net; 86 87 const struct frag_v4_compare_key *key = a; 88 89 q->key.v4 = *key; 90 qp->ecn = 0; 91 qp->peer = q->fqdir->max_dist ? 92 inet_getpeer_v4(net->ipv4.peers, key->saddr, key->vif, 1) : 93 NULL; 94 } 95 96 static void ip4_frag_free(struct inet_frag_queue *q) 97 { 98 struct ipq *qp; 99 100 qp = container_of(q, struct ipq, q); 101 if (qp->peer) 102 inet_putpeer(qp->peer); 103 } 104 105 106 /* Destruction primitives. */ 107 108 static void ipq_put(struct ipq *ipq) 109 { 110 inet_frag_put(&ipq->q); 111 } 112 113 /* Kill ipq entry. It is not destroyed immediately, 114 * because caller (and someone more) holds reference count. 115 */ 116 static void ipq_kill(struct ipq *ipq) 117 { 118 inet_frag_kill(&ipq->q); 119 } 120 121 static bool frag_expire_skip_icmp(u32 user) 122 { 123 return user == IP_DEFRAG_AF_PACKET || 124 ip_defrag_user_in_between(user, IP_DEFRAG_CONNTRACK_IN, 125 __IP_DEFRAG_CONNTRACK_IN_END) || 126 ip_defrag_user_in_between(user, IP_DEFRAG_CONNTRACK_BRIDGE_IN, 127 __IP_DEFRAG_CONNTRACK_BRIDGE_IN); 128 } 129 130 /* 131 * Oops, a fragment queue timed out. Kill it and send an ICMP reply. 132 */ 133 static void ip_expire(struct timer_list *t) 134 { 135 enum skb_drop_reason reason = SKB_DROP_REASON_FRAG_REASM_TIMEOUT; 136 struct inet_frag_queue *frag = from_timer(frag, t, timer); 137 const struct iphdr *iph; 138 struct sk_buff *head = NULL; 139 struct net *net; 140 struct ipq *qp; 141 142 qp = container_of(frag, struct ipq, q); 143 net = qp->q.fqdir->net; 144 145 rcu_read_lock(); 146 147 /* Paired with WRITE_ONCE() in fqdir_pre_exit(). */ 148 if (READ_ONCE(qp->q.fqdir->dead)) 149 goto out_rcu_unlock; 150 151 spin_lock(&qp->q.lock); 152 153 if (qp->q.flags & INET_FRAG_COMPLETE) 154 goto out; 155 156 qp->q.flags |= INET_FRAG_DROP; 157 ipq_kill(qp); 158 __IP_INC_STATS(net, IPSTATS_MIB_REASMFAILS); 159 __IP_INC_STATS(net, IPSTATS_MIB_REASMTIMEOUT); 160 161 if (!(qp->q.flags & INET_FRAG_FIRST_IN)) 162 goto out; 163 164 /* sk_buff::dev and sk_buff::rbnode are unionized. So we 165 * pull the head out of the tree in order to be able to 166 * deal with head->dev. 167 */ 168 head = inet_frag_pull_head(&qp->q); 169 if (!head) 170 goto out; 171 head->dev = dev_get_by_index_rcu(net, qp->iif); 172 if (!head->dev) 173 goto out; 174 175 176 /* skb has no dst, perform route lookup again */ 177 iph = ip_hdr(head); 178 reason = ip_route_input_noref(head, iph->daddr, iph->saddr, 179 ip4h_dscp(iph), head->dev); 180 if (reason) 181 goto out; 182 183 /* Only an end host needs to send an ICMP 184 * "Fragment Reassembly Timeout" message, per RFC792. 185 */ 186 reason = SKB_DROP_REASON_FRAG_REASM_TIMEOUT; 187 if (frag_expire_skip_icmp(qp->q.key.v4.user) && 188 (skb_rtable(head)->rt_type != RTN_LOCAL)) 189 goto out; 190 191 spin_unlock(&qp->q.lock); 192 icmp_send(head, ICMP_TIME_EXCEEDED, ICMP_EXC_FRAGTIME, 0); 193 goto out_rcu_unlock; 194 195 out: 196 spin_unlock(&qp->q.lock); 197 out_rcu_unlock: 198 rcu_read_unlock(); 199 kfree_skb_reason(head, reason); 200 ipq_put(qp); 201 } 202 203 /* Find the correct entry in the "incomplete datagrams" queue for 204 * this IP datagram, and create new one, if nothing is found. 205 */ 206 static struct ipq *ip_find(struct net *net, struct iphdr *iph, 207 u32 user, int vif) 208 { 209 struct frag_v4_compare_key key = { 210 .saddr = iph->saddr, 211 .daddr = iph->daddr, 212 .user = user, 213 .vif = vif, 214 .id = iph->id, 215 .protocol = iph->protocol, 216 }; 217 struct inet_frag_queue *q; 218 219 q = inet_frag_find(net->ipv4.fqdir, &key); 220 if (!q) 221 return NULL; 222 223 return container_of(q, struct ipq, q); 224 } 225 226 /* Is the fragment too far ahead to be part of ipq? */ 227 static int ip_frag_too_far(struct ipq *qp) 228 { 229 struct inet_peer *peer = qp->peer; 230 unsigned int max = qp->q.fqdir->max_dist; 231 unsigned int start, end; 232 233 int rc; 234 235 if (!peer || !max) 236 return 0; 237 238 start = qp->rid; 239 end = atomic_inc_return(&peer->rid); 240 qp->rid = end; 241 242 rc = qp->q.fragments_tail && (end - start) > max; 243 244 if (rc) 245 __IP_INC_STATS(qp->q.fqdir->net, IPSTATS_MIB_REASMFAILS); 246 247 return rc; 248 } 249 250 static int ip_frag_reinit(struct ipq *qp) 251 { 252 unsigned int sum_truesize = 0; 253 254 if (!mod_timer(&qp->q.timer, jiffies + qp->q.fqdir->timeout)) { 255 refcount_inc(&qp->q.refcnt); 256 return -ETIMEDOUT; 257 } 258 259 sum_truesize = inet_frag_rbtree_purge(&qp->q.rb_fragments, 260 SKB_DROP_REASON_FRAG_TOO_FAR); 261 sub_frag_mem_limit(qp->q.fqdir, sum_truesize); 262 263 qp->q.flags = 0; 264 qp->q.len = 0; 265 qp->q.meat = 0; 266 qp->q.rb_fragments = RB_ROOT; 267 qp->q.fragments_tail = NULL; 268 qp->q.last_run_head = NULL; 269 qp->iif = 0; 270 qp->ecn = 0; 271 272 return 0; 273 } 274 275 /* Add new segment to existing queue. */ 276 static int ip_frag_queue(struct ipq *qp, struct sk_buff *skb) 277 { 278 struct net *net = qp->q.fqdir->net; 279 int ihl, end, flags, offset; 280 struct sk_buff *prev_tail; 281 struct net_device *dev; 282 unsigned int fragsize; 283 int err = -ENOENT; 284 SKB_DR(reason); 285 u8 ecn; 286 287 /* If reassembly is already done, @skb must be a duplicate frag. */ 288 if (qp->q.flags & INET_FRAG_COMPLETE) { 289 SKB_DR_SET(reason, DUP_FRAG); 290 goto err; 291 } 292 293 if (!(IPCB(skb)->flags & IPSKB_FRAG_COMPLETE) && 294 unlikely(ip_frag_too_far(qp)) && 295 unlikely(err = ip_frag_reinit(qp))) { 296 ipq_kill(qp); 297 goto err; 298 } 299 300 ecn = ip4_frag_ecn(ip_hdr(skb)->tos); 301 offset = ntohs(ip_hdr(skb)->frag_off); 302 flags = offset & ~IP_OFFSET; 303 offset &= IP_OFFSET; 304 offset <<= 3; /* offset is in 8-byte chunks */ 305 ihl = ip_hdrlen(skb); 306 307 /* Determine the position of this fragment. */ 308 end = offset + skb->len - skb_network_offset(skb) - ihl; 309 err = -EINVAL; 310 311 /* Is this the final fragment? */ 312 if ((flags & IP_MF) == 0) { 313 /* If we already have some bits beyond end 314 * or have different end, the segment is corrupted. 315 */ 316 if (end < qp->q.len || 317 ((qp->q.flags & INET_FRAG_LAST_IN) && end != qp->q.len)) 318 goto discard_qp; 319 qp->q.flags |= INET_FRAG_LAST_IN; 320 qp->q.len = end; 321 } else { 322 if (end&7) { 323 end &= ~7; 324 if (skb->ip_summed != CHECKSUM_UNNECESSARY) 325 skb->ip_summed = CHECKSUM_NONE; 326 } 327 if (end > qp->q.len) { 328 /* Some bits beyond end -> corruption. */ 329 if (qp->q.flags & INET_FRAG_LAST_IN) 330 goto discard_qp; 331 qp->q.len = end; 332 } 333 } 334 if (end == offset) 335 goto discard_qp; 336 337 err = -ENOMEM; 338 if (!pskb_pull(skb, skb_network_offset(skb) + ihl)) 339 goto discard_qp; 340 341 err = pskb_trim_rcsum(skb, end - offset); 342 if (err) 343 goto discard_qp; 344 345 /* Note : skb->rbnode and skb->dev share the same location. */ 346 dev = skb->dev; 347 /* Makes sure compiler wont do silly aliasing games */ 348 barrier(); 349 350 prev_tail = qp->q.fragments_tail; 351 err = inet_frag_queue_insert(&qp->q, skb, offset, end); 352 if (err) 353 goto insert_error; 354 355 if (dev) 356 qp->iif = dev->ifindex; 357 358 qp->q.stamp = skb->tstamp; 359 qp->q.tstamp_type = skb->tstamp_type; 360 qp->q.meat += skb->len; 361 qp->ecn |= ecn; 362 add_frag_mem_limit(qp->q.fqdir, skb->truesize); 363 if (offset == 0) 364 qp->q.flags |= INET_FRAG_FIRST_IN; 365 366 fragsize = skb->len + ihl; 367 368 if (fragsize > qp->q.max_size) 369 qp->q.max_size = fragsize; 370 371 if (ip_hdr(skb)->frag_off & htons(IP_DF) && 372 fragsize > qp->max_df_size) 373 qp->max_df_size = fragsize; 374 375 if (qp->q.flags == (INET_FRAG_FIRST_IN | INET_FRAG_LAST_IN) && 376 qp->q.meat == qp->q.len) { 377 unsigned long orefdst = skb->_skb_refdst; 378 379 skb->_skb_refdst = 0UL; 380 err = ip_frag_reasm(qp, skb, prev_tail, dev); 381 skb->_skb_refdst = orefdst; 382 if (err) 383 inet_frag_kill(&qp->q); 384 return err; 385 } 386 387 skb_dst_drop(skb); 388 skb_orphan(skb); 389 return -EINPROGRESS; 390 391 insert_error: 392 if (err == IPFRAG_DUP) { 393 SKB_DR_SET(reason, DUP_FRAG); 394 err = -EINVAL; 395 goto err; 396 } 397 err = -EINVAL; 398 __IP_INC_STATS(net, IPSTATS_MIB_REASM_OVERLAPS); 399 discard_qp: 400 inet_frag_kill(&qp->q); 401 __IP_INC_STATS(net, IPSTATS_MIB_REASMFAILS); 402 err: 403 kfree_skb_reason(skb, reason); 404 return err; 405 } 406 407 static bool ip_frag_coalesce_ok(const struct ipq *qp) 408 { 409 return qp->q.key.v4.user == IP_DEFRAG_LOCAL_DELIVER; 410 } 411 412 /* Build a new IP datagram from all its fragments. */ 413 static int ip_frag_reasm(struct ipq *qp, struct sk_buff *skb, 414 struct sk_buff *prev_tail, struct net_device *dev) 415 { 416 struct net *net = qp->q.fqdir->net; 417 struct iphdr *iph; 418 void *reasm_data; 419 int len, err; 420 u8 ecn; 421 422 ipq_kill(qp); 423 424 ecn = ip_frag_ecn_table[qp->ecn]; 425 if (unlikely(ecn == 0xff)) { 426 err = -EINVAL; 427 goto out_fail; 428 } 429 430 /* Make the one we just received the head. */ 431 reasm_data = inet_frag_reasm_prepare(&qp->q, skb, prev_tail); 432 if (!reasm_data) 433 goto out_nomem; 434 435 len = ip_hdrlen(skb) + qp->q.len; 436 err = -E2BIG; 437 if (len > 65535) 438 goto out_oversize; 439 440 inet_frag_reasm_finish(&qp->q, skb, reasm_data, 441 ip_frag_coalesce_ok(qp)); 442 443 skb->dev = dev; 444 IPCB(skb)->frag_max_size = max(qp->max_df_size, qp->q.max_size); 445 446 iph = ip_hdr(skb); 447 iph->tot_len = htons(len); 448 iph->tos |= ecn; 449 450 /* When we set IP_DF on a refragmented skb we must also force a 451 * call to ip_fragment to avoid forwarding a DF-skb of size s while 452 * original sender only sent fragments of size f (where f < s). 453 * 454 * We only set DF/IPSKB_FRAG_PMTU if such DF fragment was the largest 455 * frag seen to avoid sending tiny DF-fragments in case skb was built 456 * from one very small df-fragment and one large non-df frag. 457 */ 458 if (qp->max_df_size == qp->q.max_size) { 459 IPCB(skb)->flags |= IPSKB_FRAG_PMTU; 460 iph->frag_off = htons(IP_DF); 461 } else { 462 iph->frag_off = 0; 463 } 464 465 ip_send_check(iph); 466 467 __IP_INC_STATS(net, IPSTATS_MIB_REASMOKS); 468 qp->q.rb_fragments = RB_ROOT; 469 qp->q.fragments_tail = NULL; 470 qp->q.last_run_head = NULL; 471 return 0; 472 473 out_nomem: 474 net_dbg_ratelimited("queue_glue: no memory for gluing queue %p\n", qp); 475 err = -ENOMEM; 476 goto out_fail; 477 out_oversize: 478 net_info_ratelimited("Oversized IP packet from %pI4\n", &qp->q.key.v4.saddr); 479 out_fail: 480 __IP_INC_STATS(net, IPSTATS_MIB_REASMFAILS); 481 return err; 482 } 483 484 /* Process an incoming IP datagram fragment. */ 485 int ip_defrag(struct net *net, struct sk_buff *skb, u32 user) 486 { 487 struct net_device *dev = skb->dev ? : skb_dst(skb)->dev; 488 int vif = l3mdev_master_ifindex_rcu(dev); 489 struct ipq *qp; 490 491 __IP_INC_STATS(net, IPSTATS_MIB_REASMREQDS); 492 493 /* Lookup (or create) queue header */ 494 qp = ip_find(net, ip_hdr(skb), user, vif); 495 if (qp) { 496 int ret; 497 498 spin_lock(&qp->q.lock); 499 500 ret = ip_frag_queue(qp, skb); 501 502 spin_unlock(&qp->q.lock); 503 ipq_put(qp); 504 return ret; 505 } 506 507 __IP_INC_STATS(net, IPSTATS_MIB_REASMFAILS); 508 kfree_skb(skb); 509 return -ENOMEM; 510 } 511 EXPORT_SYMBOL(ip_defrag); 512 513 struct sk_buff *ip_check_defrag(struct net *net, struct sk_buff *skb, u32 user) 514 { 515 struct iphdr iph; 516 int netoff; 517 u32 len; 518 519 if (skb->protocol != htons(ETH_P_IP)) 520 return skb; 521 522 netoff = skb_network_offset(skb); 523 524 if (skb_copy_bits(skb, netoff, &iph, sizeof(iph)) < 0) 525 return skb; 526 527 if (iph.ihl < 5 || iph.version != 4) 528 return skb; 529 530 len = ntohs(iph.tot_len); 531 if (skb->len < netoff + len || len < (iph.ihl * 4)) 532 return skb; 533 534 if (ip_is_fragment(&iph)) { 535 skb = skb_share_check(skb, GFP_ATOMIC); 536 if (skb) { 537 if (!pskb_may_pull(skb, netoff + iph.ihl * 4)) { 538 kfree_skb(skb); 539 return NULL; 540 } 541 if (pskb_trim_rcsum(skb, netoff + len)) { 542 kfree_skb(skb); 543 return NULL; 544 } 545 memset(IPCB(skb), 0, sizeof(struct inet_skb_parm)); 546 if (ip_defrag(net, skb, user)) 547 return NULL; 548 skb_clear_hash(skb); 549 } 550 } 551 return skb; 552 } 553 EXPORT_SYMBOL(ip_check_defrag); 554 555 #ifdef CONFIG_SYSCTL 556 static int dist_min; 557 558 static struct ctl_table ip4_frags_ns_ctl_table[] = { 559 { 560 .procname = "ipfrag_high_thresh", 561 .maxlen = sizeof(unsigned long), 562 .mode = 0644, 563 .proc_handler = proc_doulongvec_minmax, 564 }, 565 { 566 .procname = "ipfrag_low_thresh", 567 .maxlen = sizeof(unsigned long), 568 .mode = 0644, 569 .proc_handler = proc_doulongvec_minmax, 570 }, 571 { 572 .procname = "ipfrag_time", 573 .maxlen = sizeof(int), 574 .mode = 0644, 575 .proc_handler = proc_dointvec_jiffies, 576 }, 577 { 578 .procname = "ipfrag_max_dist", 579 .maxlen = sizeof(int), 580 .mode = 0644, 581 .proc_handler = proc_dointvec_minmax, 582 .extra1 = &dist_min, 583 }, 584 }; 585 586 /* secret interval has been deprecated */ 587 static int ip4_frags_secret_interval_unused; 588 static struct ctl_table ip4_frags_ctl_table[] = { 589 { 590 .procname = "ipfrag_secret_interval", 591 .data = &ip4_frags_secret_interval_unused, 592 .maxlen = sizeof(int), 593 .mode = 0644, 594 .proc_handler = proc_dointvec_jiffies, 595 }, 596 }; 597 598 static int __net_init ip4_frags_ns_ctl_register(struct net *net) 599 { 600 struct ctl_table *table; 601 struct ctl_table_header *hdr; 602 603 table = ip4_frags_ns_ctl_table; 604 if (!net_eq(net, &init_net)) { 605 table = kmemdup(table, sizeof(ip4_frags_ns_ctl_table), GFP_KERNEL); 606 if (!table) 607 goto err_alloc; 608 609 } 610 table[0].data = &net->ipv4.fqdir->high_thresh; 611 table[0].extra1 = &net->ipv4.fqdir->low_thresh; 612 table[1].data = &net->ipv4.fqdir->low_thresh; 613 table[1].extra2 = &net->ipv4.fqdir->high_thresh; 614 table[2].data = &net->ipv4.fqdir->timeout; 615 table[3].data = &net->ipv4.fqdir->max_dist; 616 617 hdr = register_net_sysctl_sz(net, "net/ipv4", table, 618 ARRAY_SIZE(ip4_frags_ns_ctl_table)); 619 if (!hdr) 620 goto err_reg; 621 622 net->ipv4.frags_hdr = hdr; 623 return 0; 624 625 err_reg: 626 if (!net_eq(net, &init_net)) 627 kfree(table); 628 err_alloc: 629 return -ENOMEM; 630 } 631 632 static void __net_exit ip4_frags_ns_ctl_unregister(struct net *net) 633 { 634 const struct ctl_table *table; 635 636 table = net->ipv4.frags_hdr->ctl_table_arg; 637 unregister_net_sysctl_table(net->ipv4.frags_hdr); 638 kfree(table); 639 } 640 641 static void __init ip4_frags_ctl_register(void) 642 { 643 register_net_sysctl(&init_net, "net/ipv4", ip4_frags_ctl_table); 644 } 645 #else 646 static int ip4_frags_ns_ctl_register(struct net *net) 647 { 648 return 0; 649 } 650 651 static void ip4_frags_ns_ctl_unregister(struct net *net) 652 { 653 } 654 655 static void __init ip4_frags_ctl_register(void) 656 { 657 } 658 #endif 659 660 static int __net_init ipv4_frags_init_net(struct net *net) 661 { 662 int res; 663 664 res = fqdir_init(&net->ipv4.fqdir, &ip4_frags, net); 665 if (res < 0) 666 return res; 667 /* Fragment cache limits. 668 * 669 * The fragment memory accounting code, (tries to) account for 670 * the real memory usage, by measuring both the size of frag 671 * queue struct (inet_frag_queue (ipv4:ipq/ipv6:frag_queue)) 672 * and the SKB's truesize. 673 * 674 * A 64K fragment consumes 129736 bytes (44*2944)+200 675 * (1500 truesize == 2944, sizeof(struct ipq) == 200) 676 * 677 * We will commit 4MB at one time. Should we cross that limit 678 * we will prune down to 3MB, making room for approx 8 big 64K 679 * fragments 8x128k. 680 */ 681 net->ipv4.fqdir->high_thresh = 4 * 1024 * 1024; 682 net->ipv4.fqdir->low_thresh = 3 * 1024 * 1024; 683 /* 684 * Important NOTE! Fragment queue must be destroyed before MSL expires. 685 * RFC791 is wrong proposing to prolongate timer each fragment arrival 686 * by TTL. 687 */ 688 net->ipv4.fqdir->timeout = IP_FRAG_TIME; 689 690 net->ipv4.fqdir->max_dist = 64; 691 692 res = ip4_frags_ns_ctl_register(net); 693 if (res < 0) 694 fqdir_exit(net->ipv4.fqdir); 695 return res; 696 } 697 698 static void __net_exit ipv4_frags_pre_exit_net(struct net *net) 699 { 700 fqdir_pre_exit(net->ipv4.fqdir); 701 } 702 703 static void __net_exit ipv4_frags_exit_net(struct net *net) 704 { 705 ip4_frags_ns_ctl_unregister(net); 706 fqdir_exit(net->ipv4.fqdir); 707 } 708 709 static struct pernet_operations ip4_frags_ops = { 710 .init = ipv4_frags_init_net, 711 .pre_exit = ipv4_frags_pre_exit_net, 712 .exit = ipv4_frags_exit_net, 713 }; 714 715 716 static u32 ip4_key_hashfn(const void *data, u32 len, u32 seed) 717 { 718 return jhash2(data, 719 sizeof(struct frag_v4_compare_key) / sizeof(u32), seed); 720 } 721 722 static u32 ip4_obj_hashfn(const void *data, u32 len, u32 seed) 723 { 724 const struct inet_frag_queue *fq = data; 725 726 return jhash2((const u32 *)&fq->key.v4, 727 sizeof(struct frag_v4_compare_key) / sizeof(u32), seed); 728 } 729 730 static int ip4_obj_cmpfn(struct rhashtable_compare_arg *arg, const void *ptr) 731 { 732 const struct frag_v4_compare_key *key = arg->key; 733 const struct inet_frag_queue *fq = ptr; 734 735 return !!memcmp(&fq->key, key, sizeof(*key)); 736 } 737 738 static const struct rhashtable_params ip4_rhash_params = { 739 .head_offset = offsetof(struct inet_frag_queue, node), 740 .key_offset = offsetof(struct inet_frag_queue, key), 741 .key_len = sizeof(struct frag_v4_compare_key), 742 .hashfn = ip4_key_hashfn, 743 .obj_hashfn = ip4_obj_hashfn, 744 .obj_cmpfn = ip4_obj_cmpfn, 745 .automatic_shrinking = true, 746 }; 747 748 void __init ipfrag_init(void) 749 { 750 ip4_frags.constructor = ip4_frag_init; 751 ip4_frags.destructor = ip4_frag_free; 752 ip4_frags.qsize = sizeof(struct ipq); 753 ip4_frags.frag_expire = ip_expire; 754 ip4_frags.frags_cache_name = ip_frag_cache_name; 755 ip4_frags.rhash_params = ip4_rhash_params; 756 if (inet_frags_init(&ip4_frags)) 757 panic("IP: failed to allocate ip4_frags cache\n"); 758 ip4_frags_ctl_register(); 759 register_pernet_subsys(&ip4_frags_ops); 760 } 761