Author: melifaro
Date: Thu Jan 14 13:14:12 2016
New Revision: 293914
URL: https://svnweb.freebsd.org/changeset/base/293914

Log:
  Make ng_netflow(9) use new routing KPI.
  
  Netflow module is supposed to store (along with fields like
    gateway address and interface index) matched netmask for each record.
    This (currently) requires returning individual route entries, instead
    of optimized next-hop structure. Given that, use control-plane
    rib_lookup_info() function to avoid accessing rtentries directly.
  While rib_lookup_info() might be slower, than fibX_lookup() flavours,
    it is more scalable than rtalloc1_fib(), because rtentry mutex is
    not acquired.

Modified:
  head/sys/netgraph/netflow/netflow.c

Modified: head/sys/netgraph/netflow/netflow.c
==============================================================================
--- head/sys/netgraph/netflow/netflow.c Thu Jan 14 11:25:28 2016        
(r293913)
+++ head/sys/netgraph/netflow/netflow.c Thu Jan 14 13:14:12 2016        
(r293914)
@@ -44,6 +44,7 @@ __FBSDID("$FreeBSD$");
 #include <sys/socket.h>
 
 #include <net/if.h>
+#include <net/if_dl.h>
 #include <net/if_var.h>
 #include <net/route.h>
 #include <net/ethernet.h>
