1 // SPDX-License-Identifier: GPL-2.0-or-later
3 * Copyright (c) 2014 Jiri Pirko <jiri@resnulli.us>
6 #include <linux/module.h>
7 #include <linux/init.h>
8 #include <linux/kernel.h>
9 #include <linux/skbuff.h>
10 #include <linux/rtnetlink.h>
11 #include <linux/if_vlan.h>
12 #include <net/netlink.h>
13 #include <net/pkt_sched.h>
14 #include <net/pkt_cls.h>
16 #include <linux/tc_act/tc_vlan.h>
17 #include <net/tc_act/tc_vlan.h>
19 static unsigned int vlan_net_id
;
20 static struct tc_action_ops act_vlan_ops
;
22 static int tcf_vlan_act(struct sk_buff
*skb
, const struct tc_action
*a
,
23 struct tcf_result
*res
)
25 struct tcf_vlan
*v
= to_vlan(a
);
26 struct tcf_vlan_params
*p
;
31 tcf_lastuse_update(&v
->tcf_tm
);
32 tcf_action_update_bstats(&v
->common
, skb
);
34 /* Ensure 'data' points at mac_header prior calling vlan manipulating
37 if (skb_at_tc_ingress(skb
))
38 skb_push_rcsum(skb
, skb
->mac_len
);
40 action
= READ_ONCE(v
->tcf_action
);
42 p
= rcu_dereference_bh(v
->vlan_p
);
44 switch (p
->tcfv_action
) {
45 case TCA_VLAN_ACT_POP
:
46 err
= skb_vlan_pop(skb
);
50 case TCA_VLAN_ACT_PUSH
:
51 err
= skb_vlan_push(skb
, p
->tcfv_push_proto
, p
->tcfv_push_vid
|
52 (p
->tcfv_push_prio
<< VLAN_PRIO_SHIFT
));
56 case TCA_VLAN_ACT_MODIFY
:
57 /* No-op if no vlan tag (either hw-accel or in-payload) */
58 if (!skb_vlan_tagged(skb
))
60 /* extract existing tag (and guarantee no hw-accel tag) */
61 if (skb_vlan_tag_present(skb
)) {
62 tci
= skb_vlan_tag_get(skb
);
63 __vlan_hwaccel_clear_tag(skb
);
65 /* in-payload vlan tag, pop it */
66 err
= __skb_vlan_pop(skb
, &tci
);
71 tci
= (tci
& ~VLAN_VID_MASK
) | p
->tcfv_push_vid
;
72 /* replace prio bits, if tcfv_push_prio specified */
73 if (p
->tcfv_push_prio_exists
) {
74 tci
&= ~VLAN_PRIO_MASK
;
75 tci
|= p
->tcfv_push_prio
<< VLAN_PRIO_SHIFT
;
77 /* put updated tci as hwaccel tag */
78 __vlan_hwaccel_put_tag(skb
, p
->tcfv_push_proto
, tci
);
80 case TCA_VLAN_ACT_POP_ETH
:
81 err
= skb_eth_pop(skb
);
85 case TCA_VLAN_ACT_PUSH_ETH
:
86 err
= skb_eth_push(skb
, p
->tcfv_push_dst
, p
->tcfv_push_src
);
95 if (skb_at_tc_ingress(skb
))
96 skb_pull_rcsum(skb
, skb
->mac_len
);
101 tcf_action_inc_drop_qstats(&v
->common
);
105 static const struct nla_policy vlan_policy
[TCA_VLAN_MAX
+ 1] = {
106 [TCA_VLAN_UNSPEC
] = { .strict_start_type
= TCA_VLAN_PUSH_ETH_DST
},
107 [TCA_VLAN_PARMS
] = { .len
= sizeof(struct tc_vlan
) },
108 [TCA_VLAN_PUSH_VLAN_ID
] = { .type
= NLA_U16
},
109 [TCA_VLAN_PUSH_VLAN_PROTOCOL
] = { .type
= NLA_U16
},
110 [TCA_VLAN_PUSH_VLAN_PRIORITY
] = { .type
= NLA_U8
},
111 [TCA_VLAN_PUSH_ETH_DST
] = NLA_POLICY_ETH_ADDR
,
112 [TCA_VLAN_PUSH_ETH_SRC
] = NLA_POLICY_ETH_ADDR
,
115 static int tcf_vlan_init(struct net
*net
, struct nlattr
*nla
,
116 struct nlattr
*est
, struct tc_action
**a
,
117 struct tcf_proto
*tp
, u32 flags
,
118 struct netlink_ext_ack
*extack
)
120 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
121 bool bind
= flags
& TCA_ACT_FLAGS_BIND
;
122 struct nlattr
*tb
[TCA_VLAN_MAX
+ 1];
123 struct tcf_chain
*goto_ch
= NULL
;
124 bool push_prio_exists
= false;
125 struct tcf_vlan_params
*p
;
126 struct tc_vlan
*parm
;
130 __be16 push_proto
= 0;
139 err
= nla_parse_nested_deprecated(tb
, TCA_VLAN_MAX
, nla
, vlan_policy
,
144 if (!tb
[TCA_VLAN_PARMS
])
146 parm
= nla_data(tb
[TCA_VLAN_PARMS
]);
148 err
= tcf_idr_check_alloc(tn
, &index
, a
, bind
);
155 switch (parm
->v_action
) {
156 case TCA_VLAN_ACT_POP
:
158 case TCA_VLAN_ACT_PUSH
:
159 case TCA_VLAN_ACT_MODIFY
:
160 if (!tb
[TCA_VLAN_PUSH_VLAN_ID
]) {
162 tcf_idr_release(*a
, bind
);
164 tcf_idr_cleanup(tn
, index
);
167 push_vid
= nla_get_u16(tb
[TCA_VLAN_PUSH_VLAN_ID
]);
168 if (push_vid
>= VLAN_VID_MASK
) {
170 tcf_idr_release(*a
, bind
);
172 tcf_idr_cleanup(tn
, index
);
176 if (tb
[TCA_VLAN_PUSH_VLAN_PROTOCOL
]) {
177 push_proto
= nla_get_be16(tb
[TCA_VLAN_PUSH_VLAN_PROTOCOL
]);
178 switch (push_proto
) {
179 case htons(ETH_P_8021Q
):
180 case htons(ETH_P_8021AD
):
184 tcf_idr_release(*a
, bind
);
186 tcf_idr_cleanup(tn
, index
);
187 return -EPROTONOSUPPORT
;
190 push_proto
= htons(ETH_P_8021Q
);
193 push_prio_exists
= !!tb
[TCA_VLAN_PUSH_VLAN_PRIORITY
];
194 if (push_prio_exists
)
195 push_prio
= nla_get_u8(tb
[TCA_VLAN_PUSH_VLAN_PRIORITY
]);
197 case TCA_VLAN_ACT_POP_ETH
:
199 case TCA_VLAN_ACT_PUSH_ETH
:
200 if (!tb
[TCA_VLAN_PUSH_ETH_DST
] || !tb
[TCA_VLAN_PUSH_ETH_SRC
]) {
202 tcf_idr_release(*a
, bind
);
204 tcf_idr_cleanup(tn
, index
);
210 tcf_idr_release(*a
, bind
);
212 tcf_idr_cleanup(tn
, index
);
215 action
= parm
->v_action
;
218 ret
= tcf_idr_create_from_flags(tn
, index
, est
, a
,
219 &act_vlan_ops
, bind
, flags
);
221 tcf_idr_cleanup(tn
, index
);
226 } else if (!(flags
& TCA_ACT_FLAGS_REPLACE
)) {
227 tcf_idr_release(*a
, bind
);
231 err
= tcf_action_check_ctrlact(parm
->action
, tp
, &goto_ch
, extack
);
237 p
= kzalloc(sizeof(*p
), GFP_KERNEL
);
243 p
->tcfv_action
= action
;
244 p
->tcfv_push_vid
= push_vid
;
245 p
->tcfv_push_prio
= push_prio
;
246 p
->tcfv_push_prio_exists
= push_prio_exists
|| action
== TCA_VLAN_ACT_PUSH
;
247 p
->tcfv_push_proto
= push_proto
;
249 if (action
== TCA_VLAN_ACT_PUSH_ETH
) {
250 nla_memcpy(&p
->tcfv_push_dst
, tb
[TCA_VLAN_PUSH_ETH_DST
],
252 nla_memcpy(&p
->tcfv_push_src
, tb
[TCA_VLAN_PUSH_ETH_SRC
],
256 spin_lock_bh(&v
->tcf_lock
);
257 goto_ch
= tcf_action_set_ctrlact(*a
, parm
->action
, goto_ch
);
258 p
= rcu_replace_pointer(v
->vlan_p
, p
, lockdep_is_held(&v
->tcf_lock
));
259 spin_unlock_bh(&v
->tcf_lock
);
262 tcf_chain_put_by_act(goto_ch
);
269 tcf_chain_put_by_act(goto_ch
);
271 tcf_idr_release(*a
, bind
);
275 static void tcf_vlan_cleanup(struct tc_action
*a
)
277 struct tcf_vlan
*v
= to_vlan(a
);
278 struct tcf_vlan_params
*p
;
280 p
= rcu_dereference_protected(v
->vlan_p
, 1);
285 static int tcf_vlan_dump(struct sk_buff
*skb
, struct tc_action
*a
,
288 unsigned char *b
= skb_tail_pointer(skb
);
289 struct tcf_vlan
*v
= to_vlan(a
);
290 struct tcf_vlan_params
*p
;
291 struct tc_vlan opt
= {
292 .index
= v
->tcf_index
,
293 .refcnt
= refcount_read(&v
->tcf_refcnt
) - ref
,
294 .bindcnt
= atomic_read(&v
->tcf_bindcnt
) - bind
,
298 spin_lock_bh(&v
->tcf_lock
);
299 opt
.action
= v
->tcf_action
;
300 p
= rcu_dereference_protected(v
->vlan_p
, lockdep_is_held(&v
->tcf_lock
));
301 opt
.v_action
= p
->tcfv_action
;
302 if (nla_put(skb
, TCA_VLAN_PARMS
, sizeof(opt
), &opt
))
303 goto nla_put_failure
;
305 if ((p
->tcfv_action
== TCA_VLAN_ACT_PUSH
||
306 p
->tcfv_action
== TCA_VLAN_ACT_MODIFY
) &&
307 (nla_put_u16(skb
, TCA_VLAN_PUSH_VLAN_ID
, p
->tcfv_push_vid
) ||
308 nla_put_be16(skb
, TCA_VLAN_PUSH_VLAN_PROTOCOL
,
309 p
->tcfv_push_proto
) ||
310 (p
->tcfv_push_prio_exists
&&
311 nla_put_u8(skb
, TCA_VLAN_PUSH_VLAN_PRIORITY
, p
->tcfv_push_prio
))))
312 goto nla_put_failure
;
314 if (p
->tcfv_action
== TCA_VLAN_ACT_PUSH_ETH
) {
315 if (nla_put(skb
, TCA_VLAN_PUSH_ETH_DST
, ETH_ALEN
,
317 goto nla_put_failure
;
318 if (nla_put(skb
, TCA_VLAN_PUSH_ETH_SRC
, ETH_ALEN
,
320 goto nla_put_failure
;
323 tcf_tm_dump(&t
, &v
->tcf_tm
);
324 if (nla_put_64bit(skb
, TCA_VLAN_TM
, sizeof(t
), &t
, TCA_VLAN_PAD
))
325 goto nla_put_failure
;
326 spin_unlock_bh(&v
->tcf_lock
);
331 spin_unlock_bh(&v
->tcf_lock
);
336 static int tcf_vlan_walker(struct net
*net
, struct sk_buff
*skb
,
337 struct netlink_callback
*cb
, int type
,
338 const struct tc_action_ops
*ops
,
339 struct netlink_ext_ack
*extack
)
341 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
343 return tcf_generic_walker(tn
, skb
, cb
, type
, ops
, extack
);
346 static void tcf_vlan_stats_update(struct tc_action
*a
, u64 bytes
, u64 packets
,
347 u64 drops
, u64 lastuse
, bool hw
)
349 struct tcf_vlan
*v
= to_vlan(a
);
350 struct tcf_t
*tm
= &v
->tcf_tm
;
352 tcf_action_update_stats(a
, bytes
, packets
, drops
, hw
);
353 tm
->lastuse
= max_t(u64
, tm
->lastuse
, lastuse
);
356 static int tcf_vlan_search(struct net
*net
, struct tc_action
**a
, u32 index
)
358 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
360 return tcf_idr_search(tn
, a
, index
);
363 static size_t tcf_vlan_get_fill_size(const struct tc_action
*act
)
365 return nla_total_size(sizeof(struct tc_vlan
))
366 + nla_total_size(sizeof(u16
)) /* TCA_VLAN_PUSH_VLAN_ID */
367 + nla_total_size(sizeof(u16
)) /* TCA_VLAN_PUSH_VLAN_PROTOCOL */
368 + nla_total_size(sizeof(u8
)); /* TCA_VLAN_PUSH_VLAN_PRIORITY */
371 static int tcf_vlan_offload_act_setup(struct tc_action
*act
, void *entry_data
,
372 u32
*index_inc
, bool bind
,
373 struct netlink_ext_ack
*extack
)
376 struct flow_action_entry
*entry
= entry_data
;
378 switch (tcf_vlan_action(act
)) {
379 case TCA_VLAN_ACT_PUSH
:
380 entry
->id
= FLOW_ACTION_VLAN_PUSH
;
381 entry
->vlan
.vid
= tcf_vlan_push_vid(act
);
382 entry
->vlan
.proto
= tcf_vlan_push_proto(act
);
383 entry
->vlan
.prio
= tcf_vlan_push_prio(act
);
385 case TCA_VLAN_ACT_POP
:
386 entry
->id
= FLOW_ACTION_VLAN_POP
;
388 case TCA_VLAN_ACT_MODIFY
:
389 entry
->id
= FLOW_ACTION_VLAN_MANGLE
;
390 entry
->vlan
.vid
= tcf_vlan_push_vid(act
);
391 entry
->vlan
.proto
= tcf_vlan_push_proto(act
);
392 entry
->vlan
.prio
= tcf_vlan_push_prio(act
);
394 case TCA_VLAN_ACT_POP_ETH
:
395 entry
->id
= FLOW_ACTION_VLAN_POP_ETH
;
397 case TCA_VLAN_ACT_PUSH_ETH
:
398 entry
->id
= FLOW_ACTION_VLAN_PUSH_ETH
;
399 tcf_vlan_push_eth(entry
->vlan_push_eth
.src
, entry
->vlan_push_eth
.dst
, act
);
402 NL_SET_ERR_MSG_MOD(extack
, "Unsupported vlan action mode offload");
407 struct flow_offload_action
*fl_action
= entry_data
;
409 switch (tcf_vlan_action(act
)) {
410 case TCA_VLAN_ACT_PUSH
:
411 fl_action
->id
= FLOW_ACTION_VLAN_PUSH
;
413 case TCA_VLAN_ACT_POP
:
414 fl_action
->id
= FLOW_ACTION_VLAN_POP
;
416 case TCA_VLAN_ACT_MODIFY
:
417 fl_action
->id
= FLOW_ACTION_VLAN_MANGLE
;
419 case TCA_VLAN_ACT_POP_ETH
:
420 fl_action
->id
= FLOW_ACTION_VLAN_POP_ETH
;
422 case TCA_VLAN_ACT_PUSH_ETH
:
423 fl_action
->id
= FLOW_ACTION_VLAN_PUSH_ETH
;
433 static struct tc_action_ops act_vlan_ops
= {
436 .owner
= THIS_MODULE
,
438 .dump
= tcf_vlan_dump
,
439 .init
= tcf_vlan_init
,
440 .cleanup
= tcf_vlan_cleanup
,
441 .walk
= tcf_vlan_walker
,
442 .stats_update
= tcf_vlan_stats_update
,
443 .get_fill_size
= tcf_vlan_get_fill_size
,
444 .lookup
= tcf_vlan_search
,
445 .offload_act_setup
= tcf_vlan_offload_act_setup
,
446 .size
= sizeof(struct tcf_vlan
),
449 static __net_init
int vlan_init_net(struct net
*net
)
451 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
453 return tc_action_net_init(net
, tn
, &act_vlan_ops
);
456 static void __net_exit
vlan_exit_net(struct list_head
*net_list
)
458 tc_action_net_exit(net_list
, vlan_net_id
);
461 static struct pernet_operations vlan_net_ops
= {
462 .init
= vlan_init_net
,
463 .exit_batch
= vlan_exit_net
,
465 .size
= sizeof(struct tc_action_net
),
468 static int __init
vlan_init_module(void)
470 return tcf_register_action(&act_vlan_ops
, &vlan_net_ops
);
473 static void __exit
vlan_cleanup_module(void)
475 tcf_unregister_action(&act_vlan_ops
, &vlan_net_ops
);
478 module_init(vlan_init_module
);
479 module_exit(vlan_cleanup_module
);
481 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
482 MODULE_DESCRIPTION("vlan manipulation actions");
483 MODULE_LICENSE("GPL v2");