Wed, Dec 07, 2016 at 12:34:12PM CET, simon.hor...@netronome.com wrote:
>Support matching on ICMP type and code.
>
>Example usage:
>
>tc qdisc add dev eth0 ingress
>
>tc filter add dev eth0 protocol ip parent ffff: flower \
>       indev eth0 ip_proto icmp type 8 code 0 action drop
>
>tc filter add dev eth0 protocol ipv6 parent ffff: flower \
>       indev eth0 ip_proto icmpv6 type 128 code 0 action drop
>
>Signed-off-by: Simon Horman <simon.hor...@netronome.com>
>---
>v3
>* Use separate dissector key FLOW_DISSECTOR_KEY_ICMP
>* Use separate structure to hold icmp fields, this does not share
>  storage with ports
>
>v2
>* Move helpers to another patch
>---
> include/uapi/linux/pkt_cls.h | 10 +++++++++
> net/sched/cls_flower.c       | 49 ++++++++++++++++++++++++++++++++++++++++++++
> 2 files changed, 59 insertions(+)
>
>diff --git a/include/uapi/linux/pkt_cls.h b/include/uapi/linux/pkt_cls.h
>index 1adc0b654996..884c5aa515c4 100644
>--- a/include/uapi/linux/pkt_cls.h
>+++ b/include/uapi/linux/pkt_cls.h
>@@ -458,6 +458,16 @@ enum {
>       TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,   /* be16 */
>       TCA_FLOWER_KEY_ENC_UDP_DST_PORT,        /* be16 */
>       TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,   /* be16 */
>+
>+      TCA_FLOWER_KEY_ICMPV4_CODE,     /* u8 */
>+      TCA_FLOWER_KEY_ICMPV4_CODE_MASK,/* u8 */
>+      TCA_FLOWER_KEY_ICMPV4_TYPE,     /* u8 */
>+      TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,/* u8 */
>+      TCA_FLOWER_KEY_ICMPV6_CODE,     /* u8 */
>+      TCA_FLOWER_KEY_ICMPV6_CODE_MASK,/* u8 */
>+      TCA_FLOWER_KEY_ICMPV6_TYPE,     /* u8 */
>+      TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,/* u8 */
>+
>       __TCA_FLOWER_MAX,
> };
> 
>diff --git a/net/sched/cls_flower.c b/net/sched/cls_flower.c
>index 29a9e6d9f274..56df0368125a 100644
>--- a/net/sched/cls_flower.c
>+++ b/net/sched/cls_flower.c
>@@ -39,6 +39,7 @@ struct fl_flow_key {
>               struct flow_dissector_key_ipv6_addrs ipv6;
>       };
>       struct flow_dissector_key_ports tp;
>+      struct flow_dissector_key_icmp icmp;
>       struct flow_dissector_key_keyid enc_key_id;
>       union {
>               struct flow_dissector_key_ipv4_addrs enc_ipv4;
>@@ -386,6 +387,14 @@ static const struct nla_policy fl_policy[TCA_FLOWER_MAX + 
>1] = {
>       [TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK]  = { .type = NLA_U16 },
>       [TCA_FLOWER_KEY_ENC_UDP_DST_PORT]       = { .type = NLA_U16 },
>       [TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK]  = { .type = NLA_U16 },
>+      [TCA_FLOWER_KEY_ICMPV4_TYPE]    = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV4_TYPE_MASK] = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV4_CODE]    = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV4_CODE_MASK] = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV6_TYPE]    = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV6_TYPE_MASK] = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV6_CODE]    = { .type = NLA_U8 },
>+      [TCA_FLOWER_KEY_ICMPV6_CODE_MASK] = { .type = NLA_U8 },
> };
> 
> static void fl_set_key_val(struct nlattr **tb,
>@@ -502,6 +511,24 @@ static int fl_set_key(struct net *net, struct nlattr **tb,
>               fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
>                              &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
>                              sizeof(key->tp.dst));
>+      } else if (flow_basic_key_is_icmpv4(&key->basic)) {

Please just use:
key->basic.n_proto and key->basic.ip_proto
like the rest of the code. And drop the helpers.

The rest looks fine to me.


>+              fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV4_TYPE,
>+                             &mask->icmp.type,
>+                             TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
>+                             sizeof(key->icmp.type));
>+              fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
>+                             &mask->icmp.code,
>+                             TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
>+                             sizeof(key->icmp.code));
>+      } else if (flow_basic_key_is_icmpv6(&key->basic)) {
>+              fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV6_TYPE,
>+                             &mask->icmp.type,
>+                             TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
>+                             sizeof(key->icmp.type));
>+              fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
>+                             &mask->icmp.code,
>+                             TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
>+                             sizeof(key->icmp.code));
>       }
> 
>       if (tb[TCA_FLOWER_KEY_ENC_IPV4_SRC] ||
>@@ -612,6 +639,8 @@ static void fl_init_dissector(struct cls_fl_head *head,
>       FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
>                            FLOW_DISSECTOR_KEY_PORTS, tp);
>       FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
>+                           FLOW_DISSECTOR_KEY_ICMP, icmp);
>+      FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
>                            FLOW_DISSECTOR_KEY_VLAN, vlan);
>       FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
>                            FLOW_DISSECTOR_KEY_ENC_KEYID, enc_key_id);
>@@ -977,6 +1006,26 @@ static int fl_dump(struct net *net, struct tcf_proto 
>*tp, unsigned long fh,
>                                 &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
>                                 sizeof(key->tp.dst))))
>               goto nla_put_failure;
>+      else if (flow_basic_key_is_icmpv4(&key->basic) &&
>+               (fl_dump_key_val(skb, &key->icmp.type,
>+                                TCA_FLOWER_KEY_ICMPV4_TYPE, &mask->icmp.type,
>+                                TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
>+                                sizeof(key->icmp.type)) ||
>+                fl_dump_key_val(skb, &key->icmp.code,
>+                                TCA_FLOWER_KEY_ICMPV4_CODE, &mask->icmp.code,
>+                                TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
>+                                sizeof(key->icmp.code))))
>+              goto nla_put_failure;
>+      else if (flow_basic_key_is_icmpv6(&key->basic) &&
>+               (fl_dump_key_val(skb, &key->icmp.type,
>+                                TCA_FLOWER_KEY_ICMPV6_TYPE, &mask->icmp.type,
>+                                TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
>+                                sizeof(key->icmp.type)) ||
>+                fl_dump_key_val(skb, &key->icmp.code,
>+                                TCA_FLOWER_KEY_ICMPV6_CODE, &mask->icmp.code,
>+                                TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
>+                                sizeof(key->icmp.code))))
>+              goto nla_put_failure;
> 
>       if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
>           (fl_dump_key_val(skb, &key->enc_ipv4.src,
>-- 
>2.7.0.rc3.207.g0ac5344
>

Reply via email to