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"); 1397