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