skbuff: Make __skb_set_sw_hash a general function

[firefly-linux-kernel-4.4.55.git] / net / core / flow_dissector.c

diff --git a/net/core/flow_dissector.c b/net/core/flow_dissector.c
index 2a834c6179b9973e45274d793e7d744939e5f49e..151b6e48b81f1d99020e710672f8b8134f82dda7 100644 (file)
--- a/net/core/flow_dissector.c
+++ b/net/core/flow_dissector.c
@@ -609,10 +609,9 @@ void __skb_get_hash(struct sk_buff *skb)
        hash = ___skb_get_hash(skb, &keys, hashrnd);
        if (!hash)
                return;
-       if (keys.ports.ports)
-               skb->l4_hash = 1;
-       skb->sw_hash = 1;
-       skb->hash = hash;
+
+       __skb_set_sw_hash(skb, hash,
+                         flow_keys_have_l4(&keys));
 }
 EXPORT_SYMBOL(__skb_get_hash);
 
@@ -624,6 +623,51 @@ __u32 skb_get_hash_perturb(const struct sk_buff *skb, u32 perturb)
 }
 EXPORT_SYMBOL(skb_get_hash_perturb);
 
+__u32 __skb_get_hash_flowi6(struct sk_buff *skb, struct flowi6 *fl6)
+{
+       struct flow_keys keys;
+
+       memset(&keys, 0, sizeof(keys));
+
+       memcpy(&keys.addrs.v6addrs.src, &fl6->saddr,
+              sizeof(keys.addrs.v6addrs.src));
+       memcpy(&keys.addrs.v6addrs.dst, &fl6->daddr,
+              sizeof(keys.addrs.v6addrs.dst));
+       keys.control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
+       keys.ports.src = fl6->fl6_sport;
+       keys.ports.dst = fl6->fl6_dport;
+       keys.keyid.keyid = fl6->fl6_gre_key;
+       keys.tags.flow_label = (__force u32)fl6->flowlabel;
+       keys.basic.ip_proto = fl6->flowi6_proto;
+
+       __skb_set_sw_hash(skb, flow_hash_from_keys(&keys),
+                         flow_keys_have_l4(&keys));
+
+       return skb->hash;
+}
+EXPORT_SYMBOL(__skb_get_hash_flowi6);
+
+__u32 __skb_get_hash_flowi4(struct sk_buff *skb, struct flowi4 *fl4)
+{
+       struct flow_keys keys;
+
+       memset(&keys, 0, sizeof(keys));
+
+       keys.addrs.v4addrs.src = fl4->saddr;
+       keys.addrs.v4addrs.dst = fl4->daddr;
+       keys.control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
+       keys.ports.src = fl4->fl4_sport;
+       keys.ports.dst = fl4->fl4_dport;
+       keys.keyid.keyid = fl4->fl4_gre_key;
+       keys.basic.ip_proto = fl4->flowi4_proto;
+
+       __skb_set_sw_hash(skb, flow_hash_from_keys(&keys),
+                         flow_keys_have_l4(&keys));
+
+       return skb->hash;
+}
+EXPORT_SYMBOL(__skb_get_hash_flowi4);
+
 u32 __skb_get_poff(const struct sk_buff *skb, void *data,
                   const struct flow_keys *keys, int hlen)
 {