@@ -307,8 +308,9 @@ hash_insert(priv_p priv, struct flow_has
        int plen, uint8_t flags, uint8_t tcp_flags)
 {
        struct flow_entry *fle;
-       struct sockaddr_in sin;
-       struct rtentry *rt;
+       struct sockaddr_in sin, sin_mask;
+       struct sockaddr_dl rt_gateway;
+       struct rt_addrinfo info;
 
        mtx_assert(&hsh->mtx, MA_OWNED);
 
@@ -339,23 +341,30 @@ hash_insert(priv_p priv, struct flow_has
                sin.sin_len = sizeof(struct sockaddr_in);
                sin.sin_family = AF_INET;
                sin.sin_addr = fle->f.r.r_dst;
-               rt = rtalloc1_fib((struct sockaddr *)&sin, 0, 0, r->fib);
-               if (rt != NULL) {
-                       fle->f.fle_o_ifx = rt->rt_ifp->if_index;
 
-                       if (rt->rt_flags & RTF_GATEWAY &&
-                           rt->rt_gateway->sa_family == AF_INET)
+               rt_gateway.sdl_len = sizeof(rt_gateway);
+               sin_mask.sin_len = sizeof(struct sockaddr_in);
+               bzero(&info, sizeof(info));
+
+               info.rti_info[RTAX_GATEWAY] = (struct sockaddr *)&rt_gateway;
+               info.rti_info[RTAX_NETMASK] = (struct sockaddr *)&sin_mask;
+
+               if (rib_lookup_info(r->fib, (struct sockaddr *)&sin, NHR_REF, 0,
+                   &info) == 0) {
+                       fle->f.fle_o_ifx = info.rti_ifp->if_index;
+
+                       if (info.rti_flags & RTF_GATEWAY &&
+                           rt_gateway.sdl_family == AF_INET)
                                fle->f.next_hop =
-                                   ((struct sockaddr_in 
*)(rt->rt_gateway))->sin_addr;
+                                   ((struct sockaddr_in 
*)&rt_gateway)->sin_addr;
 
-                       if (rt_mask(rt))
-                               fle->f.dst_mask =
-                                   bitcount32(((struct sockaddr_in 
*)rt_mask(rt))->sin_addr.s_addr);
-                       else if (rt->rt_flags & RTF_HOST)
+                       if (info.rti_addrs & RTA_NETMASK)
+                               fle->f.dst_mask = 
bitcount32(sin_mask.sin_addr.s_addr);
+                       else if (info.rti_flags & RTF_HOST)
                                /* Give up. We can't determine mask :( */
                                fle->f.dst_mask = 32;
 
-                       RTFREE_LOCKED(rt);
+                       rib_free_info(&info);
                }
        }
 
@@ -365,16 +374,20 @@ hash_insert(priv_p priv, struct flow_has
                sin.sin_len = sizeof(struct sockaddr_in);
                sin.sin_family = AF_INET;
                sin.sin_addr = fle->f.r.r_src;
-               rt = rtalloc1_fib((struct sockaddr *)&sin, 0, 0, r->fib);
-               if (rt != NULL) {
-                       if (rt_mask(rt))
+
+               sin_mask.sin_len = sizeof(struct sockaddr_in);
+               bzero(&info, sizeof(info));
+
+               info.rti_info[RTAX_NETMASK] = (struct sockaddr *)&sin_mask;
+
+               if (rib_lookup_info(r->fib, (struct sockaddr *)&sin, 0, 0,
+                   &info) == 0) {
+                       if (info.rti_addrs & RTA_NETMASK)
                                fle->f.src_mask =
-                                   bitcount32(((struct sockaddr_in 
*)rt_mask(rt))->sin_addr.s_addr);
-                       else if (rt->rt_flags & RTF_HOST)
+                                   bitcount32(sin_mask.sin_addr.s_addr);
+                       else if (info.rti_flags & RTF_HOST)
                                /* Give up. We can't determine mask :( */
                                fle->f.src_mask = 32;
-
-                       RTFREE_LOCKED(rt);
                }
        }
 
@@ -390,14 +403,14 @@ hash_insert(priv_p priv, struct flow_has
                                bitcount32((x).__u6_addr.__u6_addr32[1]) + \
                                bitcount32((x).__u6_addr.__u6_addr32[2]) + \
                                bitcount32((x).__u6_addr.__u6_addr32[3])
-#define RT_MASK6(x)    (ipv6_masklen(((struct sockaddr_in6 
*)rt_mask(x))->sin6_addr))
 static int
 hash6_insert(priv_p priv, struct flow_hash_entry *hsh6, struct flow6_rec *r,
        int plen, uint8_t flags, uint8_t tcp_flags)
 {
        struct flow6_entry *fle6;
-       struct sockaddr_in6 sin6;
-       struct rtentry *rt;
+       struct sockaddr_in6 sin6, sin6_mask;
+       struct sockaddr_dl rt_gateway;
+       struct rt_addrinfo info;
 
        mtx_assert(&hsh6->mtx, MA_OWNED);
 
@@ -430,22 +443,29 @@ hash6_insert(priv_p priv, struct flow_ha
                sin6.sin6_family = AF_INET6;
                sin6.sin6_addr = r->dst.r_dst6;
 
-               rt = rtalloc1_fib((struct sockaddr *)&sin6, 0, 0, r->fib);
-
-               if (rt != NULL) {
-                       fle6->f.fle_o_ifx = rt->rt_ifp->if_index;
+               rt_gateway.sdl_len = sizeof(rt_gateway);
+               sin6_mask.sin6_len = sizeof(struct sockaddr_in6);
+               bzero(&info, sizeof(info));
+
+               info.rti_info[RTAX_GATEWAY] = (struct sockaddr *)&rt_gateway;
+               info.rti_info[RTAX_NETMASK] = (struct sockaddr *)&sin6_mask;
+
+               if (rib_lookup_info(r->fib, (struct sockaddr *)&sin6, NHR_REF,
+                   0, &info) == 0) {
+                       fle6->f.fle_o_ifx = info.rti_ifp->if_index;
 
-                       if (rt->rt_flags & RTF_GATEWAY &&
-                           rt->rt_gateway->sa_family == AF_INET6)
+                       if (info.rti_flags & RTF_GATEWAY &&
+                           rt_gateway.sdl_family == AF_INET6)
                                fle6->f.n.next_hop6 =
-                                   ((struct sockaddr_in6 
*)(rt->rt_gateway))->sin6_addr;
+                                   ((struct sockaddr_in6 
*)&rt_gateway)->sin6_addr;
 
-                       if (rt_mask(rt))
-                               fle6->f.dst_mask = RT_MASK6(rt);
+                       if (info.rti_addrs & RTA_NETMASK)
+                               fle6->f.dst_mask =
+                                   ipv6_masklen(sin6_mask.sin6_addr);
                        else
                                fle6->f.dst_mask = 128;
 
-                       RTFREE_LOCKED(rt);
+                       rib_free_info(&info);
                }
        }
 
@@ -456,15 +476,18 @@ hash6_insert(priv_p priv, struct flow_ha
                sin6.sin6_family = AF_INET6;
                sin6.sin6_addr = r->src.r_src6;
 
-               rt = rtalloc1_fib((struct sockaddr *)&sin6, 0, 0, r->fib);
+               sin6_mask.sin6_len = sizeof(struct sockaddr_in6);
+               bzero(&info, sizeof(info));
 
-               if (rt != NULL) {
-                       if (rt_mask(rt))
-                               fle6->f.src_mask = RT_MASK6(rt);
+               info.rti_info[RTAX_NETMASK] = (struct sockaddr *)&sin6_mask;
+
+               if (rib_lookup_info(r->fib, (struct sockaddr *)&sin6, 0, 0,
+                   &info) == 0) {
+                       if (info.rti_addrs & RTA_NETMASK)
+                               fle6->f.src_mask =
+                                   ipv6_masklen(sin6_mask.sin6_addr);
                        else
                                fle6->f.src_mask = 128;
-
-                       RTFREE_LOCKED(rt);
                }
        }
 
@@ -474,7 +497,6 @@ hash6_insert(priv_p priv, struct flow_ha
        return (0);
 }
 #undef ipv6_masklen
-#undef RT_MASK6
 #endif
 
 
_______________________________________________
svn-src-all@freebsd.org mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-all
To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"

Reply via email to