net/sched: cls_flower: Support matching on SCTP ports

[linux.git] / net / sched / cls_flower.c

diff --git a/net/sched/cls_flower.c b/net/sched/cls_flower.c
index a8fb1ca03b3ea513a2822621db1e03aff92695b6..db4cd882a989c5196aede640d14ce009cb2c2cf6 100644 (file)
--- a/net/sched/cls_flower.c
+++ b/net/sched/cls_flower.c
@@ -344,6 +344,10 @@ static const struct nla_policy fl_policy[TCA_FLOWER_MAX + 1] = {
        [TCA_FLOWER_KEY_TCP_DST_MASK]   = { .type = NLA_U16 },
        [TCA_FLOWER_KEY_UDP_SRC_MASK]   = { .type = NLA_U16 },
        [TCA_FLOWER_KEY_UDP_DST_MASK]   = { .type = NLA_U16 },
+       [TCA_FLOWER_KEY_SCTP_SRC_MASK]  = { .type = NLA_U16 },
+       [TCA_FLOWER_KEY_SCTP_DST_MASK]  = { .type = NLA_U16 },
+       [TCA_FLOWER_KEY_SCTP_SRC]       = { .type = NLA_U16 },
+       [TCA_FLOWER_KEY_SCTP_DST]       = { .type = NLA_U16 },
 };
 
 static void fl_set_key_val(struct nlattr **tb,
@@ -453,6 +457,13 @@ static int fl_set_key(struct net *net, struct nlattr **tb,
                fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
                               &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
                               sizeof(key->tp.dst));
+       } else if (key->basic.ip_proto == IPPROTO_SCTP) {
+               fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
+                              &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
+                              sizeof(key->tp.src));
+               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));
        }
 
        if (tb[TCA_FLOWER_KEY_ENC_IPV4_SRC] ||
@@ -897,6 +908,14 @@ static int fl_dump(struct net *net, struct tcf_proto *tp, unsigned long fh,
                                  &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
                                  sizeof(key->tp.dst))))
                goto nla_put_failure;
+       else if (key->basic.ip_proto == IPPROTO_SCTP &&
+                (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
+                                 &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
+                                 sizeof(key->tp.src)) ||
+                 fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
+                                 &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
+                                 sizeof(key->tp.dst))))
+               goto nla_put_failure;
 
        if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
            (fl_dump_key_val(skb, &key->enc_ipv4.src,