net/sched/cls_flower.c

   1 /*
   2  * net/sched/cls_flower.c               Flower classifier
   3  *
   4  * Copyright (c) 2015 Jiri Pirko <jiri@resnulli.us>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  */
  11
  12 #include <linux/kernel.h>
  13 #include <linux/init.h>
  14 #include <linux/module.h>
  15 #include <linux/rhashtable.h>
  16 #include <linux/workqueue.h>
  17
  18 #include <linux/if_ether.h>
  19 #include <linux/in6.h>
  20 #include <linux/ip.h>
  21 #include <linux/mpls.h>
  22
  23 #include <net/sch_generic.h>
  24 #include <net/pkt_cls.h>
  25 #include <net/ip.h>
  26 #include <net/flow_dissector.h>
  27
  28 #include <net/dst.h>
  29 #include <net/dst_metadata.h>
  30
  31 struct fl_flow_key {
  32         int     indev_ifindex;
  33         struct flow_dissector_key_control control;
  34         struct flow_dissector_key_control enc_control;
  35         struct flow_dissector_key_basic basic;
  36         struct flow_dissector_key_eth_addrs eth;
  37         struct flow_dissector_key_vlan vlan;
  38         union {
  39                 struct flow_dissector_key_ipv4_addrs ipv4;
  40                 struct flow_dissector_key_ipv6_addrs ipv6;
  41         };
  42         struct flow_dissector_key_ports tp;
  43         struct flow_dissector_key_icmp icmp;
  44         struct flow_dissector_key_arp arp;
  45         struct flow_dissector_key_keyid enc_key_id;
  46         union {
  47                 struct flow_dissector_key_ipv4_addrs enc_ipv4;
  48                 struct flow_dissector_key_ipv6_addrs enc_ipv6;
  49         };
  50         struct flow_dissector_key_ports enc_tp;
  51         struct flow_dissector_key_mpls mpls;
  52         struct flow_dissector_key_tcp tcp;
  53         struct flow_dissector_key_ip ip;
  54 } __aligned(BITS_PER_LONG / 8); /* Ensure that we can do comparisons as longs. */
  55
  56 struct fl_flow_mask_range {
  57         unsigned short int start;
  58         unsigned short int end;
  59 };
  60
  61 struct fl_flow_mask {
  62         struct fl_flow_key key;
  63         struct fl_flow_mask_range range;
  64         struct rcu_head rcu;
  65 };
  66
  67 struct cls_fl_head {
  68         struct rhashtable ht;
  69         struct fl_flow_mask mask;
  70         struct flow_dissector dissector;
  71         bool mask_assigned;
  72         struct list_head filters;
  73         struct rhashtable_params ht_params;
  74         union {
  75                 struct work_struct work;
  76                 struct rcu_head rcu;
  77         };
  78         struct idr handle_idr;
  79 };
  80
  81 struct cls_fl_filter {
  82         struct rhash_head ht_node;
  83         struct fl_flow_key mkey;
  84         struct tcf_exts exts;
  85         struct tcf_result res;
  86         struct fl_flow_key key;
  87         struct list_head list;
  88         u32 handle;
  89         u32 flags;
  90         struct rcu_head rcu;
  91         struct net_device *hw_dev;
  92 };
  93
  94 static unsigned short int fl_mask_range(const struct fl_flow_mask *mask)
  95 {
  96         return mask->range.end - mask->range.start;
  97 }
  98
  99 static void fl_mask_update_range(struct fl_flow_mask *mask)
 100 {
 101         const u8 *bytes = (const u8 *) &mask->key;
 102         size_t size = sizeof(mask->key);
 103         size_t i, first = 0, last = size - 1;
 104
 105         for (i = 0; i < sizeof(mask->key); i++) {
 106                 if (bytes[i]) {
 107                         if (!first && i)
 108                                 first = i;
 109                         last = i;
 110                 }
 111         }
 112         mask->range.start = rounddown(first, sizeof(long));
 113         mask->range.end = roundup(last + 1, sizeof(long));
 114 }
 115
 116 static void *fl_key_get_start(struct fl_flow_key *key,
 117                               const struct fl_flow_mask *mask)
 118 {
 119         return (u8 *) key + mask->range.start;
 120 }
 121
 122 static void fl_set_masked_key(struct fl_flow_key *mkey, struct fl_flow_key *key,
 123                               struct fl_flow_mask *mask)
 124 {
 125         const long *lkey = fl_key_get_start(key, mask);
 126         const long *lmask = fl_key_get_start(&mask->key, mask);
 127         long *lmkey = fl_key_get_start(mkey, mask);
 128         int i;
 129
 130         for (i = 0; i < fl_mask_range(mask); i += sizeof(long))
 131                 *lmkey++ = *lkey++ & *lmask++;
 132 }
 133
 134 static void fl_clear_masked_range(struct fl_flow_key *key,
 135                                   struct fl_flow_mask *mask)
 136 {
 137         memset(fl_key_get_start(key, mask), 0, fl_mask_range(mask));
 138 }
 139
 140 static struct cls_fl_filter *fl_lookup(struct cls_fl_head *head,
 141                                        struct fl_flow_key *mkey)
 142 {
 143         return rhashtable_lookup_fast(&head->ht,
 144                                       fl_key_get_start(mkey, &head->mask),
 145                                       head->ht_params);
 146 }
 147
 148 static int fl_classify(struct sk_buff *skb, const struct tcf_proto *tp,
 149                        struct tcf_result *res)
 150 {
 151         struct cls_fl_head *head = rcu_dereference_bh(tp->root);
 152         struct cls_fl_filter *f;
 153         struct fl_flow_key skb_key;
 154         struct fl_flow_key skb_mkey;
 155         struct ip_tunnel_info *info;
 156
 157         if (!atomic_read(&head->ht.nelems))
 158                 return -1;
 159
 160         fl_clear_masked_range(&skb_key, &head->mask);
 161
 162         info = skb_tunnel_info(skb);
 163         if (info) {
 164                 struct ip_tunnel_key *key = &info->key;
 165
 166                 switch (ip_tunnel_info_af(info)) {
 167                 case AF_INET:
 168                         skb_key.enc_control.addr_type =
 169                                 FLOW_DISSECTOR_KEY_IPV4_ADDRS;
 170                         skb_key.enc_ipv4.src = key->u.ipv4.src;
 171                         skb_key.enc_ipv4.dst = key->u.ipv4.dst;
 172                         break;
 173                 case AF_INET6:
 174                         skb_key.enc_control.addr_type =
 175                                 FLOW_DISSECTOR_KEY_IPV6_ADDRS;
 176                         skb_key.enc_ipv6.src = key->u.ipv6.src;
 177                         skb_key.enc_ipv6.dst = key->u.ipv6.dst;
 178                         break;
 179                 }
 180
 181                 skb_key.enc_key_id.keyid = tunnel_id_to_key32(key->tun_id);
 182                 skb_key.enc_tp.src = key->tp_src;
 183                 skb_key.enc_tp.dst = key->tp_dst;
 184         }
 185
 186         skb_key.indev_ifindex = skb->skb_iif;
 187         /* skb_flow_dissect() does not set n_proto in case an unknown protocol,
 188          * so do it rather here.
 189          */
 190         skb_key.basic.n_proto = skb->protocol;
 191         skb_flow_dissect(skb, &head->dissector, &skb_key, 0);
 192
 193         fl_set_masked_key(&skb_mkey, &skb_key, &head->mask);
 194
 195         f = fl_lookup(head, &skb_mkey);
 196         if (f && !tc_skip_sw(f->flags)) {
 197                 *res = f->res;
 198                 return tcf_exts_exec(skb, &f->exts, res);
 199         }
 200         return -1;
 201 }
 202
 203 static int fl_init(struct tcf_proto *tp)
 204 {
 205         struct cls_fl_head *head;
 206
 207         head = kzalloc(sizeof(*head), GFP_KERNEL);
 208         if (!head)
 209                 return -ENOBUFS;
 210
 211         INIT_LIST_HEAD_RCU(&head->filters);
 212         rcu_assign_pointer(tp->root, head);
 213         idr_init(&head->handle_idr);
 214
 215         return 0;
 216 }
 217
 218 static void fl_destroy_filter(struct rcu_head *head)
 219 {
 220         struct cls_fl_filter *f = container_of(head, struct cls_fl_filter, rcu);
 221
 222         tcf_exts_destroy(&f->exts);
 223         kfree(f);
 224 }
 225
 226 static void fl_hw_destroy_filter(struct tcf_proto *tp, struct cls_fl_filter *f)
 227 {
 228         struct tc_cls_flower_offload cls_flower = {};
 229         struct net_device *dev = f->hw_dev;
 230
 231         if (!tc_can_offload(dev))
 232                 return;
 233
 234         tc_cls_common_offload_init(&cls_flower.common, tp);
 235         cls_flower.command = TC_CLSFLOWER_DESTROY;
 236         cls_flower.cookie = (unsigned long) f;
 237         cls_flower.egress_dev = f->hw_dev != tp->q->dev_queue->dev;
 238
 239         dev->netdev_ops->ndo_setup_tc(dev, TC_SETUP_CLSFLOWER, &cls_flower);
 240 }
 241
 242 static int fl_hw_replace_filter(struct tcf_proto *tp,
 243                                 struct flow_dissector *dissector,
 244                                 struct fl_flow_key *mask,
 245                                 struct cls_fl_filter *f)
 246 {
 247         struct net_device *dev = tp->q->dev_queue->dev;
 248         struct tc_cls_flower_offload cls_flower = {};
 249         int err;
 250
 251         if (!tc_can_offload(dev)) {
 252                 if (tcf_exts_get_dev(dev, &f->exts, &f->hw_dev) ||
 253                     (f->hw_dev && !tc_can_offload(f->hw_dev))) {
 254                         f->hw_dev = dev;
 255                         return tc_skip_sw(f->flags) ? -EINVAL : 0;
 256                 }
 257                 dev = f->hw_dev;
 258                 cls_flower.egress_dev = true;
 259         } else {
 260                 f->hw_dev = dev;
 261         }
 262
 263         tc_cls_common_offload_init(&cls_flower.common, tp);
 264         cls_flower.command = TC_CLSFLOWER_REPLACE;
 265         cls_flower.cookie = (unsigned long) f;
 266         cls_flower.dissector = dissector;
 267         cls_flower.mask = mask;
 268         cls_flower.key = &f->mkey;
 269         cls_flower.exts = &f->exts;
 270
 271         err = dev->netdev_ops->ndo_setup_tc(dev, TC_SETUP_CLSFLOWER,
 272                                             &cls_flower);
 273         if (!err)
 274                 f->flags |= TCA_CLS_FLAGS_IN_HW;
 275
 276         if (tc_skip_sw(f->flags))
 277                 return err;
 278         return 0;
 279 }
 280
 281 static void fl_hw_update_stats(struct tcf_proto *tp, struct cls_fl_filter *f)
 282 {
 283         struct tc_cls_flower_offload cls_flower = {};
 284         struct net_device *dev = f->hw_dev;
 285
 286         if (!tc_can_offload(dev))
 287                 return;
 288
 289         tc_cls_common_offload_init(&cls_flower.common, tp);
 290         cls_flower.command = TC_CLSFLOWER_STATS;
 291         cls_flower.cookie = (unsigned long) f;
 292         cls_flower.exts = &f->exts;
 293         cls_flower.egress_dev = f->hw_dev != tp->q->dev_queue->dev;
 294
 295         dev->netdev_ops->ndo_setup_tc(dev, TC_SETUP_CLSFLOWER,
 296                                       &cls_flower);
 297 }
 298
 299 static void __fl_delete(struct tcf_proto *tp, struct cls_fl_filter *f)
 300 {
 301         struct cls_fl_head *head = rtnl_dereference(tp->root);
 302
 303         idr_remove_ext(&head->handle_idr, f->handle);
 304         list_del_rcu(&f->list);
 305         if (!tc_skip_hw(f->flags))
 306                 fl_hw_destroy_filter(tp, f);
 307         tcf_unbind_filter(tp, &f->res);
 308         call_rcu(&f->rcu, fl_destroy_filter);
 309 }
 310
 311 static void fl_destroy_sleepable(struct work_struct *work)
 312 {
 313         struct cls_fl_head *head = container_of(work, struct cls_fl_head,
 314                                                 work);
 315         if (head->mask_assigned)
 316                 rhashtable_destroy(&head->ht);
 317         kfree(head);
 318         module_put(THIS_MODULE);
 319 }
 320
 321 static void fl_destroy_rcu(struct rcu_head *rcu)
 322 {
 323         struct cls_fl_head *head = container_of(rcu, struct cls_fl_head, rcu);
 324
 325         INIT_WORK(&head->work, fl_destroy_sleepable);
 326         schedule_work(&head->work);
 327 }
 328
 329 static void fl_destroy(struct tcf_proto *tp)
 330 {
 331         struct cls_fl_head *head = rtnl_dereference(tp->root);
 332         struct cls_fl_filter *f, *next;
 333
 334         list_for_each_entry_safe(f, next, &head->filters, list)
 335                 __fl_delete(tp, f);
 336         idr_destroy(&head->handle_idr);
 337
 338         __module_get(THIS_MODULE);
 339         call_rcu(&head->rcu, fl_destroy_rcu);
 340 }
 341
 342 static void *fl_get(struct tcf_proto *tp, u32 handle)
 343 {
 344         struct cls_fl_head *head = rtnl_dereference(tp->root);
 345
 346         return idr_find_ext(&head->handle_idr, handle);
 347 }
 348
 349 static const struct nla_policy fl_policy[TCA_FLOWER_MAX + 1] = {
 350         [TCA_FLOWER_UNSPEC]             = { .type = NLA_UNSPEC },
 351         [TCA_FLOWER_CLASSID]            = { .type = NLA_U32 },
 352         [TCA_FLOWER_INDEV]              = { .type = NLA_STRING,
 353                                             .len = IFNAMSIZ },
 354         [TCA_FLOWER_KEY_ETH_DST]        = { .len = ETH_ALEN },
 355         [TCA_FLOWER_KEY_ETH_DST_MASK]   = { .len = ETH_ALEN },
 356         [TCA_FLOWER_KEY_ETH_SRC]        = { .len = ETH_ALEN },
 357         [TCA_FLOWER_KEY_ETH_SRC_MASK]   = { .len = ETH_ALEN },
 358         [TCA_FLOWER_KEY_ETH_TYPE]       = { .type = NLA_U16 },
 359         [TCA_FLOWER_KEY_IP_PROTO]       = { .type = NLA_U8 },
 360         [TCA_FLOWER_KEY_IPV4_SRC]       = { .type = NLA_U32 },
 361         [TCA_FLOWER_KEY_IPV4_SRC_MASK]  = { .type = NLA_U32 },
 362         [TCA_FLOWER_KEY_IPV4_DST]       = { .type = NLA_U32 },
 363         [TCA_FLOWER_KEY_IPV4_DST_MASK]  = { .type = NLA_U32 },
 364         [TCA_FLOWER_KEY_IPV6_SRC]       = { .len = sizeof(struct in6_addr) },
 365         [TCA_FLOWER_KEY_IPV6_SRC_MASK]  = { .len = sizeof(struct in6_addr) },
 366         [TCA_FLOWER_KEY_IPV6_DST]       = { .len = sizeof(struct in6_addr) },
 367         [TCA_FLOWER_KEY_IPV6_DST_MASK]  = { .len = sizeof(struct in6_addr) },
 368         [TCA_FLOWER_KEY_TCP_SRC]        = { .type = NLA_U16 },
 369         [TCA_FLOWER_KEY_TCP_DST]        = { .type = NLA_U16 },
 370         [TCA_FLOWER_KEY_UDP_SRC]        = { .type = NLA_U16 },
 371         [TCA_FLOWER_KEY_UDP_DST]        = { .type = NLA_U16 },
 372         [TCA_FLOWER_KEY_VLAN_ID]        = { .type = NLA_U16 },
 373         [TCA_FLOWER_KEY_VLAN_PRIO]      = { .type = NLA_U8 },
 374         [TCA_FLOWER_KEY_VLAN_ETH_TYPE]  = { .type = NLA_U16 },
 375         [TCA_FLOWER_KEY_ENC_KEY_ID]     = { .type = NLA_U32 },
 376         [TCA_FLOWER_KEY_ENC_IPV4_SRC]   = { .type = NLA_U32 },
 377         [TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK] = { .type = NLA_U32 },
 378         [TCA_FLOWER_KEY_ENC_IPV4_DST]   = { .type = NLA_U32 },
 379         [TCA_FLOWER_KEY_ENC_IPV4_DST_MASK] = { .type = NLA_U32 },
 380         [TCA_FLOWER_KEY_ENC_IPV6_SRC]   = { .len = sizeof(struct in6_addr) },
 381         [TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK] = { .len = sizeof(struct in6_addr) },
 382         [TCA_FLOWER_KEY_ENC_IPV6_DST]   = { .len = sizeof(struct in6_addr) },
 383         [TCA_FLOWER_KEY_ENC_IPV6_DST_MASK] = { .len = sizeof(struct in6_addr) },
 384         [TCA_FLOWER_KEY_TCP_SRC_MASK]   = { .type = NLA_U16 },
 385         [TCA_FLOWER_KEY_TCP_DST_MASK]   = { .type = NLA_U16 },
 386         [TCA_FLOWER_KEY_UDP_SRC_MASK]   = { .type = NLA_U16 },
 387         [TCA_FLOWER_KEY_UDP_DST_MASK]   = { .type = NLA_U16 },
 388         [TCA_FLOWER_KEY_SCTP_SRC_MASK]  = { .type = NLA_U16 },
 389         [TCA_FLOWER_KEY_SCTP_DST_MASK]  = { .type = NLA_U16 },
 390         [TCA_FLOWER_KEY_SCTP_SRC]       = { .type = NLA_U16 },
 391         [TCA_FLOWER_KEY_SCTP_DST]       = { .type = NLA_U16 },
 392         [TCA_FLOWER_KEY_ENC_UDP_SRC_PORT]       = { .type = NLA_U16 },
 393         [TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK]  = { .type = NLA_U16 },
 394         [TCA_FLOWER_KEY_ENC_UDP_DST_PORT]       = { .type = NLA_U16 },
 395         [TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK]  = { .type = NLA_U16 },
 396         [TCA_FLOWER_KEY_FLAGS]          = { .type = NLA_U32 },
 397         [TCA_FLOWER_KEY_FLAGS_MASK]     = { .type = NLA_U32 },
 398         [TCA_FLOWER_KEY_ICMPV4_TYPE]    = { .type = NLA_U8 },
 399         [TCA_FLOWER_KEY_ICMPV4_TYPE_MASK] = { .type = NLA_U8 },
 400         [TCA_FLOWER_KEY_ICMPV4_CODE]    = { .type = NLA_U8 },
 401         [TCA_FLOWER_KEY_ICMPV4_CODE_MASK] = { .type = NLA_U8 },
 402         [TCA_FLOWER_KEY_ICMPV6_TYPE]    = { .type = NLA_U8 },
 403         [TCA_FLOWER_KEY_ICMPV6_TYPE_MASK] = { .type = NLA_U8 },
 404         [TCA_FLOWER_KEY_ICMPV6_CODE]    = { .type = NLA_U8 },
 405         [TCA_FLOWER_KEY_ICMPV6_CODE_MASK] = { .type = NLA_U8 },
 406         [TCA_FLOWER_KEY_ARP_SIP]        = { .type = NLA_U32 },
 407         [TCA_FLOWER_KEY_ARP_SIP_MASK]   = { .type = NLA_U32 },
 408         [TCA_FLOWER_KEY_ARP_TIP]        = { .type = NLA_U32 },
 409         [TCA_FLOWER_KEY_ARP_TIP_MASK]   = { .type = NLA_U32 },
 410         [TCA_FLOWER_KEY_ARP_OP]         = { .type = NLA_U8 },
 411         [TCA_FLOWER_KEY_ARP_OP_MASK]    = { .type = NLA_U8 },
 412         [TCA_FLOWER_KEY_ARP_SHA]        = { .len = ETH_ALEN },
 413         [TCA_FLOWER_KEY_ARP_SHA_MASK]   = { .len = ETH_ALEN },
 414         [TCA_FLOWER_KEY_ARP_THA]        = { .len = ETH_ALEN },
 415         [TCA_FLOWER_KEY_ARP_THA_MASK]   = { .len = ETH_ALEN },
 416         [TCA_FLOWER_KEY_MPLS_TTL]       = { .type = NLA_U8 },
 417         [TCA_FLOWER_KEY_MPLS_BOS]       = { .type = NLA_U8 },
 418         [TCA_FLOWER_KEY_MPLS_TC]        = { .type = NLA_U8 },
 419         [TCA_FLOWER_KEY_MPLS_LABEL]     = { .type = NLA_U32 },
 420         [TCA_FLOWER_KEY_TCP_FLAGS]      = { .type = NLA_U16 },
 421         [TCA_FLOWER_KEY_TCP_FLAGS_MASK] = { .type = NLA_U16 },
 422         [TCA_FLOWER_KEY_IP_TOS]         = { .type = NLA_U8 },
 423         [TCA_FLOWER_KEY_IP_TOS_MASK]    = { .type = NLA_U8 },
 424         [TCA_FLOWER_KEY_IP_TTL]         = { .type = NLA_U8 },
 425         [TCA_FLOWER_KEY_IP_TTL_MASK]    = { .type = NLA_U8 },
 426 };
 427
 428 static void fl_set_key_val(struct nlattr **tb,
 429                            void *val, int val_type,
 430                            void *mask, int mask_type, int len)
 431 {
 432         if (!tb[val_type])
 433                 return;
 434         memcpy(val, nla_data(tb[val_type]), len);
 435         if (mask_type == TCA_FLOWER_UNSPEC || !tb[mask_type])
 436                 memset(mask, 0xff, len);
 437         else
 438                 memcpy(mask, nla_data(tb[mask_type]), len);
 439 }
 440
 441 static int fl_set_key_mpls(struct nlattr **tb,
 442                            struct flow_dissector_key_mpls *key_val,
 443                            struct flow_dissector_key_mpls *key_mask)
 444 {
 445         if (tb[TCA_FLOWER_KEY_MPLS_TTL]) {
 446                 key_val->mpls_ttl = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_TTL]);
 447                 key_mask->mpls_ttl = MPLS_TTL_MASK;
 448         }
 449         if (tb[TCA_FLOWER_KEY_MPLS_BOS]) {
 450                 u8 bos = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_BOS]);
 451
 452                 if (bos & ~MPLS_BOS_MASK)
 453                         return -EINVAL;
 454                 key_val->mpls_bos = bos;
 455                 key_mask->mpls_bos = MPLS_BOS_MASK;
 456         }
 457         if (tb[TCA_FLOWER_KEY_MPLS_TC]) {
 458                 u8 tc = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_TC]);
 459
 460                 if (tc & ~MPLS_TC_MASK)
 461                         return -EINVAL;
 462                 key_val->mpls_tc = tc;
 463                 key_mask->mpls_tc = MPLS_TC_MASK;
 464         }
 465         if (tb[TCA_FLOWER_KEY_MPLS_LABEL]) {
 466                 u32 label = nla_get_u32(tb[TCA_FLOWER_KEY_MPLS_LABEL]);
 467
 468                 if (label & ~MPLS_LABEL_MASK)
 469                         return -EINVAL;
 470                 key_val->mpls_label = label;
 471                 key_mask->mpls_label = MPLS_LABEL_MASK;
 472         }
 473         return 0;
 474 }
 475
 476 static void fl_set_key_vlan(struct nlattr **tb,
 477                             struct flow_dissector_key_vlan *key_val,
 478                             struct flow_dissector_key_vlan *key_mask)
 479 {
 480 #define VLAN_PRIORITY_MASK      0x7
 481
 482         if (tb[TCA_FLOWER_KEY_VLAN_ID]) {
 483                 key_val->vlan_id =
 484                         nla_get_u16(tb[TCA_FLOWER_KEY_VLAN_ID]) & VLAN_VID_MASK;
 485                 key_mask->vlan_id = VLAN_VID_MASK;
 486         }
 487         if (tb[TCA_FLOWER_KEY_VLAN_PRIO]) {
 488                 key_val->vlan_priority =
 489                         nla_get_u8(tb[TCA_FLOWER_KEY_VLAN_PRIO]) &
 490                         VLAN_PRIORITY_MASK;
 491                 key_mask->vlan_priority = VLAN_PRIORITY_MASK;
 492         }
 493 }
 494
 495 static void fl_set_key_flag(u32 flower_key, u32 flower_mask,
 496                             u32 *dissector_key, u32 *dissector_mask,
 497                             u32 flower_flag_bit, u32 dissector_flag_bit)
 498 {
 499         if (flower_mask & flower_flag_bit) {
 500                 *dissector_mask |= dissector_flag_bit;
 501                 if (flower_key & flower_flag_bit)
 502                         *dissector_key |= dissector_flag_bit;
 503         }
 504 }
 505
 506 static int fl_set_key_flags(struct nlattr **tb,
 507                             u32 *flags_key, u32 *flags_mask)
 508 {
 509         u32 key, mask;
 510
 511         /* mask is mandatory for flags */
 512         if (!tb[TCA_FLOWER_KEY_FLAGS_MASK])
 513                 return -EINVAL;
 514
 515         key = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS]));
 516         mask = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS_MASK]));
 517
 518         *flags_key  = 0;
 519         *flags_mask = 0;
 520
 521         fl_set_key_flag(key, mask, flags_key, flags_mask,
 522                         TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
 523
 524         return 0;
 525 }
 526
 527 static void fl_set_key_ip(struct nlattr **tb,
 528                           struct flow_dissector_key_ip *key,
 529                           struct flow_dissector_key_ip *mask)
 530 {
 531                 fl_set_key_val(tb, &key->tos, TCA_FLOWER_KEY_IP_TOS,
 532                                &mask->tos, TCA_FLOWER_KEY_IP_TOS_MASK,
 533                                sizeof(key->tos));
 534
 535                 fl_set_key_val(tb, &key->ttl, TCA_FLOWER_KEY_IP_TTL,
 536                                &mask->ttl, TCA_FLOWER_KEY_IP_TTL_MASK,
 537                                sizeof(key->ttl));
 538 }
 539
 540 static int fl_set_key(struct net *net, struct nlattr **tb,
 541                       struct fl_flow_key *key, struct fl_flow_key *mask)
 542 {
 543         __be16 ethertype;
 544         int ret = 0;
 545 #ifdef CONFIG_NET_CLS_IND
 546         if (tb[TCA_FLOWER_INDEV]) {
 547                 int err = tcf_change_indev(net, tb[TCA_FLOWER_INDEV]);
 548                 if (err < 0)
 549                         return err;
 550                 key->indev_ifindex = err;
 551                 mask->indev_ifindex = 0xffffffff;
 552         }
 553 #endif
 554
 555         fl_set_key_val(tb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
 556                        mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
 557                        sizeof(key->eth.dst));
 558         fl_set_key_val(tb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
 559                        mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
 560                        sizeof(key->eth.src));
 561
 562         if (tb[TCA_FLOWER_KEY_ETH_TYPE]) {
 563                 ethertype = nla_get_be16(tb[TCA_FLOWER_KEY_ETH_TYPE]);
 564
 565                 if (ethertype == htons(ETH_P_8021Q)) {
 566                         fl_set_key_vlan(tb, &key->vlan, &mask->vlan);
 567                         fl_set_key_val(tb, &key->basic.n_proto,
 568                                        TCA_FLOWER_KEY_VLAN_ETH_TYPE,
 569                                        &mask->basic.n_proto, TCA_FLOWER_UNSPEC,
 570                                        sizeof(key->basic.n_proto));
 571                 } else {
 572                         key->basic.n_proto = ethertype;
 573                         mask->basic.n_proto = cpu_to_be16(~0);
 574                 }
 575         }
 576
 577         if (key->basic.n_proto == htons(ETH_P_IP) ||
 578             key->basic.n_proto == htons(ETH_P_IPV6)) {
 579                 fl_set_key_val(tb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
 580                                &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
 581                                sizeof(key->basic.ip_proto));
 582                 fl_set_key_ip(tb, &key->ip, &mask->ip);
 583         }
 584
 585         if (tb[TCA_FLOWER_KEY_IPV4_SRC] || tb[TCA_FLOWER_KEY_IPV4_DST]) {
 586                 key->control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
 587                 mask->control.addr_type = ~0;
 588                 fl_set_key_val(tb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
 589                                &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
 590                                sizeof(key->ipv4.src));
 591                 fl_set_key_val(tb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
 592                                &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
 593                                sizeof(key->ipv4.dst));
 594         } else if (tb[TCA_FLOWER_KEY_IPV6_SRC] || tb[TCA_FLOWER_KEY_IPV6_DST]) {
 595                 key->control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
 596                 mask->control.addr_type = ~0;
 597                 fl_set_key_val(tb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
 598                                &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
 599                                sizeof(key->ipv6.src));
 600                 fl_set_key_val(tb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
 601                                &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
 602                                sizeof(key->ipv6.dst));
 603         }
 604
 605         if (key->basic.ip_proto == IPPROTO_TCP) {
 606                 fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
 607                                &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
 608                                sizeof(key->tp.src));
 609                 fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
 610                                &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
 611                                sizeof(key->tp.dst));
 612                 fl_set_key_val(tb, &key->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS,
 613                                &mask->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS_MASK,
 614                                sizeof(key->tcp.flags));
 615         } else if (key->basic.ip_proto == IPPROTO_UDP) {
 616                 fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
 617                                &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
 618                                sizeof(key->tp.src));
 619                 fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
 620                                &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
 621                                sizeof(key->tp.dst));
 622         } else if (key->basic.ip_proto == IPPROTO_SCTP) {
 623                 fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
 624                                &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
 625                                sizeof(key->tp.src));
 626                 fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
 627                                &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
 628                                sizeof(key->tp.dst));
 629         } else if (key->basic.n_proto == htons(ETH_P_IP) &&
 630                    key->basic.ip_proto == IPPROTO_ICMP) {
 631                 fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV4_TYPE,
 632                                &mask->icmp.type,
 633                                TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
 634                                sizeof(key->icmp.type));
 635                 fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
 636                                &mask->icmp.code,
 637                                TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
 638                                sizeof(key->icmp.code));
 639         } else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
 640                    key->basic.ip_proto == IPPROTO_ICMPV6) {
 641                 fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV6_TYPE,
 642                                &mask->icmp.type,
 643                                TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
 644                                sizeof(key->icmp.type));
 645                 fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV6_CODE,
 646                                &mask->icmp.code,
 647                                TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
 648                                sizeof(key->icmp.code));
 649         } else if (key->basic.n_proto == htons(ETH_P_MPLS_UC) ||
 650                    key->basic.n_proto == htons(ETH_P_MPLS_MC)) {
 651                 ret = fl_set_key_mpls(tb, &key->mpls, &mask->mpls);
 652                 if (ret)
 653                         return ret;
 654         } else if (key->basic.n_proto == htons(ETH_P_ARP) ||
 655                    key->basic.n_proto == htons(ETH_P_RARP)) {
 656                 fl_set_key_val(tb, &key->arp.sip, TCA_FLOWER_KEY_ARP_SIP,
 657                                &mask->arp.sip, TCA_FLOWER_KEY_ARP_SIP_MASK,
 658                                sizeof(key->arp.sip));
 659                 fl_set_key_val(tb, &key->arp.tip, TCA_FLOWER_KEY_ARP_TIP,
 660                                &mask->arp.tip, TCA_FLOWER_KEY_ARP_TIP_MASK,
 661                                sizeof(key->arp.tip));
 662                 fl_set_key_val(tb, &key->arp.op, TCA_FLOWER_KEY_ARP_OP,
 663                                &mask->arp.op, TCA_FLOWER_KEY_ARP_OP_MASK,
 664                                sizeof(key->arp.op));
 665                 fl_set_key_val(tb, key->arp.sha, TCA_FLOWER_KEY_ARP_SHA,
 666                                mask->arp.sha, TCA_FLOWER_KEY_ARP_SHA_MASK,
 667                                sizeof(key->arp.sha));
 668                 fl_set_key_val(tb, key->arp.tha, TCA_FLOWER_KEY_ARP_THA,
 669                                mask->arp.tha, TCA_FLOWER_KEY_ARP_THA_MASK,
 670                                sizeof(key->arp.tha));
 671         }
 672
 673         if (tb[TCA_FLOWER_KEY_ENC_IPV4_SRC] ||
 674             tb[TCA_FLOWER_KEY_ENC_IPV4_DST]) {
 675                 key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
 676                 mask->enc_control.addr_type = ~0;
 677                 fl_set_key_val(tb, &key->enc_ipv4.src,
 678                                TCA_FLOWER_KEY_ENC_IPV4_SRC,
 679                                &mask->enc_ipv4.src,
 680                                TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
 681                                sizeof(key->enc_ipv4.src));
 682                 fl_set_key_val(tb, &key->enc_ipv4.dst,
 683                                TCA_FLOWER_KEY_ENC_IPV4_DST,
 684                                &mask->enc_ipv4.dst,
 685                                TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
 686                                sizeof(key->enc_ipv4.dst));
 687         }
 688
 689         if (tb[TCA_FLOWER_KEY_ENC_IPV6_SRC] ||
 690             tb[TCA_FLOWER_KEY_ENC_IPV6_DST]) {
 691                 key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
 692                 mask->enc_control.addr_type = ~0;
 693                 fl_set_key_val(tb, &key->enc_ipv6.src,
 694                                TCA_FLOWER_KEY_ENC_IPV6_SRC,
 695                                &mask->enc_ipv6.src,
 696                                TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
 697                                sizeof(key->enc_ipv6.src));
 698                 fl_set_key_val(tb, &key->enc_ipv6.dst,
 699                                TCA_FLOWER_KEY_ENC_IPV6_DST,
 700                                &mask->enc_ipv6.dst,
 701                                TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
 702                                sizeof(key->enc_ipv6.dst));
 703         }
 704
 705         fl_set_key_val(tb, &key->enc_key_id.keyid, TCA_FLOWER_KEY_ENC_KEY_ID,
 706                        &mask->enc_key_id.keyid, TCA_FLOWER_UNSPEC,
 707                        sizeof(key->enc_key_id.keyid));
 708
 709         fl_set_key_val(tb, &key->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
 710                        &mask->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
 711                        sizeof(key->enc_tp.src));
 712
 713         fl_set_key_val(tb, &key->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
 714                        &mask->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
 715                        sizeof(key->enc_tp.dst));
 716
 717         if (tb[TCA_FLOWER_KEY_FLAGS])
 718                 ret = fl_set_key_flags(tb, &key->control.flags, &mask->control.flags);
 719
 720         return ret;
 721 }
 722
 723 static bool fl_mask_eq(struct fl_flow_mask *mask1,
 724                        struct fl_flow_mask *mask2)
 725 {
 726         const long *lmask1 = fl_key_get_start(&mask1->key, mask1);
 727         const long *lmask2 = fl_key_get_start(&mask2->key, mask2);
 728
 729         return !memcmp(&mask1->range, &mask2->range, sizeof(mask1->range)) &&
 730                !memcmp(lmask1, lmask2, fl_mask_range(mask1));
 731 }
 732
 733 static const struct rhashtable_params fl_ht_params = {
 734         .key_offset = offsetof(struct cls_fl_filter, mkey), /* base offset */
 735         .head_offset = offsetof(struct cls_fl_filter, ht_node),
 736         .automatic_shrinking = true,
 737 };
 738
 739 static int fl_init_hashtable(struct cls_fl_head *head,
 740                              struct fl_flow_mask *mask)
 741 {
 742         head->ht_params = fl_ht_params;
 743         head->ht_params.key_len = fl_mask_range(mask);
 744         head->ht_params.key_offset += mask->range.start;
 745
 746         return rhashtable_init(&head->ht, &head->ht_params);
 747 }
 748
 749 #define FL_KEY_MEMBER_OFFSET(member) offsetof(struct fl_flow_key, member)
 750 #define FL_KEY_MEMBER_SIZE(member) (sizeof(((struct fl_flow_key *) 0)->member))
 751
 752 #define FL_KEY_IS_MASKED(mask, member)                                          \
 753         memchr_inv(((char *)mask) + FL_KEY_MEMBER_OFFSET(member),               \
 754                    0, FL_KEY_MEMBER_SIZE(member))                               \
 755
 756 #define FL_KEY_SET(keys, cnt, id, member)                                       \
 757         do {                                                                    \
 758                 keys[cnt].key_id = id;                                          \
 759                 keys[cnt].offset = FL_KEY_MEMBER_OFFSET(member);                \
 760                 cnt++;                                                          \
 761         } while(0);
 762
 763 #define FL_KEY_SET_IF_MASKED(mask, keys, cnt, id, member)                       \
 764         do {                                                                    \
 765                 if (FL_KEY_IS_MASKED(mask, member))                             \
 766                         FL_KEY_SET(keys, cnt, id, member);                      \
 767         } while(0);
 768
 769 static void fl_init_dissector(struct cls_fl_head *head,
 770                               struct fl_flow_mask *mask)
 771 {
 772         struct flow_dissector_key keys[FLOW_DISSECTOR_KEY_MAX];
 773         size_t cnt = 0;
 774
 775         FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_CONTROL, control);
 776         FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_BASIC, basic);
 777         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 778                              FLOW_DISSECTOR_KEY_ETH_ADDRS, eth);
 779         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 780                              FLOW_DISSECTOR_KEY_IPV4_ADDRS, ipv4);
 781         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 782                              FLOW_DISSECTOR_KEY_IPV6_ADDRS, ipv6);
 783         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 784                              FLOW_DISSECTOR_KEY_PORTS, tp);
 785         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 786                              FLOW_DISSECTOR_KEY_IP, ip);
 787         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 788                              FLOW_DISSECTOR_KEY_TCP, tcp);
 789         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 790                              FLOW_DISSECTOR_KEY_ICMP, icmp);
 791         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 792                              FLOW_DISSECTOR_KEY_ARP, arp);
 793         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 794                              FLOW_DISSECTOR_KEY_MPLS, mpls);
 795         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 796                              FLOW_DISSECTOR_KEY_VLAN, vlan);
 797         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 798                              FLOW_DISSECTOR_KEY_ENC_KEYID, enc_key_id);
 799         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 800                              FLOW_DISSECTOR_KEY_ENC_IPV4_ADDRS, enc_ipv4);
 801         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 802                              FLOW_DISSECTOR_KEY_ENC_IPV6_ADDRS, enc_ipv6);
 803         if (FL_KEY_IS_MASKED(&mask->key, enc_ipv4) ||
 804             FL_KEY_IS_MASKED(&mask->key, enc_ipv6))
 805                 FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_ENC_CONTROL,
 806                            enc_control);
 807         FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
 808                              FLOW_DISSECTOR_KEY_ENC_PORTS, enc_tp);
 809
 810         skb_flow_dissector_init(&head->dissector, keys, cnt);
 811 }
 812
 813 static int fl_check_assign_mask(struct cls_fl_head *head,
 814                                 struct fl_flow_mask *mask)
 815 {
 816         int err;
 817
 818         if (head->mask_assigned) {
 819                 if (!fl_mask_eq(&head->mask, mask))
 820                         return -EINVAL;
 821                 else
 822                         return 0;
 823         }
 824
 825         /* Mask is not assigned yet. So assign it and init hashtable
 826          * according to that.
 827          */
 828         err = fl_init_hashtable(head, mask);
 829         if (err)
 830                 return err;
 831         memcpy(&head->mask, mask, sizeof(head->mask));
 832         head->mask_assigned = true;
 833
 834         fl_init_dissector(head, mask);
 835
 836         return 0;
 837 }
 838
 839 static int fl_set_parms(struct net *net, struct tcf_proto *tp,
 840                         struct cls_fl_filter *f, struct fl_flow_mask *mask,
 841                         unsigned long base, struct nlattr **tb,
 842                         struct nlattr *est, bool ovr)
 843 {
 844         int err;
 845
 846         err = tcf_exts_validate(net, tp, tb, est, &f->exts, ovr);
 847         if (err < 0)
 848                 return err;
 849
 850         if (tb[TCA_FLOWER_CLASSID]) {
 851                 f->res.classid = nla_get_u32(tb[TCA_FLOWER_CLASSID]);
 852                 tcf_bind_filter(tp, &f->res, base);
 853         }
 854
 855         err = fl_set_key(net, tb, &f->key, &mask->key);
 856         if (err)
 857                 return err;
 858
 859         fl_mask_update_range(mask);
 860         fl_set_masked_key(&f->mkey, &f->key, mask);
 861
 862         return 0;
 863 }
 864
 865 static int fl_change(struct net *net, struct sk_buff *in_skb,
 866                      struct tcf_proto *tp, unsigned long base,
 867                      u32 handle, struct nlattr **tca,
 868                      void **arg, bool ovr)
 869 {
 870         struct cls_fl_head *head = rtnl_dereference(tp->root);
 871         struct cls_fl_filter *fold = *arg;
 872         struct cls_fl_filter *fnew;
 873         struct nlattr **tb;
 874         struct fl_flow_mask mask = {};
 875         unsigned long idr_index;
 876         int err;
 877
 878         if (!tca[TCA_OPTIONS])
 879                 return -EINVAL;
 880
 881         tb = kcalloc(TCA_FLOWER_MAX + 1, sizeof(struct nlattr *), GFP_KERNEL);
 882         if (!tb)
 883                 return -ENOBUFS;
 884
 885         err = nla_parse_nested(tb, TCA_FLOWER_MAX, tca[TCA_OPTIONS],
 886                                fl_policy, NULL);
 887         if (err < 0)
 888                 goto errout_tb;
 889
 890         if (fold && handle && fold->handle != handle) {
 891                 err = -EINVAL;
 892                 goto errout_tb;
 893         }
 894
 895         fnew = kzalloc(sizeof(*fnew), GFP_KERNEL);
 896         if (!fnew) {
 897                 err = -ENOBUFS;
 898                 goto errout_tb;
 899         }
 900
 901         err = tcf_exts_init(&fnew->exts, TCA_FLOWER_ACT, 0);
 902         if (err < 0)
 903                 goto errout;
 904
 905         if (!handle) {
 906                 err = idr_alloc_ext(&head->handle_idr, fnew, &idr_index,
 907                                     1, 0x80000000, GFP_KERNEL);
 908                 if (err)
 909                         goto errout;
 910                 fnew->handle = idr_index;
 911         }
 912
 913         /* user specifies a handle and it doesn't exist */
 914         if (handle && !fold) {
 915                 err = idr_alloc_ext(&head->handle_idr, fnew, &idr_index,
 916                                     handle, handle + 1, GFP_KERNEL);
 917                 if (err)
 918                         goto errout;
 919                 fnew->handle = idr_index;
 920         }
 921
 922         if (tb[TCA_FLOWER_FLAGS]) {
 923                 fnew->flags = nla_get_u32(tb[TCA_FLOWER_FLAGS]);
 924
 925                 if (!tc_flags_valid(fnew->flags)) {
 926                         err = -EINVAL;
 927                         goto errout_idr;
 928                 }
 929         }
 930
 931         err = fl_set_parms(net, tp, fnew, &mask, base, tb, tca[TCA_RATE], ovr);
 932         if (err)
 933                 goto errout_idr;
 934
 935         err = fl_check_assign_mask(head, &mask);
 936         if (err)
 937                 goto errout_idr;
 938
 939         if (!tc_skip_sw(fnew->flags)) {
 940                 if (!fold && fl_lookup(head, &fnew->mkey)) {
 941                         err = -EEXIST;
 942                         goto errout_idr;
 943                 }
 944
 945                 err = rhashtable_insert_fast(&head->ht, &fnew->ht_node,
 946                                              head->ht_params);
 947                 if (err)
 948                         goto errout_idr;
 949         }
 950
 951         if (!tc_skip_hw(fnew->flags)) {
 952                 err = fl_hw_replace_filter(tp,
 953                                            &head->dissector,
 954                                            &mask.key,
 955                                            fnew);
 956                 if (err)
 957                         goto errout_idr;
 958         }
 959
 960         if (!tc_in_hw(fnew->flags))
 961                 fnew->flags |= TCA_CLS_FLAGS_NOT_IN_HW;
 962
 963         if (fold) {
 964                 if (!tc_skip_sw(fold->flags))
 965                         rhashtable_remove_fast(&head->ht, &fold->ht_node,
 966                                                head->ht_params);
 967                 if (!tc_skip_hw(fold->flags))
 968                         fl_hw_destroy_filter(tp, fold);
 969         }
 970
 971         *arg = fnew;
 972
 973         if (fold) {
 974                 fnew->handle = handle;
 975                 idr_replace_ext(&head->handle_idr, fnew, fnew->handle);
 976                 list_replace_rcu(&fold->list, &fnew->list);
 977                 tcf_unbind_filter(tp, &fold->res);
 978                 call_rcu(&fold->rcu, fl_destroy_filter);
 979         } else {
 980                 list_add_tail_rcu(&fnew->list, &head->filters);
 981         }
 982
 983         kfree(tb);
 984         return 0;
 985
 986 errout_idr:
 987         if (fnew->handle)
 988                 idr_remove_ext(&head->handle_idr, fnew->handle);
 989 errout:
 990         tcf_exts_destroy(&fnew->exts);
 991         kfree(fnew);
 992 errout_tb:
 993         kfree(tb);
 994         return err;
 995 }
 996
 997 static int fl_delete(struct tcf_proto *tp, void *arg, bool *last)
 998 {
 999         struct cls_fl_head *head = rtnl_dereference(tp->root);
1000         struct cls_fl_filter *f = arg;
1001
1002         if (!tc_skip_sw(f->flags))
1003                 rhashtable_remove_fast(&head->ht, &f->ht_node,
1004                                        head->ht_params);
1005         __fl_delete(tp, f);
1006         *last = list_empty(&head->filters);
1007         return 0;
1008 }
1009
1010 static void fl_walk(struct tcf_proto *tp, struct tcf_walker *arg)
1011 {
1012         struct cls_fl_head *head = rtnl_dereference(tp->root);
1013         struct cls_fl_filter *f;
1014
1015         list_for_each_entry_rcu(f, &head->filters, list) {
1016                 if (arg->count < arg->skip)
1017                         goto skip;
1018                 if (arg->fn(tp, f, arg) < 0) {
1019                         arg->stop = 1;
1020                         break;
1021                 }
1022 skip:
1023                 arg->count++;
1024         }
1025 }
1026
1027 static int fl_dump_key_val(struct sk_buff *skb,
1028                            void *val, int val_type,
1029                            void *mask, int mask_type, int len)
1030 {
1031         int err;
1032
1033         if (!memchr_inv(mask, 0, len))
1034                 return 0;
1035         err = nla_put(skb, val_type, len, val);
1036         if (err)
1037                 return err;
1038         if (mask_type != TCA_FLOWER_UNSPEC) {
1039                 err = nla_put(skb, mask_type, len, mask);
1040                 if (err)
1041                         return err;
1042         }
1043         return 0;
1044 }
1045
1046 static int fl_dump_key_mpls(struct sk_buff *skb,
1047                             struct flow_dissector_key_mpls *mpls_key,
1048                             struct flow_dissector_key_mpls *mpls_mask)
1049 {
1050         int err;
1051
1052         if (!memchr_inv(mpls_mask, 0, sizeof(*mpls_mask)))
1053                 return 0;
1054         if (mpls_mask->mpls_ttl) {
1055                 err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_TTL,
1056                                  mpls_key->mpls_ttl);
1057                 if (err)
1058                         return err;
1059         }
1060         if (mpls_mask->mpls_tc) {
1061                 err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_TC,
1062                                  mpls_key->mpls_tc);
1063                 if (err)
1064                         return err;
1065         }
1066         if (mpls_mask->mpls_label) {
1067                 err = nla_put_u32(skb, TCA_FLOWER_KEY_MPLS_LABEL,
1068                                   mpls_key->mpls_label);
1069                 if (err)
1070                         return err;
1071         }
1072         if (mpls_mask->mpls_bos) {
1073                 err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_BOS,
1074                                  mpls_key->mpls_bos);
1075                 if (err)
1076                         return err;
1077         }
1078         return 0;
1079 }
1080
1081 static int fl_dump_key_ip(struct sk_buff *skb,
1082                           struct flow_dissector_key_ip *key,
1083                           struct flow_dissector_key_ip *mask)
1084 {
1085         if (fl_dump_key_val(skb, &key->tos, TCA_FLOWER_KEY_IP_TOS, &mask->tos,
1086                             TCA_FLOWER_KEY_IP_TOS_MASK, sizeof(key->tos)) ||
1087             fl_dump_key_val(skb, &key->ttl, TCA_FLOWER_KEY_IP_TTL, &mask->ttl,
1088                             TCA_FLOWER_KEY_IP_TTL_MASK, sizeof(key->ttl)))
1089                 return -1;
1090
1091         return 0;
1092 }
1093
1094 static int fl_dump_key_vlan(struct sk_buff *skb,
1095                             struct flow_dissector_key_vlan *vlan_key,
1096                             struct flow_dissector_key_vlan *vlan_mask)
1097 {
1098         int err;
1099
1100         if (!memchr_inv(vlan_mask, 0, sizeof(*vlan_mask)))
1101                 return 0;
1102         if (vlan_mask->vlan_id) {
1103                 err = nla_put_u16(skb, TCA_FLOWER_KEY_VLAN_ID,
1104                                   vlan_key->vlan_id);
1105                 if (err)
1106                         return err;
1107         }
1108         if (vlan_mask->vlan_priority) {
1109                 err = nla_put_u8(skb, TCA_FLOWER_KEY_VLAN_PRIO,
1110                                  vlan_key->vlan_priority);
1111                 if (err)
1112                         return err;
1113         }
1114         return 0;
1115 }
1116
1117 static void fl_get_key_flag(u32 dissector_key, u32 dissector_mask,
1118                             u32 *flower_key, u32 *flower_mask,
1119                             u32 flower_flag_bit, u32 dissector_flag_bit)
1120 {
1121         if (dissector_mask & dissector_flag_bit) {
1122                 *flower_mask |= flower_flag_bit;
1123                 if (dissector_key & dissector_flag_bit)
1124                         *flower_key |= flower_flag_bit;
1125         }
1126 }
1127
1128 static int fl_dump_key_flags(struct sk_buff *skb, u32 flags_key, u32 flags_mask)
1129 {
1130         u32 key, mask;
1131         __be32 _key, _mask;
1132         int err;
1133
1134         if (!memchr_inv(&flags_mask, 0, sizeof(flags_mask)))
1135                 return 0;
1136
1137         key = 0;
1138         mask = 0;
1139
1140         fl_get_key_flag(flags_key, flags_mask, &key, &mask,
1141                         TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
1142
1143         _key = cpu_to_be32(key);
1144         _mask = cpu_to_be32(mask);
1145
1146         err = nla_put(skb, TCA_FLOWER_KEY_FLAGS, 4, &_key);
1147         if (err)
1148                 return err;
1149
1150         return nla_put(skb, TCA_FLOWER_KEY_FLAGS_MASK, 4, &_mask);
1151 }
1152
1153 static int fl_dump(struct net *net, struct tcf_proto *tp, void *fh,
1154                    struct sk_buff *skb, struct tcmsg *t)
1155 {
1156         struct cls_fl_head *head = rtnl_dereference(tp->root);
1157         struct cls_fl_filter *f = fh;
1158         struct nlattr *nest;
1159         struct fl_flow_key *key, *mask;
1160
1161         if (!f)
1162                 return skb->len;
1163
1164         t->tcm_handle = f->handle;
1165
1166         nest = nla_nest_start(skb, TCA_OPTIONS);
1167         if (!nest)
1168                 goto nla_put_failure;
1169
1170         if (f->res.classid &&
1171             nla_put_u32(skb, TCA_FLOWER_CLASSID, f->res.classid))
1172                 goto nla_put_failure;
1173
1174         key = &f->key;
1175         mask = &head->mask.key;
1176
1177         if (mask->indev_ifindex) {
1178                 struct net_device *dev;
1179
1180                 dev = __dev_get_by_index(net, key->indev_ifindex);
1181                 if (dev && nla_put_string(skb, TCA_FLOWER_INDEV, dev->name))
1182                         goto nla_put_failure;
1183         }
1184
1185         if (!tc_skip_hw(f->flags))
1186                 fl_hw_update_stats(tp, f);
1187
1188         if (fl_dump_key_val(skb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
1189                             mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
1190                             sizeof(key->eth.dst)) ||
1191             fl_dump_key_val(skb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
1192                             mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
1193                             sizeof(key->eth.src)) ||
1194             fl_dump_key_val(skb, &key->basic.n_proto, TCA_FLOWER_KEY_ETH_TYPE,
1195                             &mask->basic.n_proto, TCA_FLOWER_UNSPEC,
1196                             sizeof(key->basic.n_proto)))
1197                 goto nla_put_failure;
1198
1199         if (fl_dump_key_mpls(skb, &key->mpls, &mask->mpls))
1200                 goto nla_put_failure;
1201
1202         if (fl_dump_key_vlan(skb, &key->vlan, &mask->vlan))
1203                 goto nla_put_failure;
1204
1205         if ((key->basic.n_proto == htons(ETH_P_IP) ||
1206              key->basic.n_proto == htons(ETH_P_IPV6)) &&
1207             (fl_dump_key_val(skb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
1208                             &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
1209                             sizeof(key->basic.ip_proto)) ||
1210             fl_dump_key_ip(skb, &key->ip, &mask->ip)))
1211                 goto nla_put_failure;
1212
1213         if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1214             (fl_dump_key_val(skb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
1215                              &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
1216                              sizeof(key->ipv4.src)) ||
1217              fl_dump_key_val(skb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
1218                              &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
1219                              sizeof(key->ipv4.dst))))
1220                 goto nla_put_failure;
1221         else if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1222                  (fl_dump_key_val(skb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
1223                                   &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
1224                                   sizeof(key->ipv6.src)) ||
1225                   fl_dump_key_val(skb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
1226                                   &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
1227                                   sizeof(key->ipv6.dst))))
1228                 goto nla_put_failure;
1229
1230         if (key->basic.ip_proto == IPPROTO_TCP &&
1231             (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
1232                              &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
1233                              sizeof(key->tp.src)) ||
1234              fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
1235                              &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
1236                              sizeof(key->tp.dst)) ||
1237              fl_dump_key_val(skb, &key->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS,
1238                              &mask->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS_MASK,
1239                              sizeof(key->tcp.flags))))
1240                 goto nla_put_failure;
1241         else if (key->basic.ip_proto == IPPROTO_UDP &&
1242                  (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
1243                                   &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
1244                                   sizeof(key->tp.src)) ||
1245                   fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
1246                                   &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
1247                                   sizeof(key->tp.dst))))
1248                 goto nla_put_failure;
1249         else if (key->basic.ip_proto == IPPROTO_SCTP &&
1250                  (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
1251                                   &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
1252                                   sizeof(key->tp.src)) ||
1253                   fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
1254                                   &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
1255                                   sizeof(key->tp.dst))))
1256                 goto nla_put_failure;
1257         else if (key->basic.n_proto == htons(ETH_P_IP) &&
1258                  key->basic.ip_proto == IPPROTO_ICMP &&
1259                  (fl_dump_key_val(skb, &key->icmp.type,
1260                                   TCA_FLOWER_KEY_ICMPV4_TYPE, &mask->icmp.type,
1261                                   TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
1262                                   sizeof(key->icmp.type)) ||
1263                   fl_dump_key_val(skb, &key->icmp.code,
1264                                   TCA_FLOWER_KEY_ICMPV4_CODE, &mask->icmp.code,
1265                                   TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
1266                                   sizeof(key->icmp.code))))
1267                 goto nla_put_failure;
1268         else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
1269                  key->basic.ip_proto == IPPROTO_ICMPV6 &&
1270                  (fl_dump_key_val(skb, &key->icmp.type,
1271                                   TCA_FLOWER_KEY_ICMPV6_TYPE, &mask->icmp.type,
1272                                   TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
1273                                   sizeof(key->icmp.type)) ||
1274                   fl_dump_key_val(skb, &key->icmp.code,
1275                                   TCA_FLOWER_KEY_ICMPV6_CODE, &mask->icmp.code,
1276                                   TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
1277                                   sizeof(key->icmp.code))))
1278                 goto nla_put_failure;
1279         else if ((key->basic.n_proto == htons(ETH_P_ARP) ||
1280                   key->basic.n_proto == htons(ETH_P_RARP)) &&
1281                  (fl_dump_key_val(skb, &key->arp.sip,
1282                                   TCA_FLOWER_KEY_ARP_SIP, &mask->arp.sip,
1283                                   TCA_FLOWER_KEY_ARP_SIP_MASK,
1284                                   sizeof(key->arp.sip)) ||
1285                   fl_dump_key_val(skb, &key->arp.tip,
1286                                   TCA_FLOWER_KEY_ARP_TIP, &mask->arp.tip,
1287                                   TCA_FLOWER_KEY_ARP_TIP_MASK,
1288                                   sizeof(key->arp.tip)) ||
1289                   fl_dump_key_val(skb, &key->arp.op,
1290                                   TCA_FLOWER_KEY_ARP_OP, &mask->arp.op,
1291                                   TCA_FLOWER_KEY_ARP_OP_MASK,
1292                                   sizeof(key->arp.op)) ||
1293                   fl_dump_key_val(skb, key->arp.sha, TCA_FLOWER_KEY_ARP_SHA,
1294                                   mask->arp.sha, TCA_FLOWER_KEY_ARP_SHA_MASK,
1295                                   sizeof(key->arp.sha)) ||
1296                   fl_dump_key_val(skb, key->arp.tha, TCA_FLOWER_KEY_ARP_THA,
1297                                   mask->arp.tha, TCA_FLOWER_KEY_ARP_THA_MASK,
1298                                   sizeof(key->arp.tha))))
1299                 goto nla_put_failure;
1300
1301         if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1302             (fl_dump_key_val(skb, &key->enc_ipv4.src,
1303                             TCA_FLOWER_KEY_ENC_IPV4_SRC, &mask->enc_ipv4.src,
1304                             TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
1305                             sizeof(key->enc_ipv4.src)) ||
1306              fl_dump_key_val(skb, &key->enc_ipv4.dst,
1307                              TCA_FLOWER_KEY_ENC_IPV4_DST, &mask->enc_ipv4.dst,
1308                              TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
1309                              sizeof(key->enc_ipv4.dst))))
1310                 goto nla_put_failure;
1311         else if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1312                  (fl_dump_key_val(skb, &key->enc_ipv6.src,
1313                             TCA_FLOWER_KEY_ENC_IPV6_SRC, &mask->enc_ipv6.src,
1314                             TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
1315                             sizeof(key->enc_ipv6.src)) ||
1316                  fl_dump_key_val(skb, &key->enc_ipv6.dst,
1317                                  TCA_FLOWER_KEY_ENC_IPV6_DST,
1318                                  &mask->enc_ipv6.dst,
1319                                  TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
1320                             sizeof(key->enc_ipv6.dst))))
1321                 goto nla_put_failure;
1322
1323         if (fl_dump_key_val(skb, &key->enc_key_id, TCA_FLOWER_KEY_ENC_KEY_ID,
1324                             &mask->enc_key_id, TCA_FLOWER_UNSPEC,
1325                             sizeof(key->enc_key_id)) ||
1326             fl_dump_key_val(skb, &key->enc_tp.src,
1327                             TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
1328                             &mask->enc_tp.src,
1329                             TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
1330                             sizeof(key->enc_tp.src)) ||
1331             fl_dump_key_val(skb, &key->enc_tp.dst,
1332                             TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
1333                             &mask->enc_tp.dst,
1334                             TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
1335                             sizeof(key->enc_tp.dst)))
1336                 goto nla_put_failure;
1337
1338         if (fl_dump_key_flags(skb, key->control.flags, mask->control.flags))
1339                 goto nla_put_failure;
1340
1341         if (f->flags && nla_put_u32(skb, TCA_FLOWER_FLAGS, f->flags))
1342                 goto nla_put_failure;
1343
1344         if (tcf_exts_dump(skb, &f->exts))
1345                 goto nla_put_failure;
1346
1347         nla_nest_end(skb, nest);
1348
1349         if (tcf_exts_dump_stats(skb, &f->exts) < 0)
1350                 goto nla_put_failure;
1351
1352         return skb->len;
1353
1354 nla_put_failure:
1355         nla_nest_cancel(skb, nest);
1356         return -1;
1357 }
1358
1359 static void fl_bind_class(void *fh, u32 classid, unsigned long cl)
1360 {
1361         struct cls_fl_filter *f = fh;
1362
1363         if (f && f->res.classid == classid)
1364                 f->res.class = cl;
1365 }
1366
1367 static struct tcf_proto_ops cls_fl_ops __read_mostly = {
1368         .kind           = "flower",
1369         .classify       = fl_classify,
1370         .init           = fl_init,
1371         .destroy        = fl_destroy,
1372         .get            = fl_get,
1373         .change         = fl_change,
1374         .delete         = fl_delete,
1375         .walk           = fl_walk,
1376         .dump           = fl_dump,
1377         .bind_class     = fl_bind_class,
1378         .owner          = THIS_MODULE,
1379 };
1380
1381 static int __init cls_fl_init(void)
1382 {
1383         return register_tcf_proto_ops(&cls_fl_ops);
1384 }
1385
1386 static void __exit cls_fl_exit(void)
1387 {
1388         unregister_tcf_proto_ops(&cls_fl_ops);
1389 }
1390
1391 module_init(cls_fl_init);
1392 module_exit(cls_fl_exit);
1393
1394 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
1395 MODULE_DESCRIPTION("Flower classifier");
1396 MODULE_LICENSE("GPL v2");