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