xref: /freebsd/sys/net/route/nhop_ctl.c (revision 15fccd86b9dd17ca2cc1a70b060a2b1a1bf8470b)
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_route.h"
32 
33 #include <sys/param.h>
34 #include <sys/systm.h>
35 #include <sys/lock.h>
36 #include <sys/rwlock.h>
37 #include <sys/malloc.h>
38 #include <sys/socket.h>
39 #include <sys/sysctl.h>
40 #include <sys/kernel.h>
41 #include <sys/epoch.h>
42 
43 #include <net/if.h>
44 #include <net/if_var.h>
45 #include <net/if_dl.h>
46 #include <net/route.h>
47 #include <net/route/route_ctl.h>
48 #include <net/route/route_var.h>
49 #include <net/route/nhop_utils.h>
50 #include <net/route/nhop.h>
51 #include <net/route/nhop_var.h>
52 #include <net/vnet.h>
53 
54 /*
55  * This file contains core functionality for the nexthop ("nhop") route subsystem.
56  * The business logic needed to create nexhop objects is implemented here.
57  *
58  * Nexthops in the original sense are the objects containing all the necessary
59  * information to forward the packet to the selected destination.
60  * In particular, nexthop is defined by a combination of
61  *  ifp, ifa, aifp, mtu, gw addr(if set), nh_type, nh_family, mask of rt_flags and
62  *    NHF_DEFAULT
63  *
64  * Additionally, each nexthop gets assigned its unique index (nexthop index).
65  * It serves two purposes: first one is to ease the ability of userland programs to
66  *  reference nexthops by their index. The second one allows lookup algorithms to
67  *  to store index instead of pointer (2 bytes vs 8) as a lookup result.
68  * All nexthops are stored in the resizable hash table.
69  *
70  * Basically, this file revolves around supporting 3 functions:
71  * 1) nhop_create_from_info / nhop_create_from_nhop, which contains all
72  *  business logic on filling the nexthop fields based on the provided request.
73  * 2) nhop_get(), which gets a usable referenced nexthops.
74  *
75  * Conventions:
76  * 1) non-exported functions start with verb
77  * 2) exported function starts with the subsystem prefix: "nhop"
78  */
79 
80 static int dump_nhop_entry(struct rib_head *rh, struct nhop_object *nh, struct sysctl_req *w);
81 
82 static struct nhop_priv *alloc_nhop_structure(void);
83 static int get_nhop(struct rib_head *rnh, struct rt_addrinfo *info,
84     struct nhop_priv **pnh_priv);
85 static int finalize_nhop(struct nh_control *ctl, struct rt_addrinfo *info,
86     struct nhop_priv *nh_priv);
87 static struct ifnet *get_aifp(const struct nhop_object *nh);
88 static void fill_sdl_from_ifp(struct sockaddr_dl_short *sdl, const struct ifnet *ifp);
89 
90 static void destroy_nhop_epoch(epoch_context_t ctx);
91 static void destroy_nhop(struct nhop_priv *nh_priv);
92 
93 static void print_nhop(const char *prefix, const struct nhop_object *nh);
94 
95 _Static_assert(__offsetof(struct nhop_object, nh_ifp) == 32,
96     "nhop_object: wrong nh_ifp offset");
97 _Static_assert(sizeof(struct nhop_object) <= 128,
98     "nhop_object: size exceeds 128 bytes");
99 
100 static uma_zone_t nhops_zone;	/* Global zone for each and every nexthop */
101 
102 #define	NHOP_OBJECT_ALIGNED_SIZE	roundup2(sizeof(struct nhop_object), \
103 							2 * CACHE_LINE_SIZE)
104 #define	NHOP_PRIV_ALIGNED_SIZE		roundup2(sizeof(struct nhop_priv), \
105 							2 * CACHE_LINE_SIZE)
106 void
107 nhops_init(void)
108 {
109 
110 	nhops_zone = uma_zcreate("routing nhops",
111 	    NHOP_OBJECT_ALIGNED_SIZE + NHOP_PRIV_ALIGNED_SIZE,
112 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
113 }
114 
115 /*
116  * Fetches the interface of source address used by the route.
117  * In all cases except interface-address-route it would be the
118  * same as the transmit interfaces.
119  * However, for the interface address this function will return
120  * this interface ifp instead of loopback. This is needed to support
121  * link-local IPv6 loopback communications.
122  *
123  * Returns found ifp.
124  */
125 static struct ifnet *
126 get_aifp(const struct nhop_object *nh)
127 {
128 	struct ifnet *aifp = NULL;
129 
130 	/*
131 	 * Adjust the "outgoing" interface.  If we're going to loop
132 	 * the packet back to ourselves, the ifp would be the loopback
133 	 * interface. However, we'd rather know the interface associated
134 	 * to the destination address (which should probably be one of
135 	 * our own addresses).
136 	 */
137 	if ((nh->nh_ifp->if_flags & IFF_LOOPBACK) &&
138 			nh->gw_sa.sa_family == AF_LINK) {
139 		aifp = ifnet_byindex(nh->gwl_sa.sdl_index);
140 		if (aifp == NULL) {
141 			DPRINTF("unable to get aifp for %s index %d",
142 				if_name(nh->nh_ifp), nh->gwl_sa.sdl_index);
143 		}
144 	}
145 
146 	if (aifp == NULL)
147 		aifp = nh->nh_ifp;
148 
149 	return (aifp);
150 }
151 
152 int
153 cmp_priv(const struct nhop_priv *_one, const struct nhop_priv *_two)
154 {
155 
156 	if (memcmp(_one->nh, _two->nh, NHOP_END_CMP) != 0)
157 		return (0);
158 
159 	if (memcmp(_one, _two, NH_PRIV_END_CMP) != 0)
160 		return (0);
161 
162 	return (1);
163 }
164 
165 /*
166  * Conditionally sets @nh mtu data based on the @info data.
167  */
168 static void
169 set_nhop_mtu_from_info(struct nhop_object *nh, const struct rt_addrinfo *info)
170 {
171 
172 	if (info->rti_mflags & RTV_MTU) {
173 		if (info->rti_rmx->rmx_mtu != 0) {
174 			/*
175 			 * MTU was explicitly provided by user.
176 			 * Keep it.
177 			 */
178 
179 			nh->nh_priv->rt_flags |= RTF_FIXEDMTU;
180 		} else {
181 			/*
182 			 * User explicitly sets MTU to 0.
183 			 * Assume rollback to default.
184 			 */
185 			nh->nh_priv->rt_flags &= ~RTF_FIXEDMTU;
186 		}
187 		nh->nh_mtu = info->rti_rmx->rmx_mtu;
188 	}
189 }
190 
191 /*
192  * Fills in shorted link-level sockadd version suitable to be stored inside the
193  *  nexthop gateway buffer.
194  */
195 static void
196 fill_sdl_from_ifp(struct sockaddr_dl_short *sdl, const struct ifnet *ifp)
197 {
198 
199 	bzero(sdl, sizeof(struct sockaddr_dl_short));
200 	sdl->sdl_family = AF_LINK;
201 	sdl->sdl_len = sizeof(struct sockaddr_dl_short);
202 	sdl->sdl_index = ifp->if_index;
203 	sdl->sdl_type = ifp->if_type;
204 }
205 
206 static int
207 set_nhop_gw_from_info(struct nhop_object *nh, struct rt_addrinfo *info)
208 {
209 	struct sockaddr *gw;
210 
211 	gw = info->rti_info[RTAX_GATEWAY];
212 	KASSERT(gw != NULL, ("gw is NULL"));
213 
214 	if ((gw->sa_family == AF_LINK) && !(info->rti_flags & RTF_GATEWAY)) {
215 
216 		/*
217 		 * Interface route with interface specified by the interface
218 		 * index in sockadd_dl structure. It is used in the IPv6 loopback
219 		 * output code, where we need to preserve the original interface
220 		 * to maintain proper scoping.
221 		 * Despite the fact that nexthop code stores original interface
222 		 * in the separate field (nh_aifp, see below), write AF_LINK
223 		 * compatible sa with shorter total length.
224 		 */
225 		struct sockaddr_dl *sdl = (struct sockaddr_dl *)gw;
226 		struct ifnet *ifp = ifnet_byindex(sdl->sdl_index);
227 		if (ifp == NULL) {
228 			DPRINTF("invalid ifindex %d", sdl->sdl_index);
229 			return (EINVAL);
230 		}
231 		fill_sdl_from_ifp(&nh->gwl_sa, ifp);
232 	} else {
233 
234 		/*
235 		 * Multiple options here:
236 		 *
237 		 * 1) RTF_GATEWAY with IPv4/IPv6 gateway data
238 		 * 2) Interface route with IPv4/IPv6 address of the
239 		 *   matching interface. Some routing daemons do that
240 		 *   instead of specifying ifindex in AF_LINK.
241 		 *
242 		 * In both cases, save the original nexthop to make the callers
243 		 *   happy.
244 		 */
245 		if (gw->sa_len > sizeof(struct sockaddr_in6)) {
246 			DPRINTF("nhop SA size too big: AF %d len %u",
247 			    gw->sa_family, gw->sa_len);
248 			return (ENOMEM);
249 		}
250 		memcpy(&nh->gw_sa, gw, gw->sa_len);
251 	}
252 	return (0);
253 }
254 
255 static uint16_t
256 convert_rt_to_nh_flags(int rt_flags)
257 {
258 	uint16_t res;
259 
260 	res = (rt_flags & RTF_REJECT) ? NHF_REJECT : 0;
261 	res |= (rt_flags & RTF_HOST) ? NHF_HOST : 0;
262 	res |= (rt_flags & RTF_BLACKHOLE) ? NHF_BLACKHOLE : 0;
263 	res |= (rt_flags & (RTF_DYNAMIC|RTF_MODIFIED)) ? NHF_REDIRECT : 0;
264 	res |= (rt_flags & RTF_BROADCAST) ? NHF_BROADCAST : 0;
265 	res |= (rt_flags & RTF_GATEWAY) ? NHF_GATEWAY : 0;
266 
267 	return (res);
268 }
269 
270 static int
271 fill_nhop_from_info(struct nhop_priv *nh_priv, struct rt_addrinfo *info)
272 {
273 	int error, rt_flags;
274 	struct nhop_object *nh;
275 
276 	nh = nh_priv->nh;
277 
278 	rt_flags = info->rti_flags & NHOP_RT_FLAG_MASK;
279 
280 	nh->nh_priv->rt_flags = rt_flags;
281 	nh_priv->nh_family = info->rti_info[RTAX_DST]->sa_family;
282 	nh_priv->nh_type = 0; // hook responsibility to set nhop type
283 
284 	nh->nh_flags = convert_rt_to_nh_flags(rt_flags);
285 	set_nhop_mtu_from_info(nh, info);
286 	if ((error = set_nhop_gw_from_info(nh, info)) != 0)
287 		return (error);
288 
289 	nh->nh_ifp = info->rti_ifa->ifa_ifp;
290 	nh->nh_ifa = info->rti_ifa;
291 	/* depends on the gateway */
292 	nh->nh_aifp = get_aifp(nh);
293 
294 	/*
295 	 * Note some of the remaining data is set by the
296 	 * per-address-family pre-add hook.
297 	 */
298 
299 	return (0);
300 }
301 
302 /*
303  * Creates a new nexthop based on the information in @info.
304  *
305  * Returns:
306  * 0 on success, filling @nh_ret with the desired nexthop object ptr
307  * errno otherwise
308  */
309 int
310 nhop_create_from_info(struct rib_head *rnh, struct rt_addrinfo *info,
311     struct nhop_object **nh_ret)
312 {
313 	struct nhop_priv *nh_priv;
314 	int error;
315 
316 	NET_EPOCH_ASSERT();
317 
318 	if (info->rti_info[RTAX_GATEWAY] == NULL)
319 		return (EINVAL);
320 
321 	nh_priv = alloc_nhop_structure();
322 
323 	error = fill_nhop_from_info(nh_priv, info);
324 	if (error != 0) {
325 		uma_zfree(nhops_zone, nh_priv->nh);
326 		return (error);
327 	}
328 
329 	error = get_nhop(rnh, info, &nh_priv);
330 	if (error == 0)
331 		*nh_ret = nh_priv->nh;
332 
333 	return (error);
334 }
335 
336 /*
337  * Gets linked nhop using the provided @pnh_priv nexhop data.
338  * If linked nhop is found, returns it, freeing the provided one.
339  * If there is no such nexthop, attaches the remaining data to the
340  *  provided nexthop and links it.
341  *
342  * Returns 0 on success, storing referenced nexthop in @pnh_priv.
343  * Otherwise, errno is returned.
344  */
345 static int
346 get_nhop(struct rib_head *rnh, struct rt_addrinfo *info,
347     struct nhop_priv **pnh_priv)
348 {
349 	const struct sockaddr *dst, *gateway, *netmask;
350 	struct nhop_priv *nh_priv, *tmp_priv;
351 	int error;
352 
353 	nh_priv = *pnh_priv;
354 
355 	/* Give the protocols chance to augment the request data */
356 	dst = info->rti_info[RTAX_DST];
357 	netmask = info->rti_info[RTAX_NETMASK];
358 	gateway = info->rti_info[RTAX_GATEWAY];
359 
360 	error = rnh->rnh_preadd(rnh->rib_fibnum, dst, netmask, nh_priv->nh);
361 	if (error != 0) {
362 		uma_zfree(nhops_zone, nh_priv->nh);
363 		return (error);
364 	}
365 
366 	tmp_priv = find_nhop(rnh->nh_control, nh_priv);
367 	if (tmp_priv != NULL) {
368 		uma_zfree(nhops_zone, nh_priv->nh);
369 		*pnh_priv = tmp_priv;
370 		return (0);
371 	}
372 
373 	/*
374 	 * Existing nexthop not found, need to create new one.
375 	 * Note: multiple simultaneous get_nhop() requests
376 	 *  can result in multiple equal nexhops existing in the
377 	 *  nexthop table. This is not a not a problem until the
378 	 *  relative number of such nexthops is significant, which
379 	 *  is extremely unlikely.
380 	 */
381 
382 	error = finalize_nhop(rnh->nh_control, info, nh_priv);
383 	if (error != 0)
384 		return (error);
385 
386 	return (0);
387 }
388 
389 /*
390  * Update @nh with data supplied in @info.
391  * This is a helper function to support route changes.
392  *
393  * It limits the changes that can be done to the route to the following:
394  * 1) all combination of gateway changes (gw, interface, blackhole/reject)
395  * 2) route flags (FLAG[123],STATIC,BLACKHOLE,REJECT)
396  * 3) route MTU
397  *
398  * Returns:
399  * 0 on success
400  */
401 static int
402 alter_nhop_from_info(struct nhop_object *nh, struct rt_addrinfo *info)
403 {
404 	struct sockaddr *info_gw;
405 	int error;
406 
407 	/* Update MTU if set in the request*/
408 	set_nhop_mtu_from_info(nh, info);
409 
410 	/* XXX: allow only one of BLACKHOLE,REJECT,GATEWAY */
411 
412 	/* Allow some flags (FLAG1,STATIC,BLACKHOLE,REJECT) to be toggled on change. */
413 	nh->nh_priv->rt_flags &= ~RTF_FMASK;
414 	nh->nh_priv->rt_flags |= info->rti_flags & RTF_FMASK;
415 
416 	/* Consider gateway change */
417 	info_gw = info->rti_info[RTAX_GATEWAY];
418 	if (info_gw != NULL) {
419 		error = set_nhop_gw_from_info(nh, info);
420 		if (error != 0)
421 			return (error);
422 		/* Update RTF_GATEWAY flag status */
423 		nh->nh_priv->rt_flags &= ~RTF_GATEWAY;
424 		nh->nh_priv->rt_flags |= (RTF_GATEWAY & info->rti_flags);
425 	}
426 	/* Update datapath flags */
427 	nh->nh_flags = convert_rt_to_nh_flags(nh->nh_priv->rt_flags);
428 
429 	if (info->rti_ifa != NULL)
430 		nh->nh_ifa = info->rti_ifa;
431 	if (info->rti_ifp != NULL)
432 		nh->nh_ifp = info->rti_ifp;
433 	nh->nh_aifp = get_aifp(nh);
434 
435 	return (0);
436 }
437 
438 /*
439  * Creates new nexthop based on @nh_orig and augmentation data from @info.
440  * Helper function used in the route changes, please see
441  *   alter_nhop_from_info() comments for more details.
442  *
443  * Returns:
444  * 0 on success, filling @nh_ret with the desired nexthop object
445  * errno otherwise
446  */
447 int
448 nhop_create_from_nhop(struct rib_head *rnh, const struct nhop_object *nh_orig,
449     struct rt_addrinfo *info, struct nhop_object **pnh)
450 {
451 	struct nhop_priv *nh_priv;
452 	struct nhop_object *nh;
453 	int error;
454 
455 	NET_EPOCH_ASSERT();
456 
457 	nh_priv = alloc_nhop_structure();
458 	nh = nh_priv->nh;
459 
460 	/* Start with copying data from original nexthop */
461 	nh_priv->nh_family = nh_orig->nh_priv->nh_family;
462 	nh_priv->rt_flags = nh_orig->nh_priv->rt_flags;
463 	nh_priv->nh_type = nh_orig->nh_priv->nh_type;
464 
465 	nh->nh_ifp = nh_orig->nh_ifp;
466 	nh->nh_ifa = nh_orig->nh_ifa;
467 	nh->nh_aifp = nh_orig->nh_aifp;
468 	nh->nh_mtu = nh_orig->nh_mtu;
469 	nh->nh_flags = nh_orig->nh_flags;
470 	memcpy(&nh->gw_sa, &nh_orig->gw_sa, nh_orig->gw_sa.sa_len);
471 
472 	error = alter_nhop_from_info(nh, info);
473 	if (error != 0) {
474 		uma_zfree(nhops_zone, nh_priv->nh);
475 		return (error);
476 	}
477 
478 	error = get_nhop(rnh, info, &nh_priv);
479 	if (error == 0)
480 		*pnh = nh_priv->nh;
481 
482 	return (error);
483 }
484 
485 /*
486  * Allocates memory for public/private nexthop structures.
487  *
488  * Returns pointer to nhop_priv or NULL.
489  */
490 static struct nhop_priv *
491 alloc_nhop_structure()
492 {
493 	struct nhop_object *nh;
494 	struct nhop_priv *nh_priv;
495 
496 	nh = (struct nhop_object *)uma_zalloc(nhops_zone, M_NOWAIT | M_ZERO);
497 	if (nh == NULL)
498 		return (NULL);
499 	nh_priv = (struct nhop_priv *)((char *)nh + NHOP_OBJECT_ALIGNED_SIZE);
500 
501 	nh->nh_priv = nh_priv;
502 	nh_priv->nh = nh;
503 
504 	return (nh_priv);
505 }
506 
507 static bool
508 reference_nhop_deps(struct nhop_object *nh)
509 {
510 	if (!ifa_try_ref(nh->nh_ifa))
511 		return (false);
512 	nh->nh_aifp = get_aifp(nh);
513 	if (!if_try_ref(nh->nh_aifp)) {
514 		ifa_free(nh->nh_ifa);
515 		return (false);
516 	}
517 	DPRINTF("AIFP: %p nh_ifp %p", nh->nh_aifp, nh->nh_ifp);
518 	if (!if_try_ref(nh->nh_ifp)) {
519 		ifa_free(nh->nh_ifa);
520 		if_rele(nh->nh_aifp);
521 		return (false);
522 	}
523 
524 	return (true);
525 }
526 
527 /*
528  * Alocates/references the remaining bits of nexthop data and links
529  *  it to the hash table.
530  * Returns 0 if successful,
531  *  errno otherwise. @nh_priv is freed in case of error.
532  */
533 static int
534 finalize_nhop(struct nh_control *ctl, struct rt_addrinfo *info,
535     struct nhop_priv *nh_priv)
536 {
537 	struct nhop_object *nh = nh_priv->nh;
538 
539 	/* Allocate per-cpu packet counter */
540 	nh->nh_pksent = counter_u64_alloc(M_NOWAIT);
541 	if (nh->nh_pksent == NULL) {
542 		uma_zfree(nhops_zone, nh);
543 		RTSTAT_INC(rts_nh_alloc_failure);
544 		DPRINTF("nh_alloc_finalize failed");
545 		return (ENOMEM);
546 	}
547 
548 	if (!reference_nhop_deps(nh)) {
549 		counter_u64_free(nh->nh_pksent);
550 		uma_zfree(nhops_zone, nh);
551 		RTSTAT_INC(rts_nh_alloc_failure);
552 		DPRINTF("nh_alloc_finalize failed - reference failure");
553 		return (EAGAIN);
554 	}
555 
556 	/* Save vnet to ease destruction */
557 	nh_priv->nh_vnet = curvnet;
558 
559 	refcount_init(&nh_priv->nh_refcnt, 1);
560 
561 	/* Please see nhop_free() comments on the initial value */
562 	refcount_init(&nh_priv->nh_linked, 2);
563 
564 	print_nhop("FINALIZE", nh);
565 
566 	if (link_nhop(ctl, nh_priv) == 0) {
567 		/*
568 		 * Adding nexthop to the datastructures
569 		 *  failed. Call destructor w/o waiting for
570 		 *  the epoch end, as nexthop is not used
571 		 *  and return.
572 		 */
573 		DPRINTF("link_nhop failed!");
574 		destroy_nhop(nh_priv);
575 
576 		return (ENOBUFS);
577 	}
578 
579 	return (0);
580 }
581 
582 static void
583 print_nhop_sa(char *buf, size_t buflen, const struct sockaddr *sa)
584 {
585 
586 	if (sa->sa_family == AF_INET) {
587 		const struct sockaddr_in *sin4;
588 		sin4 = (const struct sockaddr_in *)sa;
589 		inet_ntop(AF_INET, &sin4->sin_addr, buf, buflen);
590 	} else if (sa->sa_family == AF_INET6) {
591 		const struct sockaddr_in6 *sin6;
592 		sin6 = (const struct sockaddr_in6 *)sa;
593 		inet_ntop(AF_INET6, &sin6->sin6_addr, buf, buflen);
594 	} else if (sa->sa_family == AF_LINK) {
595 		const struct sockaddr_dl *sdl;
596 		sdl = (const struct sockaddr_dl *)sa;
597 		snprintf(buf, buflen, "if#%d", sdl->sdl_index);
598 	} else
599 		snprintf(buf, buflen, "af:%d", sa->sa_family);
600 }
601 
602 static void
603 print_nhop(const char *prefix, const struct nhop_object *nh)
604 {
605 	char src_buf[INET6_ADDRSTRLEN], addr_buf[INET6_ADDRSTRLEN];
606 
607 	print_nhop_sa(src_buf, sizeof(src_buf), nh->nh_ifa->ifa_addr);
608 	print_nhop_sa(addr_buf, sizeof(addr_buf), &nh->gw_sa);
609 
610 	DPRINTF("%s nhop priv %p: AF %d ifp %p %s addr %s src %p %s aifp %p %s mtu %d nh_flags %X",
611 	    prefix, nh->nh_priv, nh->nh_priv->nh_family, nh->nh_ifp,
612 	    if_name(nh->nh_ifp), addr_buf, nh->nh_ifa, src_buf, nh->nh_aifp,
613 	    if_name(nh->nh_aifp), nh->nh_mtu, nh->nh_flags);
614 }
615 
616 static void
617 destroy_nhop(struct nhop_priv *nh_priv)
618 {
619 	struct nhop_object *nh;
620 
621 	nh = nh_priv->nh;
622 
623 	print_nhop("DEL", nh);
624 
625 	if_rele(nh->nh_ifp);
626 	if_rele(nh->nh_aifp);
627 	ifa_free(nh->nh_ifa);
628 	counter_u64_free(nh->nh_pksent);
629 
630 	uma_zfree(nhops_zone, nh);
631 }
632 
633 /*
634  * Epoch callback indicating nhop is safe to destroy
635  */
636 static void
637 destroy_nhop_epoch(epoch_context_t ctx)
638 {
639 	struct nhop_priv *nh_priv;
640 
641 	nh_priv = __containerof(ctx, struct nhop_priv, nh_epoch_ctx);
642 
643 	destroy_nhop(nh_priv);
644 }
645 
646 void
647 nhop_ref_object(struct nhop_object *nh)
648 {
649 	u_int old;
650 
651 	old = refcount_acquire(&nh->nh_priv->nh_refcnt);
652 	KASSERT(old > 0, ("%s: nhop object %p has 0 refs", __func__, nh));
653 }
654 
655 int
656 nhop_try_ref_object(struct nhop_object *nh)
657 {
658 
659 	return (refcount_acquire_if_not_zero(&nh->nh_priv->nh_refcnt));
660 }
661 
662 void
663 nhop_free(struct nhop_object *nh)
664 {
665 	struct nh_control *ctl;
666 	struct nhop_priv *nh_priv = nh->nh_priv;
667 	struct epoch_tracker et;
668 
669 	if (!refcount_release(&nh_priv->nh_refcnt))
670 		return;
671 
672 	/*
673 	 * There are only 2 places, where nh_linked can be decreased:
674 	 *  rib destroy (nhops_destroy_rib) and this function.
675 	 * nh_link can never be increased.
676 	 *
677 	 * Hence, use initial value of 2 to make use of
678 	 *  refcount_release_if_not_last().
679 	 *
680 	 * There can be two scenarious when calling this function:
681 	 *
682 	 * 1) nh_linked value is 2. This means that either
683 	 *  nhops_destroy_rib() has not been called OR it is running,
684 	 *  but we are guaranteed that nh_control won't be freed in
685 	 *  this epoch. Hence, nexthop can be safely unlinked.
686 	 *
687 	 * 2) nh_linked value is 1. In that case, nhops_destroy_rib()
688 	 *  has been called and nhop unlink can be skipped.
689 	 */
690 
691 	NET_EPOCH_ENTER(et);
692 	if (refcount_release_if_not_last(&nh_priv->nh_linked)) {
693 		ctl = nh_priv->nh_control;
694 		if (unlink_nhop(ctl, nh_priv) == NULL) {
695 			/* Do not try to reclaim */
696 			DPRINTF("Failed to unlink nexhop %p", nh_priv);
697 			NET_EPOCH_EXIT(et);
698 			return;
699 		}
700 	}
701 	NET_EPOCH_EXIT(et);
702 
703 	epoch_call(net_epoch_preempt, destroy_nhop_epoch,
704 	    &nh_priv->nh_epoch_ctx);
705 }
706 
707 void
708 nhop_ref_any(struct nhop_object *nh)
709 {
710 #ifdef ROUTE_MPATH
711 	if (!NH_IS_NHGRP(nh))
712 		nhop_ref_object(nh);
713 	else
714 		nhgrp_ref_object((struct nhgrp_object *)nh);
715 #else
716 	nhop_ref_object(nh);
717 #endif
718 }
719 
720 void
721 nhop_free_any(struct nhop_object *nh)
722 {
723 
724 #ifdef ROUTE_MPATH
725 	if (!NH_IS_NHGRP(nh))
726 		nhop_free(nh);
727 	else
728 		nhgrp_free((struct nhgrp_object *)nh);
729 #else
730 	nhop_free(nh);
731 #endif
732 }
733 
734 /* Helper functions */
735 
736 uint32_t
737 nhop_get_idx(const struct nhop_object *nh)
738 {
739 
740 	return (nh->nh_priv->nh_idx);
741 }
742 
743 enum nhop_type
744 nhop_get_type(const struct nhop_object *nh)
745 {
746 
747 	return (nh->nh_priv->nh_type);
748 }
749 
750 void
751 nhop_set_type(struct nhop_object *nh, enum nhop_type nh_type)
752 {
753 
754 	nh->nh_priv->nh_type = nh_type;
755 }
756 
757 int
758 nhop_get_rtflags(const struct nhop_object *nh)
759 {
760 
761 	return (nh->nh_priv->rt_flags);
762 }
763 
764 void
765 nhop_set_rtflags(struct nhop_object *nh, int rt_flags)
766 {
767 
768 	nh->nh_priv->rt_flags = rt_flags;
769 }
770 
771 struct vnet *
772 nhop_get_vnet(const struct nhop_object *nh)
773 {
774 
775 	return (nh->nh_priv->nh_vnet);
776 }
777 
778 struct nhop_object *
779 nhop_select_func(struct nhop_object *nh, uint32_t flowid)
780 {
781 
782 	return (nhop_select(nh, flowid));
783 }
784 
785 void
786 nhops_update_ifmtu(struct rib_head *rh, struct ifnet *ifp, uint32_t mtu)
787 {
788 	struct nh_control *ctl;
789 	struct nhop_priv *nh_priv;
790 	struct nhop_object *nh;
791 
792 	ctl = rh->nh_control;
793 
794 	NHOPS_WLOCK(ctl);
795 	CHT_SLIST_FOREACH(&ctl->nh_head, nhops, nh_priv) {
796 		nh = nh_priv->nh;
797 		if (nh->nh_ifp == ifp) {
798 			if ((nh_priv->rt_flags & RTF_FIXEDMTU) == 0 ||
799 			    nh->nh_mtu > mtu) {
800 				/* Update MTU directly */
801 				nh->nh_mtu = mtu;
802 			}
803 		}
804 	} CHT_SLIST_FOREACH_END;
805 	NHOPS_WUNLOCK(ctl);
806 
807 }
808 
809 /*
810  * Dumps a single entry to sysctl buffer.
811  *
812  * Layout:
813  *  rt_msghdr - generic RTM header to allow users to skip non-understood messages
814  *  nhop_external - nexhop description structure (with length)
815  *  nhop_addrs - structure encapsulating GW/SRC sockaddrs
816  */
817 static int
818 dump_nhop_entry(struct rib_head *rh, struct nhop_object *nh, struct sysctl_req *w)
819 {
820 	struct {
821 		struct rt_msghdr	rtm;
822 		struct nhop_external	nhe;
823 		struct nhop_addrs	na;
824 	} arpc;
825 	struct nhop_external *pnhe;
826 	struct sockaddr *gw_sa, *src_sa;
827 	struct sockaddr_storage ss;
828 	size_t addrs_len;
829 	int error;
830 
831 	//DPRINTF("Dumping: head %p nh %p flags %X req %p\n", rh, nh, nh->nh_flags, w);
832 
833 	memset(&arpc, 0, sizeof(arpc));
834 
835 	arpc.rtm.rtm_msglen = sizeof(arpc);
836 	arpc.rtm.rtm_version = RTM_VERSION;
837 	arpc.rtm.rtm_type = RTM_GET;
838 	//arpc.rtm.rtm_flags = RTF_UP;
839 	arpc.rtm.rtm_flags = nh->nh_priv->rt_flags;
840 
841 	/* nhop_external */
842 	pnhe = &arpc.nhe;
843 	pnhe->nh_len = sizeof(struct nhop_external);
844 	pnhe->nh_idx = nh->nh_priv->nh_idx;
845 	pnhe->nh_fib = rh->rib_fibnum;
846 	pnhe->ifindex = nh->nh_ifp->if_index;
847 	pnhe->aifindex = nh->nh_aifp->if_index;
848 	pnhe->nh_family = nh->nh_priv->nh_family;
849 	pnhe->nh_type = nh->nh_priv->nh_type;
850 	pnhe->nh_mtu = nh->nh_mtu;
851 	pnhe->nh_flags = nh->nh_flags;
852 
853 	memcpy(pnhe->nh_prepend, nh->nh_prepend, sizeof(nh->nh_prepend));
854 	pnhe->prepend_len = nh->nh_prepend_len;
855 	pnhe->nh_refcount = nh->nh_priv->nh_refcnt;
856 	pnhe->nh_pksent = counter_u64_fetch(nh->nh_pksent);
857 
858 	/* sockaddr container */
859 	addrs_len = sizeof(struct nhop_addrs);
860 	arpc.na.gw_sa_off = addrs_len;
861 	gw_sa = (struct sockaddr *)&nh->gw4_sa;
862 	addrs_len += gw_sa->sa_len;
863 
864 	src_sa = nh->nh_ifa->ifa_addr;
865 	if (src_sa->sa_family == AF_LINK) {
866 		/* Shorten structure */
867 		memset(&ss, 0, sizeof(struct sockaddr_storage));
868 		fill_sdl_from_ifp((struct sockaddr_dl_short *)&ss,
869 		    nh->nh_ifa->ifa_ifp);
870 		src_sa = (struct sockaddr *)&ss;
871 	}
872 	arpc.na.src_sa_off = addrs_len;
873 	addrs_len += src_sa->sa_len;
874 
875 	/* Write total container length */
876 	arpc.na.na_len = addrs_len;
877 
878 	arpc.rtm.rtm_msglen += arpc.na.na_len - sizeof(struct nhop_addrs);
879 
880 	error = SYSCTL_OUT(w, &arpc, sizeof(arpc));
881 	if (error == 0)
882 		error = SYSCTL_OUT(w, gw_sa, gw_sa->sa_len);
883 	if (error == 0)
884 		error = SYSCTL_OUT(w, src_sa, src_sa->sa_len);
885 
886 	return (error);
887 }
888 
889 uint32_t
890 nhops_get_count(struct rib_head *rh)
891 {
892 	struct nh_control *ctl;
893 	uint32_t count;
894 
895 	ctl = rh->nh_control;
896 
897 	NHOPS_RLOCK(ctl);
898 	count = ctl->nh_head.items_count;
899 	NHOPS_RUNLOCK(ctl);
900 
901 	return (count);
902 }
903 
904 int
905 nhops_dump_sysctl(struct rib_head *rh, struct sysctl_req *w)
906 {
907 	struct nh_control *ctl;
908 	struct nhop_priv *nh_priv;
909 	int error;
910 
911 	ctl = rh->nh_control;
912 
913 	NHOPS_RLOCK(ctl);
914 	DPRINTF("NHDUMP: count=%u", ctl->nh_head.items_count);
915 	CHT_SLIST_FOREACH(&ctl->nh_head, nhops, nh_priv) {
916 		error = dump_nhop_entry(rh, nh_priv->nh, w);
917 		if (error != 0) {
918 			NHOPS_RUNLOCK(ctl);
919 			return (error);
920 		}
921 	} CHT_SLIST_FOREACH_END;
922 	NHOPS_RUNLOCK(ctl);
923 
924 	return (0);
925 }
926