From: Peter Nørlund <p...@ordbogen.com>

This patch adds L3 and L4 hash-based multipath routing, selectable on a
per-route basis with the reintroduced RTA_MP_ALGO attribute. The default is
now RT_MP_ALG_L3_HASH.

Signed-off-by: Peter Nørlund <p...@ordbogen.com>
---
 include/net/ip_fib.h           | 22 ++++++++++++++++-
 include/uapi/linux/rtnetlink.h | 14 ++++++++++-
 net/ipv4/fib_frontend.c        |  4 +++
 net/ipv4/fib_semantics.c       | 43 +++++++++++++++++++++++++++-----
 net/ipv4/route.c               | 56 ++++++++++++++++++++++++++++++++++++++++--
 5 files changed, 129 insertions(+), 10 deletions(-)

diff --git a/include/net/ip_fib.h b/include/net/ip_fib.h
index 18a3c7f..21e74b5 100644
--- a/include/net/ip_fib.h
+++ b/include/net/ip_fib.h
@@ -37,6 +37,7 @@ struct fib_config {
        u32                     fc_flags;
        u32                     fc_priority;
        __be32                  fc_prefsrc;
+       int                     fc_mp_alg;
        struct nlattr           *fc_mx;
        struct rtnexthop        *fc_mp;
        int                     fc_mx_len;
@@ -119,6 +120,7 @@ struct fib_info {
        int                     fib_nhs;
 #ifdef CONFIG_IP_ROUTE_MULTIPATH
        int                     fib_weight;
+       int                     fib_mp_alg;
 #endif
        struct rcu_head         rcu;
        struct fib_nh           fib_nh[0];
@@ -312,7 +314,25 @@ int ip_fib_check_default(__be32 gw, struct net_device 
*dev);
 int fib_sync_down_dev(struct net_device *dev, unsigned long event);
 int fib_sync_down_addr(struct net *net, __be32 local);
 int fib_sync_up(struct net_device *dev, unsigned int nh_flags);
-void fib_select_multipath(struct fib_result *res);
+
+struct multipath_flow4 {
+       __be32 saddr;
+       __be32 daddr;
+       union {
+               __be32 ports;
+               struct {
+                       __be16 sport;
+                       __be16 dport;
+               };
+       };
+};
+
+typedef void (*multipath_flow4_func_t)(struct multipath_flow4 *flow,
+                                      void *ctx);
+
+void fib_select_multipath(struct fib_result *res,
+                         multipath_flow4_func_t flow_func,
+                         void *ctx);
 
 /* Exported by fib_trie.c */
 void fib_trie_init(void);
diff --git a/include/uapi/linux/rtnetlink.h b/include/uapi/linux/rtnetlink.h
index 0d3d3cc..2563a96 100644
--- a/include/uapi/linux/rtnetlink.h
+++ b/include/uapi/linux/rtnetlink.h
@@ -271,6 +271,18 @@ enum rt_scope_t {
 #define RTM_F_EQUALIZE         0x400   /* Multipath equalizer: NI      */
 #define RTM_F_PREFIX           0x800   /* Prefix addresses             */
 
+/* Multipath algorithms */
+
+enum rt_mp_alg_t {
+       RT_MP_ALG_L3_HASH,      /* Was IP_MP_ALG_NONE */
+       RT_MP_ALG_PER_PACKET,   /* Was IP_MP_ALG_RR */
+       RT_MP_ALG_DRR,          /* not used */
+       RT_MP_ALG_RANDOM,       /* not used */
+       RT_MP_ALG_WRANDOM,      /* not used */
+       RT_MP_ALG_L4_HASH,
+       __RT_MP_ALG_MAX
+};
+
 /* Reserved table identifiers */
 
 enum rt_class_t {
@@ -301,7 +313,7 @@ enum rtattr_type_t {
        RTA_FLOW,
        RTA_CACHEINFO,
        RTA_SESSION, /* no longer used */
-       RTA_MP_ALGO, /* no longer used */
+       RTA_MP_ALGO,
        RTA_TABLE,
        RTA_MARK,
        RTA_MFC_STATS,
diff --git a/net/ipv4/fib_frontend.c b/net/ipv4/fib_frontend.c
index 7fa2771..5ba4442 100644
--- a/net/ipv4/fib_frontend.c
+++ b/net/ipv4/fib_frontend.c
@@ -622,6 +622,7 @@ const struct nla_policy rtm_ipv4_policy[RTA_MAX + 1] = {
        [RTA_PREFSRC]           = { .type = NLA_U32 },
        [RTA_METRICS]           = { .type = NLA_NESTED },
        [RTA_MULTIPATH]         = { .len = sizeof(struct rtnexthop) },
+       [RTA_MP_ALGO]           = { .type = NLA_U32 },
        [RTA_FLOW]              = { .type = NLA_U32 },
        [RTA_ENCAP_TYPE]        = { .type = NLA_U16 },
        [RTA_ENCAP]             = { .type = NLA_NESTED },
@@ -684,6 +685,9 @@ static int rtm_to_fib_config(struct net *net, struct 
sk_buff *skb,
                        cfg->fc_mp = nla_data(attr);
                        cfg->fc_mp_len = nla_len(attr);
                        break;
+               case RTA_MP_ALGO:
+                       cfg->fc_mp_alg = nla_get_u32(attr);
+                       break;
                case RTA_FLOW:
                        cfg->fc_flow = nla_get_u32(attr);
                        break;
diff --git a/net/ipv4/fib_semantics.c b/net/ipv4/fib_semantics.c
index becb63f..3a80b1a 100644
--- a/net/ipv4/fib_semantics.c
+++ b/net/ipv4/fib_semantics.c
@@ -259,6 +259,11 @@ static inline int nh_comp(const struct fib_info *fi, const 
struct fib_info *ofi)
 {
        const struct fib_nh *onh = ofi->fib_nh;
 
+#ifdef CONFIG_IP_ROUTE_MULTIPATH
+       if (fi->fib_mp_alg != ofi->fib_mp_alg)
+               return -1;
+#endif
+
        for_nexthops(fi) {
                if (nh->nh_oif != onh->nh_oif ||
                    nh->nh_gw  != onh->nh_gw ||
@@ -1028,6 +1033,7 @@ struct fib_info *fib_create_info(struct fib_config *cfg)
 
        if (cfg->fc_mp) {
 #ifdef CONFIG_IP_ROUTE_MULTIPATH
+               fi->fib_mp_alg = cfg->fc_mp_alg;
                err = fib_get_nhs(fi, cfg->fc_mp, cfg->fc_mp_len, cfg);
                if (err != 0)
                        goto failure;
@@ -1245,6 +1251,10 @@ int fib_dump_info(struct sk_buff *skb, u32 portid, u32 
seq, int event,
                struct rtnexthop *rtnh;
                struct nlattr *mp;
 
+               if (fi->fib_mp_alg &&
+                   nla_put_u32(skb, RTA_MP_ALGO, fi->fib_mp_alg))
+                       goto nla_put_failure;
+
                mp = nla_nest_start(skb, RTA_MULTIPATH);
                if (!mp)
                        goto nla_put_failure;
@@ -1520,16 +1530,37 @@ int fib_sync_up(struct net_device *dev, unsigned int 
nh_flags)
 
 #ifdef CONFIG_IP_ROUTE_MULTIPATH
 
-/*
- * The algorithm is suboptimal, but it provides really
- * fair weighted route distribution.
- */
-void fib_select_multipath(struct fib_result *res)
+/* Compute multipath hash based on 3- or 5-tuple */
+static int fib_multipath_hash(const struct fib_result *res,
+                             multipath_flow4_func_t flow_func, void *ctx)
+{
+       struct multipath_flow4 flow;
+
+       flow_func(&flow, ctx);
+
+       if (res->fi->fib_mp_alg == RT_MP_ALG_L4_HASH)
+               return jhash_3words(flow.saddr, flow.daddr, flow.ports, 0) >> 1;
+       else
+               return jhash_2words(flow.saddr, flow.daddr, 0) >> 1;
+}
+
+static int fib_multipath_perpacket(void)
+{
+       return bitrev32(this_cpu_inc_return(fib_multipath_rr_counter)) >> 1;
+}
+
+void fib_select_multipath(struct fib_result *res,
+                         multipath_flow4_func_t flow_func,
+                         void *ctx)
 {
        struct fib_info *fi = res->fi;
        int h;
 
-       h = bitrev32(this_cpu_inc_return(fib_multipath_rr_counter)) >> 1;
+       if (res->fi->fib_mp_alg == RT_MP_ALG_PER_PACKET) {
+               h = fib_multipath_perpacket();
+       } else {
+               h = fib_multipath_hash(res, flow_func, ctx);
+       }
 
        for_nexthops(fi) {
                if (h > atomic_read(&nh->nh_upper_bound))
diff --git a/net/ipv4/route.c b/net/ipv4/route.c
index f3087aa..f50f84f 100644
--- a/net/ipv4/route.c
+++ b/net/ipv4/route.c
@@ -1643,6 +1643,58 @@ out:
        return err;
 }
 
+#ifdef CONFIG_IP_ROUTE_MULTIPATH
+
+/* Fill multipath flow key data based on socket buffer */
+static void ip_multipath_flow_skb(struct multipath_flow4 *flow, void *ctx)
+{
+       const struct sk_buff *skb = (const struct sk_buff *)ctx;
+       const struct iphdr *iph;
+
+       iph = ip_hdr(skb);
+
+       flow->saddr = iph->saddr;
+       flow->daddr = iph->daddr;
+       flow->ports = 0;
+
+       if (unlikely(!(iph->frag_off & htons(IP_DF))))
+               return;
+
+       if (iph->protocol == IPPROTO_TCP ||
+           iph->protocol == IPPROTO_UDP ||
+           iph->protocol == IPPROTO_SCTP) {
+               __be16 _ports[2];
+               const __be16 *ports;
+
+               ports = skb_header_pointer(skb, iph->ihl * 4, sizeof(_ports),
+                                          &_ports);
+               if (ports) {
+                       flow->sport = ports[0];
+                       flow->dport = ports[1];
+               }
+       }
+}
+
+/* Fill multipath flow key data based on flowi4  */
+static void ip_multipath_flow_fl4(struct multipath_flow4 *flow, void *ctx)
+{
+       const struct flowi4 *fl4 = (const struct flowi4 *)ctx;
+
+       flow->saddr = fl4->saddr;
+       flow->daddr = fl4->daddr;
+
+       if (fl4->flowi4_proto == IPPROTO_TCP ||
+           fl4->flowi4_proto == IPPROTO_UDP ||
+           fl4->flowi4_proto == IPPROTO_SCTP) {
+               flow->sport = fl4->fl4_sport;
+               flow->dport = fl4->fl4_dport;
+       } else {
+               flow->ports = 0;
+       }
+}
+
+#endif /* CONFIG_IP_ROUTE_MULTIPATH */
+
 static int ip_mkroute_input(struct sk_buff *skb,
                            struct fib_result *res,
                            const struct flowi4 *fl4,
@@ -1651,7 +1703,7 @@ static int ip_mkroute_input(struct sk_buff *skb,
 {
 #ifdef CONFIG_IP_ROUTE_MULTIPATH
        if (res->fi && res->fi->fib_nhs > 1)
-               fib_select_multipath(res);
+               fib_select_multipath(res, ip_multipath_flow_skb, skb);
 #endif
 
        /* create a routing cache entry */
@@ -2197,7 +2249,7 @@ struct rtable *__ip_route_output_key(struct net *net, 
struct flowi4 *fl4)
 
 #ifdef CONFIG_IP_ROUTE_MULTIPATH
        if (res.fi->fib_nhs > 1 && fl4->flowi4_oif == 0)
-               fib_select_multipath(&res);
+               fib_select_multipath(&res, ip_multipath_flow_fl4, fl4);
        else
 #endif
        if (!res.prefixlen &&
-- 
2.1.4

--
To unsubscribe from this list: send the line "unsubscribe netdev" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to