1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3 * 4 * Copyright (c) 2020 Alexander V. Chernikov 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/cdefs.h> 29 __FBSDID("$FreeBSD$"); 30 #include "opt_inet.h" 31 #include "opt_inet6.h" 32 #include "opt_route.h" 33 34 #include <sys/param.h> 35 #include <sys/systm.h> 36 #include <sys/malloc.h> 37 #include <sys/mbuf.h> 38 #include <sys/socket.h> 39 #include <sys/sysctl.h> 40 #include <sys/syslog.h> 41 #include <sys/kernel.h> 42 #include <sys/lock.h> 43 #include <sys/rmlock.h> 44 45 #include <net/if.h> 46 #include <net/if_var.h> 47 #include <net/if_dl.h> 48 #include <net/vnet.h> 49 #include <net/route.h> 50 #include <net/route/route_ctl.h> 51 #include <net/route/route_var.h> 52 #include <net/route/nhop_utils.h> 53 #include <net/route/nhop.h> 54 #include <net/route/nhop_var.h> 55 #include <netinet/in.h> 56 #include <netinet6/scope6_var.h> 57 58 #include <vm/uma.h> 59 60 /* 61 * This file contains control plane routing tables functions. 62 * 63 * All functions assumes they are called in net epoch. 64 */ 65 66 struct rib_subscription { 67 CK_STAILQ_ENTRY(rib_subscription) next; 68 rib_subscription_cb_t *func; 69 void *arg; 70 struct rib_head *rnh; 71 enum rib_subscription_type type; 72 struct epoch_context epoch_ctx; 73 }; 74 75 static int add_route(struct rib_head *rnh, struct rt_addrinfo *info, 76 struct rib_cmd_info *rc); 77 static int add_route_nhop(struct rib_head *rnh, struct rtentry *rt, 78 struct rt_addrinfo *info, struct route_nhop_data *rnd, 79 struct rib_cmd_info *rc); 80 static int del_route(struct rib_head *rnh, struct rt_addrinfo *info, 81 struct rib_cmd_info *rc); 82 static int change_route(struct rib_head *rnh, struct rt_addrinfo *info, 83 struct route_nhop_data *nhd_orig, struct rib_cmd_info *rc); 84 85 static int rt_unlinkrte(struct rib_head *rnh, struct rt_addrinfo *info, 86 struct rib_cmd_info *rc); 87 88 static void rib_notify(struct rib_head *rnh, enum rib_subscription_type type, 89 struct rib_cmd_info *rc); 90 91 static void destroy_subscription_epoch(epoch_context_t ctx); 92 #ifdef ROUTE_MPATH 93 static bool rib_can_multipath(struct rib_head *rh); 94 #endif 95 96 /* Per-vnet multipath routing configuration */ 97 SYSCTL_DECL(_net_route); 98 #define V_rib_route_multipath VNET(rib_route_multipath) 99 #ifdef ROUTE_MPATH 100 #define _MP_FLAGS CTLFLAG_RW 101 #else 102 #define _MP_FLAGS CTLFLAG_RD 103 #endif 104 VNET_DEFINE(u_int, rib_route_multipath) = 1; 105 SYSCTL_UINT(_net_route, OID_AUTO, multipath, _MP_FLAGS | CTLFLAG_VNET, 106 &VNET_NAME(rib_route_multipath), 0, "Enable route multipath"); 107 #undef _MP_FLAGS 108 109 #if defined(INET) && defined(INET6) 110 FEATURE(ipv4_rfc5549_support, "Route IPv4 packets via IPv6 nexthops"); 111 #define V_rib_route_ipv6_nexthop VNET(rib_route_ipv6_nexthop) 112 VNET_DEFINE(u_int, rib_route_ipv6_nexthop) = 1; 113 SYSCTL_UINT(_net_route, OID_AUTO, ipv6_nexthop, CTLFLAG_RW | CTLFLAG_VNET, 114 &VNET_NAME(rib_route_ipv6_nexthop), 0, "Enable IPv4 route via IPv6 Next Hop address"); 115 #endif 116 117 /* Routing table UMA zone */ 118 VNET_DEFINE_STATIC(uma_zone_t, rtzone); 119 #define V_rtzone VNET(rtzone) 120 121 void 122 vnet_rtzone_init() 123 { 124 125 V_rtzone = uma_zcreate("rtentry", sizeof(struct rtentry), 126 NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0); 127 } 128 129 #ifdef VIMAGE 130 void 131 vnet_rtzone_destroy() 132 { 133 134 uma_zdestroy(V_rtzone); 135 } 136 #endif 137 138 static void 139 destroy_rtentry(struct rtentry *rt) 140 { 141 #ifdef VIMAGE 142 struct nhop_object *nh = rt->rt_nhop; 143 144 /* 145 * At this moment rnh, nh_control may be already freed. 146 * nhop interface may have been migrated to a different vnet. 147 * Use vnet stored in the nexthop to delete the entry. 148 */ 149 #ifdef ROUTE_MPATH 150 if (NH_IS_NHGRP(nh)) { 151 struct weightened_nhop *wn; 152 uint32_t num_nhops; 153 wn = nhgrp_get_nhops((struct nhgrp_object *)nh, &num_nhops); 154 nh = wn[0].nh; 155 } 156 #endif 157 CURVNET_SET(nhop_get_vnet(nh)); 158 #endif 159 160 /* Unreference nexthop */ 161 nhop_free_any(rt->rt_nhop); 162 163 uma_zfree(V_rtzone, rt); 164 165 CURVNET_RESTORE(); 166 } 167 168 /* 169 * Epoch callback indicating rtentry is safe to destroy 170 */ 171 static void 172 destroy_rtentry_epoch(epoch_context_t ctx) 173 { 174 struct rtentry *rt; 175 176 rt = __containerof(ctx, struct rtentry, rt_epoch_ctx); 177 178 destroy_rtentry(rt); 179 } 180 181 /* 182 * Schedule rtentry deletion 183 */ 184 static void 185 rtfree(struct rtentry *rt) 186 { 187 188 KASSERT(rt != NULL, ("%s: NULL rt", __func__)); 189 190 epoch_call(net_epoch_preempt, destroy_rtentry_epoch, 191 &rt->rt_epoch_ctx); 192 } 193 194 static struct rib_head * 195 get_rnh(uint32_t fibnum, const struct rt_addrinfo *info) 196 { 197 struct rib_head *rnh; 198 struct sockaddr *dst; 199 200 KASSERT((fibnum < rt_numfibs), ("rib_add_route: bad fibnum")); 201 202 dst = info->rti_info[RTAX_DST]; 203 rnh = rt_tables_get_rnh(fibnum, dst->sa_family); 204 205 return (rnh); 206 } 207 208 #if defined(INET) && defined(INET6) 209 static bool 210 rib_can_ipv6_nexthop_address(struct rib_head *rh) 211 { 212 int result; 213 214 CURVNET_SET(rh->rib_vnet); 215 result = !!V_rib_route_ipv6_nexthop; 216 CURVNET_RESTORE(); 217 218 return (result); 219 } 220 #endif 221 222 #ifdef ROUTE_MPATH 223 static bool 224 rib_can_multipath(struct rib_head *rh) 225 { 226 int result; 227 228 CURVNET_SET(rh->rib_vnet); 229 result = !!V_rib_route_multipath; 230 CURVNET_RESTORE(); 231 232 return (result); 233 } 234 235 /* 236 * Check is nhop is multipath-eligible. 237 * Avoid nhops without gateways and redirects. 238 * 239 * Returns 1 for multipath-eligible nexthop, 240 * 0 otherwise. 241 */ 242 bool 243 nhop_can_multipath(const struct nhop_object *nh) 244 { 245 246 if ((nh->nh_flags & NHF_MULTIPATH) != 0) 247 return (1); 248 if ((nh->nh_flags & NHF_GATEWAY) == 0) 249 return (0); 250 if ((nh->nh_flags & NHF_REDIRECT) != 0) 251 return (0); 252 253 return (1); 254 } 255 #endif 256 257 static int 258 get_info_weight(const struct rt_addrinfo *info, uint32_t default_weight) 259 { 260 uint32_t weight; 261 262 if (info->rti_mflags & RTV_WEIGHT) 263 weight = info->rti_rmx->rmx_weight; 264 else 265 weight = default_weight; 266 /* Keep upper 1 byte for adm distance purposes */ 267 if (weight > RT_MAX_WEIGHT) 268 weight = RT_MAX_WEIGHT; 269 else if (weight == 0) 270 weight = default_weight; 271 272 return (weight); 273 } 274 275 bool 276 rt_is_host(const struct rtentry *rt) 277 { 278 279 return (rt->rte_flags & RTF_HOST); 280 } 281 282 sa_family_t 283 rt_get_family(const struct rtentry *rt) 284 { 285 const struct sockaddr *dst; 286 287 dst = (const struct sockaddr *)rt_key_const(rt); 288 289 return (dst->sa_family); 290 } 291 292 /* 293 * Returns pointer to nexthop or nexthop group 294 * associated with @rt 295 */ 296 struct nhop_object * 297 rt_get_raw_nhop(const struct rtentry *rt) 298 { 299 300 return (rt->rt_nhop); 301 } 302 303 #ifdef INET 304 /* 305 * Stores IPv4 address and prefix length of @rt inside 306 * @paddr and @plen. 307 * @pscopeid is currently always set to 0. 308 */ 309 void 310 rt_get_inet_prefix_plen(const struct rtentry *rt, struct in_addr *paddr, 311 int *plen, uint32_t *pscopeid) 312 { 313 const struct sockaddr_in *dst; 314 315 dst = (const struct sockaddr_in *)rt_key_const(rt); 316 KASSERT((dst->sin_family == AF_INET), 317 ("rt family is %d, not inet", dst->sin_family)); 318 *paddr = dst->sin_addr; 319 dst = (const struct sockaddr_in *)rt_mask_const(rt); 320 if (dst == NULL) 321 *plen = 32; 322 else 323 *plen = bitcount32(dst->sin_addr.s_addr); 324 *pscopeid = 0; 325 } 326 327 /* 328 * Stores IPv4 address and prefix mask of @rt inside 329 * @paddr and @pmask. Sets mask to INADDR_ANY for host routes. 330 * @pscopeid is currently always set to 0. 331 */ 332 void 333 rt_get_inet_prefix_pmask(const struct rtentry *rt, struct in_addr *paddr, 334 struct in_addr *pmask, uint32_t *pscopeid) 335 { 336 const struct sockaddr_in *dst; 337 338 dst = (const struct sockaddr_in *)rt_key_const(rt); 339 KASSERT((dst->sin_family == AF_INET), 340 ("rt family is %d, not inet", dst->sin_family)); 341 *paddr = dst->sin_addr; 342 dst = (const struct sockaddr_in *)rt_mask_const(rt); 343 if (dst == NULL) 344 pmask->s_addr = INADDR_BROADCAST; 345 else 346 *pmask = dst->sin_addr; 347 *pscopeid = 0; 348 } 349 #endif 350 351 #ifdef INET6 352 static int 353 inet6_get_plen(const struct in6_addr *addr) 354 { 355 356 return (bitcount32(addr->s6_addr32[0]) + bitcount32(addr->s6_addr32[1]) + 357 bitcount32(addr->s6_addr32[2]) + bitcount32(addr->s6_addr32[3])); 358 } 359 360 /* 361 * Stores IPv6 address and prefix length of @rt inside 362 * @paddr and @plen. Addresses are returned in de-embedded form. 363 * Scopeid is set to 0 for non-LL addresses. 364 */ 365 void 366 rt_get_inet6_prefix_plen(const struct rtentry *rt, struct in6_addr *paddr, 367 int *plen, uint32_t *pscopeid) 368 { 369 const struct sockaddr_in6 *dst; 370 371 dst = (const struct sockaddr_in6 *)rt_key_const(rt); 372 KASSERT((dst->sin6_family == AF_INET6), 373 ("rt family is %d, not inet6", dst->sin6_family)); 374 if (IN6_IS_SCOPE_LINKLOCAL(&dst->sin6_addr)) 375 in6_splitscope(&dst->sin6_addr, paddr, pscopeid); 376 else 377 *paddr = dst->sin6_addr; 378 dst = (const struct sockaddr_in6 *)rt_mask_const(rt); 379 if (dst == NULL) 380 *plen = 128; 381 else 382 *plen = inet6_get_plen(&dst->sin6_addr); 383 } 384 385 /* 386 * Stores IPv6 address and prefix mask of @rt inside 387 * @paddr and @pmask. Addresses are returned in de-embedded form. 388 * Scopeid is set to 0 for non-LL addresses. 389 */ 390 void 391 rt_get_inet6_prefix_pmask(const struct rtentry *rt, struct in6_addr *paddr, 392 struct in6_addr *pmask, uint32_t *pscopeid) 393 { 394 const struct sockaddr_in6 *dst; 395 396 dst = (const struct sockaddr_in6 *)rt_key_const(rt); 397 KASSERT((dst->sin6_family == AF_INET6), 398 ("rt family is %d, not inet", dst->sin6_family)); 399 if (IN6_IS_SCOPE_LINKLOCAL(&dst->sin6_addr)) 400 in6_splitscope(&dst->sin6_addr, paddr, pscopeid); 401 else 402 *paddr = dst->sin6_addr; 403 dst = (const struct sockaddr_in6 *)rt_mask_const(rt); 404 if (dst == NULL) 405 memset(pmask, 0xFF, sizeof(struct in6_addr)); 406 else 407 *pmask = dst->sin6_addr; 408 } 409 #endif 410 411 static void 412 rt_set_expire_info(struct rtentry *rt, const struct rt_addrinfo *info) 413 { 414 415 /* Kernel -> userland timebase conversion. */ 416 if (info->rti_mflags & RTV_EXPIRE) 417 rt->rt_expire = info->rti_rmx->rmx_expire ? 418 info->rti_rmx->rmx_expire - time_second + time_uptime : 0; 419 } 420 421 /* 422 * Check if specified @gw matches gw data in the nexthop @nh. 423 * 424 * Returns true if matches, false otherwise. 425 */ 426 bool 427 match_nhop_gw(const struct nhop_object *nh, const struct sockaddr *gw) 428 { 429 430 if (nh->gw_sa.sa_family != gw->sa_family) 431 return (false); 432 433 switch (gw->sa_family) { 434 case AF_INET: 435 return (nh->gw4_sa.sin_addr.s_addr == 436 ((const struct sockaddr_in *)gw)->sin_addr.s_addr); 437 case AF_INET6: 438 { 439 const struct sockaddr_in6 *gw6; 440 gw6 = (const struct sockaddr_in6 *)gw; 441 442 /* 443 * Currently (2020-09) IPv6 gws in kernel have their 444 * scope embedded. Once this becomes false, this code 445 * has to be revisited. 446 */ 447 if (IN6_ARE_ADDR_EQUAL(&nh->gw6_sa.sin6_addr, 448 &gw6->sin6_addr)) 449 return (true); 450 return (false); 451 } 452 case AF_LINK: 453 { 454 const struct sockaddr_dl *sdl; 455 sdl = (const struct sockaddr_dl *)gw; 456 return (nh->gwl_sa.sdl_index == sdl->sdl_index); 457 } 458 default: 459 return (memcmp(&nh->gw_sa, gw, nh->gw_sa.sa_len) == 0); 460 } 461 462 /* NOTREACHED */ 463 return (false); 464 } 465 466 /* 467 * Checks if data in @info matches nexhop @nh. 468 * 469 * Returns 0 on success, 470 * ESRCH if not matched, 471 * ENOENT if filter function returned false 472 */ 473 int 474 check_info_match_nhop(const struct rt_addrinfo *info, const struct rtentry *rt, 475 const struct nhop_object *nh) 476 { 477 const struct sockaddr *gw = info->rti_info[RTAX_GATEWAY]; 478 479 if (info->rti_filter != NULL) { 480 if (info->rti_filter(rt, nh, info->rti_filterdata) == 0) 481 return (ENOENT); 482 else 483 return (0); 484 } 485 if ((gw != NULL) && !match_nhop_gw(nh, gw)) 486 return (ESRCH); 487 488 return (0); 489 } 490 491 /* 492 * Checks if nexhop @nh can be rewritten by data in @info because 493 * of higher "priority". Currently the only case for such scenario 494 * is kernel installing interface routes, marked by RTF_PINNED flag. 495 * 496 * Returns: 497 * 1 if @info data has higher priority 498 * 0 if priority is the same 499 * -1 if priority is lower 500 */ 501 int 502 can_override_nhop(const struct rt_addrinfo *info, const struct nhop_object *nh) 503 { 504 505 if (info->rti_flags & RTF_PINNED) { 506 return (NH_IS_PINNED(nh)) ? 0 : 1; 507 } else { 508 return (NH_IS_PINNED(nh)) ? -1 : 0; 509 } 510 } 511 512 /* 513 * Runs exact prefix match based on @dst and @netmask. 514 * Returns matched @rtentry if found or NULL. 515 * If rtentry was found, saves nexthop / weight value into @rnd. 516 */ 517 static struct rtentry * 518 lookup_prefix_bysa(struct rib_head *rnh, const struct sockaddr *dst, 519 const struct sockaddr *netmask, struct route_nhop_data *rnd) 520 { 521 struct rtentry *rt; 522 523 RIB_LOCK_ASSERT(rnh); 524 525 rt = (struct rtentry *)rnh->rnh_lookup(__DECONST(void *, dst), 526 __DECONST(void *, netmask), &rnh->head); 527 if (rt != NULL) { 528 rnd->rnd_nhop = rt->rt_nhop; 529 rnd->rnd_weight = rt->rt_weight; 530 } else { 531 rnd->rnd_nhop = NULL; 532 rnd->rnd_weight = 0; 533 } 534 535 return (rt); 536 } 537 538 /* 539 * Runs exact prefix match based on dst/netmask from @info. 540 * Assumes RIB lock is held. 541 * Returns matched @rtentry if found or NULL. 542 * If rtentry was found, saves nexthop / weight value into @rnd. 543 */ 544 struct rtentry * 545 lookup_prefix(struct rib_head *rnh, const struct rt_addrinfo *info, 546 struct route_nhop_data *rnd) 547 { 548 struct rtentry *rt; 549 550 rt = lookup_prefix_bysa(rnh, info->rti_info[RTAX_DST], 551 info->rti_info[RTAX_NETMASK], rnd); 552 553 return (rt); 554 } 555 556 /* 557 * Adds route defined by @info into the kernel table specified by @fibnum and 558 * sa_family in @info->rti_info[RTAX_DST]. 559 * 560 * Returns 0 on success and fills in operation metadata into @rc. 561 */ 562 int 563 rib_add_route(uint32_t fibnum, struct rt_addrinfo *info, 564 struct rib_cmd_info *rc) 565 { 566 struct rib_head *rnh; 567 int error; 568 569 NET_EPOCH_ASSERT(); 570 571 rnh = get_rnh(fibnum, info); 572 if (rnh == NULL) 573 return (EAFNOSUPPORT); 574 575 /* 576 * Check consistency between RTF_HOST flag and netmask 577 * existence. 578 */ 579 if (info->rti_flags & RTF_HOST) 580 info->rti_info[RTAX_NETMASK] = NULL; 581 else if (info->rti_info[RTAX_NETMASK] == NULL) 582 return (EINVAL); 583 584 bzero(rc, sizeof(struct rib_cmd_info)); 585 rc->rc_cmd = RTM_ADD; 586 587 error = add_route(rnh, info, rc); 588 if (error == 0) 589 rib_notify(rnh, RIB_NOTIFY_DELAYED, rc); 590 591 return (error); 592 } 593 594 /* 595 * Checks if @dst and @gateway is valid combination. 596 * 597 * Returns true if is valid, false otherwise. 598 */ 599 static bool 600 check_gateway(struct rib_head *rnh, struct sockaddr *dst, 601 struct sockaddr *gateway) 602 { 603 if (dst->sa_family == gateway->sa_family) 604 return (true); 605 else if (gateway->sa_family == AF_UNSPEC) 606 return (true); 607 else if (gateway->sa_family == AF_LINK) 608 return (true); 609 #if defined(INET) && defined(INET6) 610 else if (dst->sa_family == AF_INET && gateway->sa_family == AF_INET6 && 611 rib_can_ipv6_nexthop_address(rnh)) 612 return (true); 613 #endif 614 else 615 return (false); 616 } 617 618 /* 619 * Creates rtentry and nexthop based on @info data. 620 * Return 0 and fills in rtentry into @prt on success, 621 * return errno otherwise. 622 */ 623 static int 624 create_rtentry(struct rib_head *rnh, struct rt_addrinfo *info, 625 struct rtentry **prt) 626 { 627 struct sockaddr *dst, *ndst, *gateway, *netmask; 628 struct rtentry *rt; 629 struct nhop_object *nh; 630 struct ifaddr *ifa; 631 int error, flags; 632 633 dst = info->rti_info[RTAX_DST]; 634 gateway = info->rti_info[RTAX_GATEWAY]; 635 netmask = info->rti_info[RTAX_NETMASK]; 636 flags = info->rti_flags; 637 638 if ((flags & RTF_GATEWAY) && !gateway) 639 return (EINVAL); 640 if (dst && gateway && !check_gateway(rnh, dst, gateway)) 641 return (EINVAL); 642 643 if (dst->sa_len > sizeof(((struct rtentry *)NULL)->rt_dstb)) 644 return (EINVAL); 645 646 if (info->rti_ifa == NULL) { 647 error = rt_getifa_fib(info, rnh->rib_fibnum); 648 if (error) 649 return (error); 650 } 651 652 error = nhop_create_from_info(rnh, info, &nh); 653 if (error != 0) 654 return (error); 655 656 rt = uma_zalloc(V_rtzone, M_NOWAIT | M_ZERO); 657 if (rt == NULL) { 658 nhop_free(nh); 659 return (ENOBUFS); 660 } 661 rt->rte_flags = (RTF_UP | flags) & RTE_RT_FLAG_MASK; 662 rt->rt_nhop = nh; 663 664 /* Fill in dst */ 665 memcpy(&rt->rt_dst, dst, dst->sa_len); 666 rt_key(rt) = &rt->rt_dst; 667 668 /* 669 * point to the (possibly newly malloc'd) dest address. 670 */ 671 ndst = (struct sockaddr *)rt_key(rt); 672 673 /* 674 * make sure it contains the value we want (masked if needed). 675 */ 676 if (netmask) { 677 rt_maskedcopy(dst, ndst, netmask); 678 } else 679 bcopy(dst, ndst, dst->sa_len); 680 681 /* 682 * We use the ifa reference returned by rt_getifa_fib(). 683 * This moved from below so that rnh->rnh_addaddr() can 684 * examine the ifa and ifa->ifa_ifp if it so desires. 685 */ 686 ifa = info->rti_ifa; 687 rt->rt_weight = get_info_weight(info, RT_DEFAULT_WEIGHT); 688 rt_set_expire_info(rt, info); 689 690 *prt = rt; 691 return (0); 692 } 693 694 static int 695 add_route(struct rib_head *rnh, struct rt_addrinfo *info, 696 struct rib_cmd_info *rc) 697 { 698 struct nhop_object *nh_orig; 699 struct route_nhop_data rnd_orig, rnd_add; 700 struct nhop_object *nh; 701 struct rtentry *rt, *rt_orig; 702 int error; 703 704 error = create_rtentry(rnh, info, &rt); 705 if (error != 0) 706 return (error); 707 708 rnd_add.rnd_nhop = rt->rt_nhop; 709 rnd_add.rnd_weight = rt->rt_weight; 710 nh = rt->rt_nhop; 711 712 RIB_WLOCK(rnh); 713 error = add_route_nhop(rnh, rt, info, &rnd_add, rc); 714 if (error == 0) { 715 RIB_WUNLOCK(rnh); 716 return (0); 717 } 718 719 /* addition failed. Lookup prefix in the rib to determine the cause */ 720 rt_orig = lookup_prefix(rnh, info, &rnd_orig); 721 if (rt_orig == NULL) { 722 /* No prefix -> rnh_addaddr() failed to allocate memory */ 723 RIB_WUNLOCK(rnh); 724 nhop_free(nh); 725 uma_zfree(V_rtzone, rt); 726 return (ENOMEM); 727 } 728 729 /* We have existing route in the RIB. */ 730 nh_orig = rnd_orig.rnd_nhop; 731 /* Check if new route has higher preference */ 732 if (can_override_nhop(info, nh_orig) > 0) { 733 /* Update nexthop to the new route */ 734 change_route_nhop(rnh, rt_orig, info, &rnd_add, rc); 735 RIB_WUNLOCK(rnh); 736 uma_zfree(V_rtzone, rt); 737 nhop_free(nh_orig); 738 return (0); 739 } 740 741 RIB_WUNLOCK(rnh); 742 743 #ifdef ROUTE_MPATH 744 if (rib_can_multipath(rnh) && nhop_can_multipath(rnd_add.rnd_nhop) && 745 nhop_can_multipath(rnd_orig.rnd_nhop)) 746 error = add_route_mpath(rnh, info, rt, &rnd_add, &rnd_orig, rc); 747 else 748 #endif 749 /* Unable to add - another route with the same preference exists */ 750 error = EEXIST; 751 752 /* 753 * ROUTE_MPATH disabled: failed to add route, free both nhop and rt. 754 * ROUTE_MPATH enabled: original nhop reference is unused in any case, 755 * free rt only if not _adding_ new route to rib (e.g. the case 756 * when initial lookup returned existing route, but then it got 757 * deleted prior to multipath group insertion, leading to a simple 758 * non-multipath add as a result). 759 */ 760 nhop_free(nh); 761 if ((error != 0) || rc->rc_cmd != RTM_ADD) 762 uma_zfree(V_rtzone, rt); 763 764 return (error); 765 } 766 767 /* 768 * Removes route defined by @info from the kernel table specified by @fibnum and 769 * sa_family in @info->rti_info[RTAX_DST]. 770 * 771 * Returns 0 on success and fills in operation metadata into @rc. 772 */ 773 int 774 rib_del_route(uint32_t fibnum, struct rt_addrinfo *info, struct rib_cmd_info *rc) 775 { 776 struct rib_head *rnh; 777 struct sockaddr *dst_orig, *netmask; 778 struct sockaddr_storage mdst; 779 int error; 780 781 NET_EPOCH_ASSERT(); 782 783 rnh = get_rnh(fibnum, info); 784 if (rnh == NULL) 785 return (EAFNOSUPPORT); 786 787 bzero(rc, sizeof(struct rib_cmd_info)); 788 rc->rc_cmd = RTM_DELETE; 789 790 dst_orig = info->rti_info[RTAX_DST]; 791 netmask = info->rti_info[RTAX_NETMASK]; 792 793 if (netmask != NULL) { 794 /* Ensure @dst is always properly masked */ 795 if (dst_orig->sa_len > sizeof(mdst)) 796 return (EINVAL); 797 rt_maskedcopy(dst_orig, (struct sockaddr *)&mdst, netmask); 798 info->rti_info[RTAX_DST] = (struct sockaddr *)&mdst; 799 } 800 error = del_route(rnh, info, rc); 801 info->rti_info[RTAX_DST] = dst_orig; 802 803 return (error); 804 } 805 806 /* 807 * Conditionally unlinks rtentry matching data inside @info from @rnh. 808 * Returns 0 on success with operation result stored in @rc. 809 * On error, returns: 810 * ESRCH - if prefix was not found, 811 * EADDRINUSE - if trying to delete higher priority route. 812 * ENOENT - if supplied filter function returned 0 (not matched). 813 */ 814 static int 815 rt_unlinkrte(struct rib_head *rnh, struct rt_addrinfo *info, struct rib_cmd_info *rc) 816 { 817 struct rtentry *rt; 818 struct nhop_object *nh; 819 struct radix_node *rn; 820 struct route_nhop_data rnd; 821 int error; 822 823 rt = lookup_prefix(rnh, info, &rnd); 824 if (rt == NULL) 825 return (ESRCH); 826 827 nh = rt->rt_nhop; 828 #ifdef ROUTE_MPATH 829 if (NH_IS_NHGRP(nh)) { 830 error = del_route_mpath(rnh, info, rt, 831 (struct nhgrp_object *)nh, rc); 832 return (error); 833 } 834 #endif 835 error = check_info_match_nhop(info, rt, nh); 836 if (error != 0) 837 return (error); 838 839 if (can_override_nhop(info, nh) < 0) 840 return (EADDRINUSE); 841 842 /* 843 * Remove the item from the tree and return it. 844 * Complain if it is not there and do no more processing. 845 */ 846 rn = rnh->rnh_deladdr(info->rti_info[RTAX_DST], 847 info->rti_info[RTAX_NETMASK], &rnh->head); 848 if (rn == NULL) 849 return (ESRCH); 850 851 if (rn->rn_flags & (RNF_ACTIVE | RNF_ROOT)) 852 panic ("rtrequest delete"); 853 854 rt = RNTORT(rn); 855 rt->rte_flags &= ~RTF_UP; 856 857 /* Finalize notification */ 858 rib_bump_gen(rnh); 859 rnh->rnh_prefixes--; 860 861 rc->rc_cmd = RTM_DELETE; 862 rc->rc_rt = rt; 863 rc->rc_nh_old = rt->rt_nhop; 864 rc->rc_nh_weight = rt->rt_weight; 865 rib_notify(rnh, RIB_NOTIFY_IMMEDIATE, rc); 866 867 return (0); 868 } 869 870 static int 871 del_route(struct rib_head *rnh, struct rt_addrinfo *info, 872 struct rib_cmd_info *rc) 873 { 874 int error; 875 876 RIB_WLOCK(rnh); 877 error = rt_unlinkrte(rnh, info, rc); 878 RIB_WUNLOCK(rnh); 879 if (error != 0) 880 return (error); 881 882 rib_notify(rnh, RIB_NOTIFY_DELAYED, rc); 883 884 /* 885 * If the caller wants it, then it can have it, 886 * the entry will be deleted after the end of the current epoch. 887 */ 888 if (rc->rc_cmd == RTM_DELETE) 889 rtfree(rc->rc_rt); 890 #ifdef ROUTE_MPATH 891 else { 892 /* 893 * Deleting 1 path may result in RTM_CHANGE to 894 * a different mpath group/nhop. 895 * Free old mpath group. 896 */ 897 nhop_free_any(rc->rc_nh_old); 898 } 899 #endif 900 901 return (0); 902 } 903 904 int 905 rib_change_route(uint32_t fibnum, struct rt_addrinfo *info, 906 struct rib_cmd_info *rc) 907 { 908 RIB_RLOCK_TRACKER; 909 struct route_nhop_data rnd_orig; 910 struct rib_head *rnh; 911 struct rtentry *rt; 912 int error; 913 914 NET_EPOCH_ASSERT(); 915 916 rnh = get_rnh(fibnum, info); 917 if (rnh == NULL) 918 return (EAFNOSUPPORT); 919 920 bzero(rc, sizeof(struct rib_cmd_info)); 921 rc->rc_cmd = RTM_CHANGE; 922 923 /* Check if updated gateway exists */ 924 if ((info->rti_flags & RTF_GATEWAY) && 925 (info->rti_info[RTAX_GATEWAY] == NULL)) { 926 927 /* 928 * route(8) adds RTF_GATEWAY flag if -interface is not set. 929 * Remove RTF_GATEWAY to enforce consistency and maintain 930 * compatibility.. 931 */ 932 info->rti_flags &= ~RTF_GATEWAY; 933 } 934 935 /* 936 * route change is done in multiple steps, with dropping and 937 * reacquiring lock. In the situations with multiple processes 938 * changes the same route in can lead to the case when route 939 * is changed between the steps. Address it by retrying the operation 940 * multiple times before failing. 941 */ 942 943 RIB_RLOCK(rnh); 944 rt = (struct rtentry *)rnh->rnh_lookup(info->rti_info[RTAX_DST], 945 info->rti_info[RTAX_NETMASK], &rnh->head); 946 947 if (rt == NULL) { 948 RIB_RUNLOCK(rnh); 949 return (ESRCH); 950 } 951 952 rnd_orig.rnd_nhop = rt->rt_nhop; 953 rnd_orig.rnd_weight = rt->rt_weight; 954 955 RIB_RUNLOCK(rnh); 956 957 for (int i = 0; i < RIB_MAX_RETRIES; i++) { 958 error = change_route(rnh, info, &rnd_orig, rc); 959 if (error != EAGAIN) 960 break; 961 } 962 963 return (error); 964 } 965 966 static int 967 change_nhop(struct rib_head *rnh, struct rt_addrinfo *info, 968 struct nhop_object *nh_orig, struct nhop_object **nh_new) 969 { 970 int error; 971 972 /* 973 * New gateway could require new ifaddr, ifp; 974 * flags may also be different; ifp may be specified 975 * by ll sockaddr when protocol address is ambiguous 976 */ 977 if (((nh_orig->nh_flags & NHF_GATEWAY) && 978 info->rti_info[RTAX_GATEWAY] != NULL) || 979 info->rti_info[RTAX_IFP] != NULL || 980 (info->rti_info[RTAX_IFA] != NULL && 981 !sa_equal(info->rti_info[RTAX_IFA], nh_orig->nh_ifa->ifa_addr))) { 982 error = rt_getifa_fib(info, rnh->rib_fibnum); 983 984 if (error != 0) { 985 info->rti_ifa = NULL; 986 return (error); 987 } 988 } 989 990 error = nhop_create_from_nhop(rnh, nh_orig, info, nh_new); 991 info->rti_ifa = NULL; 992 993 return (error); 994 } 995 996 #ifdef ROUTE_MPATH 997 static int 998 change_mpath_route(struct rib_head *rnh, struct rt_addrinfo *info, 999 struct route_nhop_data *rnd_orig, struct rib_cmd_info *rc) 1000 { 1001 int error = 0; 1002 struct nhop_object *nh, *nh_orig, *nh_new; 1003 struct route_nhop_data rnd_new; 1004 1005 nh = NULL; 1006 nh_orig = rnd_orig->rnd_nhop; 1007 1008 struct weightened_nhop *wn = NULL, *wn_new; 1009 uint32_t num_nhops; 1010 1011 wn = nhgrp_get_nhops((struct nhgrp_object *)nh_orig, &num_nhops); 1012 nh_orig = NULL; 1013 for (int i = 0; i < num_nhops; i++) { 1014 if (check_info_match_nhop(info, NULL, wn[i].nh)) { 1015 nh_orig = wn[i].nh; 1016 break; 1017 } 1018 } 1019 1020 if (nh_orig == NULL) 1021 return (ESRCH); 1022 1023 error = change_nhop(rnh, info, nh_orig, &nh_new); 1024 if (error != 0) 1025 return (error); 1026 1027 wn_new = mallocarray(num_nhops, sizeof(struct weightened_nhop), 1028 M_TEMP, M_NOWAIT | M_ZERO); 1029 if (wn_new == NULL) { 1030 nhop_free(nh_new); 1031 return (EAGAIN); 1032 } 1033 1034 memcpy(wn_new, wn, num_nhops * sizeof(struct weightened_nhop)); 1035 for (int i = 0; i < num_nhops; i++) { 1036 if (wn[i].nh == nh_orig) { 1037 wn[i].nh = nh_new; 1038 wn[i].weight = get_info_weight(info, rnd_orig->rnd_weight); 1039 break; 1040 } 1041 } 1042 1043 error = nhgrp_get_group(rnh, wn_new, num_nhops, &rnd_new); 1044 nhop_free(nh_new); 1045 free(wn_new, M_TEMP); 1046 1047 if (error != 0) 1048 return (error); 1049 1050 error = change_route_conditional(rnh, NULL, info, rnd_orig, &rnd_new, rc); 1051 1052 return (error); 1053 } 1054 #endif 1055 1056 static int 1057 change_route(struct rib_head *rnh, struct rt_addrinfo *info, 1058 struct route_nhop_data *rnd_orig, struct rib_cmd_info *rc) 1059 { 1060 int error = 0; 1061 struct nhop_object *nh, *nh_orig; 1062 struct route_nhop_data rnd_new; 1063 1064 nh = NULL; 1065 nh_orig = rnd_orig->rnd_nhop; 1066 if (nh_orig == NULL) 1067 return (ESRCH); 1068 1069 #ifdef ROUTE_MPATH 1070 if (NH_IS_NHGRP(nh_orig)) 1071 return (change_mpath_route(rnh, info, rnd_orig, rc)); 1072 #endif 1073 1074 rnd_new.rnd_weight = get_info_weight(info, rnd_orig->rnd_weight); 1075 error = change_nhop(rnh, info, nh_orig, &rnd_new.rnd_nhop); 1076 if (error != 0) 1077 return (error); 1078 error = change_route_conditional(rnh, NULL, info, rnd_orig, &rnd_new, rc); 1079 1080 return (error); 1081 } 1082 1083 /* 1084 * Insert @rt with nhop data from @rnd_new to @rnh. 1085 * Returns 0 on success and stores operation results in @rc. 1086 */ 1087 static int 1088 add_route_nhop(struct rib_head *rnh, struct rtentry *rt, 1089 struct rt_addrinfo *info, struct route_nhop_data *rnd, 1090 struct rib_cmd_info *rc) 1091 { 1092 struct sockaddr *ndst, *netmask; 1093 struct radix_node *rn; 1094 int error = 0; 1095 1096 RIB_WLOCK_ASSERT(rnh); 1097 1098 ndst = (struct sockaddr *)rt_key(rt); 1099 netmask = info->rti_info[RTAX_NETMASK]; 1100 1101 rt->rt_nhop = rnd->rnd_nhop; 1102 rt->rt_weight = rnd->rnd_weight; 1103 rn = rnh->rnh_addaddr(ndst, netmask, &rnh->head, rt->rt_nodes); 1104 1105 if (rn != NULL) { 1106 if (rt->rt_expire > 0) 1107 tmproutes_update(rnh, rt); 1108 1109 /* Finalize notification */ 1110 rib_bump_gen(rnh); 1111 rnh->rnh_prefixes++; 1112 1113 rc->rc_cmd = RTM_ADD; 1114 rc->rc_rt = rt; 1115 rc->rc_nh_old = NULL; 1116 rc->rc_nh_new = rnd->rnd_nhop; 1117 rc->rc_nh_weight = rnd->rnd_weight; 1118 1119 rib_notify(rnh, RIB_NOTIFY_IMMEDIATE, rc); 1120 } else { 1121 /* Existing route or memory allocation failure */ 1122 error = EEXIST; 1123 } 1124 1125 return (error); 1126 } 1127 1128 /* 1129 * Switch @rt nhop/weigh to the ones specified in @rnd. 1130 * Conditionally set rt_expire if set in @info. 1131 * Returns 0 on success. 1132 */ 1133 int 1134 change_route_nhop(struct rib_head *rnh, struct rtentry *rt, 1135 struct rt_addrinfo *info, struct route_nhop_data *rnd, 1136 struct rib_cmd_info *rc) 1137 { 1138 struct nhop_object *nh_orig; 1139 1140 RIB_WLOCK_ASSERT(rnh); 1141 1142 nh_orig = rt->rt_nhop; 1143 1144 if (rnd->rnd_nhop != NULL) { 1145 /* Changing expiration & nexthop & weight to a new one */ 1146 rt_set_expire_info(rt, info); 1147 rt->rt_nhop = rnd->rnd_nhop; 1148 rt->rt_weight = rnd->rnd_weight; 1149 if (rt->rt_expire > 0) 1150 tmproutes_update(rnh, rt); 1151 } else { 1152 /* Route deletion requested. */ 1153 struct sockaddr *ndst, *netmask; 1154 struct radix_node *rn; 1155 1156 ndst = (struct sockaddr *)rt_key(rt); 1157 netmask = info->rti_info[RTAX_NETMASK]; 1158 rn = rnh->rnh_deladdr(ndst, netmask, &rnh->head); 1159 if (rn == NULL) 1160 return (ESRCH); 1161 rt = RNTORT(rn); 1162 rt->rte_flags &= ~RTF_UP; 1163 } 1164 1165 /* Finalize notification */ 1166 rib_bump_gen(rnh); 1167 if (rnd->rnd_nhop == NULL) 1168 rnh->rnh_prefixes--; 1169 1170 rc->rc_cmd = (rnd->rnd_nhop != NULL) ? RTM_CHANGE : RTM_DELETE; 1171 rc->rc_rt = rt; 1172 rc->rc_nh_old = nh_orig; 1173 rc->rc_nh_new = rnd->rnd_nhop; 1174 rc->rc_nh_weight = rnd->rnd_weight; 1175 1176 rib_notify(rnh, RIB_NOTIFY_IMMEDIATE, rc); 1177 1178 return (0); 1179 } 1180 1181 /* 1182 * Conditionally update route nhop/weight IFF data in @nhd_orig is 1183 * consistent with the current route data. 1184 * Nexthop in @nhd_new is consumed. 1185 */ 1186 int 1187 change_route_conditional(struct rib_head *rnh, struct rtentry *rt, 1188 struct rt_addrinfo *info, struct route_nhop_data *rnd_orig, 1189 struct route_nhop_data *rnd_new, struct rib_cmd_info *rc) 1190 { 1191 struct rtentry *rt_new; 1192 int error = 0; 1193 1194 RIB_WLOCK(rnh); 1195 1196 rt_new = (struct rtentry *)rnh->rnh_lookup(info->rti_info[RTAX_DST], 1197 info->rti_info[RTAX_NETMASK], &rnh->head); 1198 1199 if (rt_new == NULL) { 1200 if (rnd_orig->rnd_nhop == NULL) 1201 error = add_route_nhop(rnh, rt, info, rnd_new, rc); 1202 else { 1203 /* 1204 * Prefix does not exist, which was not our assumption. 1205 * Update @rnd_orig with the new data and return 1206 */ 1207 rnd_orig->rnd_nhop = NULL; 1208 rnd_orig->rnd_weight = 0; 1209 error = EAGAIN; 1210 } 1211 } else { 1212 /* Prefix exists, try to update */ 1213 if (rnd_orig->rnd_nhop == rt_new->rt_nhop) { 1214 /* 1215 * Nhop/mpath group hasn't changed. Flip 1216 * to the new precalculated one and return 1217 */ 1218 error = change_route_nhop(rnh, rt_new, info, rnd_new, rc); 1219 } else { 1220 /* Update and retry */ 1221 rnd_orig->rnd_nhop = rt_new->rt_nhop; 1222 rnd_orig->rnd_weight = rt_new->rt_weight; 1223 error = EAGAIN; 1224 } 1225 } 1226 1227 RIB_WUNLOCK(rnh); 1228 1229 if (error == 0) { 1230 rib_notify(rnh, RIB_NOTIFY_DELAYED, rc); 1231 1232 if (rnd_orig->rnd_nhop != NULL) 1233 nhop_free_any(rnd_orig->rnd_nhop); 1234 1235 } else { 1236 if (rnd_new->rnd_nhop != NULL) 1237 nhop_free_any(rnd_new->rnd_nhop); 1238 } 1239 1240 return (error); 1241 } 1242 1243 /* 1244 * Performs modification of routing table specificed by @action. 1245 * Table is specified by @fibnum and sa_family in @info->rti_info[RTAX_DST]. 1246 * Needs to be run in network epoch. 1247 * 1248 * Returns 0 on success and fills in @rc with action result. 1249 */ 1250 int 1251 rib_action(uint32_t fibnum, int action, struct rt_addrinfo *info, 1252 struct rib_cmd_info *rc) 1253 { 1254 int error; 1255 1256 switch (action) { 1257 case RTM_ADD: 1258 error = rib_add_route(fibnum, info, rc); 1259 break; 1260 case RTM_DELETE: 1261 error = rib_del_route(fibnum, info, rc); 1262 break; 1263 case RTM_CHANGE: 1264 error = rib_change_route(fibnum, info, rc); 1265 break; 1266 default: 1267 error = ENOTSUP; 1268 } 1269 1270 return (error); 1271 } 1272 1273 struct rt_delinfo 1274 { 1275 struct rt_addrinfo info; 1276 struct rib_head *rnh; 1277 struct rtentry *head; 1278 struct rib_cmd_info rc; 1279 }; 1280 1281 /* 1282 * Conditionally unlinks @rn from radix tree based 1283 * on info data passed in @arg. 1284 */ 1285 static int 1286 rt_checkdelroute(struct radix_node *rn, void *arg) 1287 { 1288 struct rt_delinfo *di; 1289 struct rt_addrinfo *info; 1290 struct rtentry *rt; 1291 1292 di = (struct rt_delinfo *)arg; 1293 rt = (struct rtentry *)rn; 1294 info = &di->info; 1295 1296 info->rti_info[RTAX_DST] = rt_key(rt); 1297 info->rti_info[RTAX_NETMASK] = rt_mask(rt); 1298 1299 if (rt_unlinkrte(di->rnh, info, &di->rc) != 0) 1300 return (0); 1301 1302 /* 1303 * Add deleted rtentries to the list to GC them 1304 * after dropping the lock. 1305 * 1306 * XXX: Delayed notifications not implemented 1307 * for nexthop updates. 1308 */ 1309 if (di->rc.rc_cmd == RTM_DELETE) { 1310 /* Add to the list and return */ 1311 rt->rt_chain = di->head; 1312 di->head = rt; 1313 #ifdef ROUTE_MPATH 1314 } else { 1315 /* 1316 * RTM_CHANGE to a diferent nexthop or nexthop group. 1317 * Free old multipath group. 1318 */ 1319 nhop_free_any(di->rc.rc_nh_old); 1320 #endif 1321 } 1322 1323 return (0); 1324 } 1325 1326 /* 1327 * Iterates over a routing table specified by @fibnum and @family and 1328 * deletes elements marked by @filter_f. 1329 * @fibnum: rtable id 1330 * @family: AF_ address family 1331 * @filter_f: function returning non-zero value for items to delete 1332 * @arg: data to pass to the @filter_f function 1333 * @report: true if rtsock notification is needed. 1334 */ 1335 void 1336 rib_walk_del(u_int fibnum, int family, rib_filter_f_t *filter_f, void *arg, bool report) 1337 { 1338 struct rib_head *rnh; 1339 struct rt_delinfo di; 1340 struct rtentry *rt; 1341 struct nhop_object *nh; 1342 struct epoch_tracker et; 1343 1344 rnh = rt_tables_get_rnh(fibnum, family); 1345 if (rnh == NULL) 1346 return; 1347 1348 bzero(&di, sizeof(di)); 1349 di.info.rti_filter = filter_f; 1350 di.info.rti_filterdata = arg; 1351 di.rnh = rnh; 1352 di.rc.rc_cmd = RTM_DELETE; 1353 1354 NET_EPOCH_ENTER(et); 1355 1356 RIB_WLOCK(rnh); 1357 rnh->rnh_walktree(&rnh->head, rt_checkdelroute, &di); 1358 RIB_WUNLOCK(rnh); 1359 1360 /* We might have something to reclaim. */ 1361 bzero(&di.rc, sizeof(di.rc)); 1362 di.rc.rc_cmd = RTM_DELETE; 1363 while (di.head != NULL) { 1364 rt = di.head; 1365 di.head = rt->rt_chain; 1366 rt->rt_chain = NULL; 1367 nh = rt->rt_nhop; 1368 1369 di.rc.rc_rt = rt; 1370 di.rc.rc_nh_old = nh; 1371 rib_notify(rnh, RIB_NOTIFY_DELAYED, &di.rc); 1372 1373 /* TODO std rt -> rt_addrinfo export */ 1374 di.info.rti_info[RTAX_DST] = rt_key(rt); 1375 di.info.rti_info[RTAX_NETMASK] = rt_mask(rt); 1376 1377 if (report) { 1378 #ifdef ROUTE_MPATH 1379 struct nhgrp_object *nhg; 1380 struct weightened_nhop *wn; 1381 uint32_t num_nhops; 1382 if (NH_IS_NHGRP(nh)) { 1383 nhg = (struct nhgrp_object *)nh; 1384 wn = nhgrp_get_nhops(nhg, &num_nhops); 1385 for (int i = 0; i < num_nhops; i++) 1386 rt_routemsg(RTM_DELETE, rt, wn[i].nh, fibnum); 1387 } else 1388 #endif 1389 rt_routemsg(RTM_DELETE, rt, nh, fibnum); 1390 } 1391 rtfree(rt); 1392 } 1393 1394 NET_EPOCH_EXIT(et); 1395 } 1396 1397 static int 1398 rt_delete_unconditional(struct radix_node *rn, void *arg) 1399 { 1400 struct rtentry *rt = RNTORT(rn); 1401 struct rib_head *rnh = (struct rib_head *)arg; 1402 1403 rn = rnh->rnh_deladdr(rt_key(rt), rt_mask(rt), &rnh->head); 1404 if (RNTORT(rn) == rt) 1405 rtfree(rt); 1406 1407 return (0); 1408 } 1409 1410 /* 1411 * Removes all routes from the routing table without executing notifications. 1412 * rtentres will be removed after the end of a current epoch. 1413 */ 1414 static void 1415 rib_flush_routes(struct rib_head *rnh) 1416 { 1417 RIB_WLOCK(rnh); 1418 rnh->rnh_walktree(&rnh->head, rt_delete_unconditional, rnh); 1419 RIB_WUNLOCK(rnh); 1420 } 1421 1422 void 1423 rib_flush_routes_family(int family) 1424 { 1425 struct rib_head *rnh; 1426 1427 for (uint32_t fibnum = 0; fibnum < rt_numfibs; fibnum++) { 1428 if ((rnh = rt_tables_get_rnh(fibnum, family)) != NULL) 1429 rib_flush_routes(rnh); 1430 } 1431 } 1432 1433 static void 1434 rib_notify(struct rib_head *rnh, enum rib_subscription_type type, 1435 struct rib_cmd_info *rc) 1436 { 1437 struct rib_subscription *rs; 1438 1439 CK_STAILQ_FOREACH(rs, &rnh->rnh_subscribers, next) { 1440 if (rs->type == type) 1441 rs->func(rnh, rc, rs->arg); 1442 } 1443 } 1444 1445 static struct rib_subscription * 1446 allocate_subscription(rib_subscription_cb_t *f, void *arg, 1447 enum rib_subscription_type type, bool waitok) 1448 { 1449 struct rib_subscription *rs; 1450 int flags = M_ZERO | (waitok ? M_WAITOK : M_NOWAIT); 1451 1452 rs = malloc(sizeof(struct rib_subscription), M_RTABLE, flags); 1453 if (rs == NULL) 1454 return (NULL); 1455 1456 rs->func = f; 1457 rs->arg = arg; 1458 rs->type = type; 1459 1460 return (rs); 1461 } 1462 1463 /* 1464 * Subscribe for the changes in the routing table specified by @fibnum and 1465 * @family. 1466 * 1467 * Returns pointer to the subscription structure on success. 1468 */ 1469 struct rib_subscription * 1470 rib_subscribe(uint32_t fibnum, int family, rib_subscription_cb_t *f, void *arg, 1471 enum rib_subscription_type type, bool waitok) 1472 { 1473 struct rib_head *rnh; 1474 struct epoch_tracker et; 1475 1476 NET_EPOCH_ENTER(et); 1477 KASSERT((fibnum < rt_numfibs), ("%s: bad fibnum", __func__)); 1478 rnh = rt_tables_get_rnh(fibnum, family); 1479 NET_EPOCH_EXIT(et); 1480 1481 return (rib_subscribe_internal(rnh, f, arg, type, waitok)); 1482 } 1483 1484 struct rib_subscription * 1485 rib_subscribe_internal(struct rib_head *rnh, rib_subscription_cb_t *f, void *arg, 1486 enum rib_subscription_type type, bool waitok) 1487 { 1488 struct rib_subscription *rs; 1489 struct epoch_tracker et; 1490 1491 if ((rs = allocate_subscription(f, arg, type, waitok)) == NULL) 1492 return (NULL); 1493 rs->rnh = rnh; 1494 1495 NET_EPOCH_ENTER(et); 1496 RIB_WLOCK(rnh); 1497 CK_STAILQ_INSERT_HEAD(&rnh->rnh_subscribers, rs, next); 1498 RIB_WUNLOCK(rnh); 1499 NET_EPOCH_EXIT(et); 1500 1501 return (rs); 1502 } 1503 1504 struct rib_subscription * 1505 rib_subscribe_locked(struct rib_head *rnh, rib_subscription_cb_t *f, void *arg, 1506 enum rib_subscription_type type) 1507 { 1508 struct rib_subscription *rs; 1509 1510 NET_EPOCH_ASSERT(); 1511 RIB_WLOCK_ASSERT(rnh); 1512 1513 if ((rs = allocate_subscription(f, arg, type, false)) == NULL) 1514 return (NULL); 1515 rs->rnh = rnh; 1516 1517 CK_STAILQ_INSERT_HEAD(&rnh->rnh_subscribers, rs, next); 1518 1519 return (rs); 1520 } 1521 1522 /* 1523 * Remove rtable subscription @rs from the routing table. 1524 * Needs to be run in network epoch. 1525 */ 1526 void 1527 rib_unsubscribe(struct rib_subscription *rs) 1528 { 1529 struct rib_head *rnh = rs->rnh; 1530 1531 NET_EPOCH_ASSERT(); 1532 1533 RIB_WLOCK(rnh); 1534 CK_STAILQ_REMOVE(&rnh->rnh_subscribers, rs, rib_subscription, next); 1535 RIB_WUNLOCK(rnh); 1536 1537 epoch_call(net_epoch_preempt, destroy_subscription_epoch, 1538 &rs->epoch_ctx); 1539 } 1540 1541 void 1542 rib_unsubscribe_locked(struct rib_subscription *rs) 1543 { 1544 struct rib_head *rnh = rs->rnh; 1545 1546 NET_EPOCH_ASSERT(); 1547 RIB_WLOCK_ASSERT(rnh); 1548 1549 CK_STAILQ_REMOVE(&rnh->rnh_subscribers, rs, rib_subscription, next); 1550 1551 epoch_call(net_epoch_preempt, destroy_subscription_epoch, 1552 &rs->epoch_ctx); 1553 } 1554 1555 /* 1556 * Epoch callback indicating subscription is safe to destroy 1557 */ 1558 static void 1559 destroy_subscription_epoch(epoch_context_t ctx) 1560 { 1561 struct rib_subscription *rs; 1562 1563 rs = __containerof(ctx, struct rib_subscription, epoch_ctx); 1564 1565 free(rs, M_RTABLE); 1566 } 1567 1568 void 1569 rib_init_subscriptions(struct rib_head *rnh) 1570 { 1571 1572 CK_STAILQ_INIT(&rnh->rnh_subscribers); 1573 } 1574 1575 void 1576 rib_destroy_subscriptions(struct rib_head *rnh) 1577 { 1578 struct rib_subscription *rs; 1579 struct epoch_tracker et; 1580 1581 NET_EPOCH_ENTER(et); 1582 RIB_WLOCK(rnh); 1583 while ((rs = CK_STAILQ_FIRST(&rnh->rnh_subscribers)) != NULL) { 1584 CK_STAILQ_REMOVE_HEAD(&rnh->rnh_subscribers, next); 1585 epoch_call(net_epoch_preempt, destroy_subscription_epoch, 1586 &rs->epoch_ctx); 1587 } 1588 RIB_WUNLOCK(rnh); 1589 NET_EPOCH_EXIT(et); 1590 } 1591