Allow dissection of ICMP(V6) type and code. This re-uses transport layer port dissection code as although ICMP is not a transport protocol and their type and code are not ports this allows sharing of both code and storage.
Signed-off-by: Simon Horman <simon.hor...@netronome.com> --- drivers/net/bonding/bond_main.c | 6 ++++-- include/linux/skbuff.h | 5 +++++ include/net/flow_dissector.h | 30 +++++++++++++++++++++++++++--- net/core/flow_dissector.c | 34 +++++++++++++++++++++++++++++++--- net/sched/cls_flow.c | 4 ++-- 5 files changed, 69 insertions(+), 10 deletions(-) diff --git a/drivers/net/bonding/bond_main.c b/drivers/net/bonding/bond_main.c index 8029dd4912b6..a6f75cfb2bf7 100644 --- a/drivers/net/bonding/bond_main.c +++ b/drivers/net/bonding/bond_main.c @@ -3181,7 +3181,8 @@ static bool bond_flow_dissect(struct bonding *bond, struct sk_buff *skb, } else { return false; } - if (bond->params.xmit_policy == BOND_XMIT_POLICY_LAYER34 && proto >= 0) + if (bond->params.xmit_policy == BOND_XMIT_POLICY_LAYER34 && + proto >= 0 && !skb_flow_is_icmp_any(skb, proto)) fk->ports.ports = skb_flow_get_ports(skb, noff, proto); return true; @@ -3209,7 +3210,8 @@ u32 bond_xmit_hash(struct bonding *bond, struct sk_buff *skb) return bond_eth_hash(skb); if (bond->params.xmit_policy == BOND_XMIT_POLICY_LAYER23 || - bond->params.xmit_policy == BOND_XMIT_POLICY_ENCAP23) + bond->params.xmit_policy == BOND_XMIT_POLICY_ENCAP23 || + flow_keys_are_icmp_any(&flow)) hash = bond_eth_hash(skb); else hash = (__force u32)flow.ports.ports; diff --git a/include/linux/skbuff.h b/include/linux/skbuff.h index 9c535fbccf2c..44a8f69a9198 100644 --- a/include/linux/skbuff.h +++ b/include/linux/skbuff.h @@ -1094,6 +1094,11 @@ u32 __skb_get_poff(const struct sk_buff *skb, void *data, __be32 __skb_flow_get_ports(const struct sk_buff *skb, int thoff, u8 ip_proto, void *data, int hlen_proto); +static inline bool skb_flow_is_icmp_any(const struct sk_buff *skb, u8 ip_proto) +{ + return flow_protos_are_icmp_any(skb->protocol, ip_proto); +} + static inline __be32 skb_flow_get_ports(const struct sk_buff *skb, int thoff, u8 ip_proto) { diff --git a/include/net/flow_dissector.h b/include/net/flow_dissector.h index c4f31666afd2..8880025914e3 100644 --- a/include/net/flow_dissector.h +++ b/include/net/flow_dissector.h @@ -2,6 +2,7 @@ #define _NET_FLOW_DISSECTOR_H #include <linux/types.h> +#include <linux/in.h> #include <linux/in6.h> #include <uapi/linux/if_ether.h> @@ -89,10 +90,15 @@ struct flow_dissector_key_addrs { }; /** - * flow_dissector_key_tp_ports: - * @ports: port numbers of Transport header + * flow_dissector_key_ports: + * @ports: port numbers of Transport header or + * type and code of ICMP header + * ports: source (high) and destination (low) port numbers * src: source port number * dst: destination port number + * icmp: ICMP type (high) and code (low) + * type: ICMP type + * type: ICMP code */ struct flow_dissector_key_ports { union { @@ -101,6 +107,11 @@ struct flow_dissector_key_ports { __be16 src; __be16 dst; }; + __be16 icmp; + struct { + u8 type; + u8 code; + }; }; }; @@ -188,9 +199,22 @@ struct flow_keys_digest { void make_flow_keys_digest(struct flow_keys_digest *digest, const struct flow_keys *flow); +static inline bool flow_protos_are_icmp_any(__be16 n_proto, u8 ip_proto) +{ + return (n_proto == htons(ETH_P_IP) && ip_proto == IPPROTO_ICMP) || + (n_proto == htons(ETH_P_IPV6) && ip_proto == IPPROTO_ICMPV6); +} + +static inline bool flow_keys_are_icmp_any(const struct flow_keys *keys) +{ + return flow_protos_are_icmp_any(keys->basic.n_proto, + keys->basic.ip_proto); +} + static inline bool flow_keys_have_l4(const struct flow_keys *keys) { - return (keys->ports.ports || keys->tags.flow_label); + return (!flow_keys_are_icmp_any(keys) && keys->ports.ports) || + keys->tags.flow_label; } u32 flow_hash_from_keys(struct flow_keys *keys); diff --git a/net/core/flow_dissector.c b/net/core/flow_dissector.c index 1eb6f949e5b2..0584b4bb4390 100644 --- a/net/core/flow_dissector.c +++ b/net/core/flow_dissector.c @@ -58,6 +58,28 @@ void skb_flow_dissector_init(struct flow_dissector *flow_dissector, EXPORT_SYMBOL(skb_flow_dissector_init); /** + * skb_flow_get_be16 - extract be16 entity + * @skb: sk_buff to extract from + * @poff: offset to extract at + * @data: raw buffer pointer to the packet + * @hlen: packet header length + * + * The function will try to retrieve a be32 entity at + * offset poff + */ +__be16 skb_flow_get_be16(const struct sk_buff *skb, int poff, void *data, + int hlen) +{ + __be16 *u, _u; + + u = __skb_header_pointer(skb, poff, sizeof(_u), data, hlen, &_u); + if (u) + return *u; + + return 0; +} + +/** * __skb_flow_get_ports - extract the upper layer ports and return them * @skb: sk_buff to extract the ports from * @thoff: transport header offset @@ -542,8 +564,13 @@ bool __skb_flow_dissect(const struct sk_buff *skb, key_ports = skb_flow_dissector_target(flow_dissector, FLOW_DISSECTOR_KEY_PORTS, target_container); - key_ports->ports = __skb_flow_get_ports(skb, nhoff, ip_proto, - data, hlen); + if (flow_protos_are_icmp_any(proto, ip_proto)) + key_ports->icmp = skb_flow_get_be16(skb, nhoff, data, + hlen); + else + key_ports->ports = __skb_flow_get_ports(skb, nhoff, + ip_proto, data, + hlen); } out_good: @@ -718,7 +745,8 @@ void make_flow_keys_digest(struct flow_keys_digest *digest, data->n_proto = flow->basic.n_proto; data->ip_proto = flow->basic.ip_proto; - data->ports = flow->ports.ports; + if (flow_keys_have_l4(flow)) + data->ports = flow->ports.ports; data->src = flow->addrs.v4addrs.src; data->dst = flow->addrs.v4addrs.dst; } diff --git a/net/sched/cls_flow.c b/net/sched/cls_flow.c index e39672394c7b..a1a7ae71aa62 100644 --- a/net/sched/cls_flow.c +++ b/net/sched/cls_flow.c @@ -96,7 +96,7 @@ static u32 flow_get_proto(const struct sk_buff *skb, static u32 flow_get_proto_src(const struct sk_buff *skb, const struct flow_keys *flow) { - if (flow->ports.ports) + if (!flow_keys_are_icmp_any(flow) && flow->ports.ports) return ntohs(flow->ports.src); return addr_fold(skb->sk); @@ -105,7 +105,7 @@ static u32 flow_get_proto_src(const struct sk_buff *skb, static u32 flow_get_proto_dst(const struct sk_buff *skb, const struct flow_keys *flow) { - if (flow->ports.ports) + if (!flow_keys_are_icmp_any(flow) && flow->ports.ports) return ntohs(flow->ports.dst); return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb); -- 2.7.0.rc3.207.g0ac5344