2 * Copyright (c) 2014 Jiri Pirko <jiri@resnulli.us>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
10 #include <linux/module.h>
11 #include <linux/init.h>
12 #include <linux/kernel.h>
13 #include <linux/skbuff.h>
14 #include <linux/rtnetlink.h>
15 #include <linux/if_vlan.h>
16 #include <net/netlink.h>
17 #include <net/pkt_sched.h>
19 #include <linux/tc_act/tc_vlan.h>
20 #include <net/tc_act/tc_vlan.h>
22 static unsigned int vlan_net_id
;
23 static struct tc_action_ops act_vlan_ops
;
25 static int tcf_vlan(struct sk_buff
*skb
, const struct tc_action
*a
,
26 struct tcf_result
*res
)
28 struct tcf_vlan
*v
= to_vlan(a
);
29 struct tcf_vlan_params
*p
;
34 tcf_lastuse_update(&v
->tcf_tm
);
35 bstats_cpu_update(this_cpu_ptr(v
->common
.cpu_bstats
), skb
);
37 /* Ensure 'data' points at mac_header prior calling vlan manipulating
40 if (skb_at_tc_ingress(skb
))
41 skb_push_rcsum(skb
, skb
->mac_len
);
45 action
= READ_ONCE(v
->tcf_action
);
47 p
= rcu_dereference(v
->vlan_p
);
49 switch (p
->tcfv_action
) {
50 case TCA_VLAN_ACT_POP
:
51 err
= skb_vlan_pop(skb
);
55 case TCA_VLAN_ACT_PUSH
:
56 err
= skb_vlan_push(skb
, p
->tcfv_push_proto
, p
->tcfv_push_vid
|
57 (p
->tcfv_push_prio
<< VLAN_PRIO_SHIFT
));
61 case TCA_VLAN_ACT_MODIFY
:
62 /* No-op if no vlan tag (either hw-accel or in-payload) */
63 if (!skb_vlan_tagged(skb
))
65 /* extract existing tag (and guarantee no hw-accel tag) */
66 if (skb_vlan_tag_present(skb
)) {
67 tci
= skb_vlan_tag_get(skb
);
70 /* in-payload vlan tag, pop it */
71 err
= __skb_vlan_pop(skb
, &tci
);
76 tci
= (tci
& ~VLAN_VID_MASK
) | p
->tcfv_push_vid
;
77 /* replace prio bits, if tcfv_push_prio specified */
78 if (p
->tcfv_push_prio
) {
79 tci
&= ~VLAN_PRIO_MASK
;
80 tci
|= p
->tcfv_push_prio
<< VLAN_PRIO_SHIFT
;
82 /* put updated tci as hwaccel tag */
83 __vlan_hwaccel_put_tag(skb
, p
->tcfv_push_proto
, tci
);
93 qstats_drop_inc(this_cpu_ptr(v
->common
.cpu_qstats
));
97 if (skb_at_tc_ingress(skb
))
98 skb_pull_rcsum(skb
, skb
->mac_len
);
103 static const struct nla_policy vlan_policy
[TCA_VLAN_MAX
+ 1] = {
104 [TCA_VLAN_PARMS
] = { .len
= sizeof(struct tc_vlan
) },
105 [TCA_VLAN_PUSH_VLAN_ID
] = { .type
= NLA_U16
},
106 [TCA_VLAN_PUSH_VLAN_PROTOCOL
] = { .type
= NLA_U16
},
107 [TCA_VLAN_PUSH_VLAN_PRIORITY
] = { .type
= NLA_U8
},
110 static int tcf_vlan_init(struct net
*net
, struct nlattr
*nla
,
111 struct nlattr
*est
, struct tc_action
**a
,
114 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
115 struct nlattr
*tb
[TCA_VLAN_MAX
+ 1];
116 struct tcf_vlan_params
*p
, *p_old
;
117 struct tc_vlan
*parm
;
121 __be16 push_proto
= 0;
129 err
= nla_parse_nested(tb
, TCA_VLAN_MAX
, nla
, vlan_policy
, NULL
);
133 if (!tb
[TCA_VLAN_PARMS
])
135 parm
= nla_data(tb
[TCA_VLAN_PARMS
]);
136 exists
= tcf_idr_check(tn
, parm
->index
, a
, bind
);
140 switch (parm
->v_action
) {
141 case TCA_VLAN_ACT_POP
:
143 case TCA_VLAN_ACT_PUSH
:
144 case TCA_VLAN_ACT_MODIFY
:
145 if (!tb
[TCA_VLAN_PUSH_VLAN_ID
]) {
147 tcf_idr_release(*a
, bind
);
150 push_vid
= nla_get_u16(tb
[TCA_VLAN_PUSH_VLAN_ID
]);
151 if (push_vid
>= VLAN_VID_MASK
) {
153 tcf_idr_release(*a
, bind
);
157 if (tb
[TCA_VLAN_PUSH_VLAN_PROTOCOL
]) {
158 push_proto
= nla_get_be16(tb
[TCA_VLAN_PUSH_VLAN_PROTOCOL
]);
159 switch (push_proto
) {
160 case htons(ETH_P_8021Q
):
161 case htons(ETH_P_8021AD
):
165 tcf_idr_release(*a
, bind
);
166 return -EPROTONOSUPPORT
;
169 push_proto
= htons(ETH_P_8021Q
);
172 if (tb
[TCA_VLAN_PUSH_VLAN_PRIORITY
])
173 push_prio
= nla_get_u8(tb
[TCA_VLAN_PUSH_VLAN_PRIORITY
]);
177 tcf_idr_release(*a
, bind
);
180 action
= parm
->v_action
;
183 ret
= tcf_idr_create(tn
, parm
->index
, est
, a
,
184 &act_vlan_ops
, bind
, true);
190 tcf_idr_release(*a
, bind
);
198 p
= kzalloc(sizeof(*p
), GFP_KERNEL
);
201 tcf_idr_release(*a
, bind
);
205 v
->tcf_action
= parm
->action
;
207 p_old
= rtnl_dereference(v
->vlan_p
);
209 p
->tcfv_action
= action
;
210 p
->tcfv_push_vid
= push_vid
;
211 p
->tcfv_push_prio
= push_prio
;
212 p
->tcfv_push_proto
= push_proto
;
214 rcu_assign_pointer(v
->vlan_p
, p
);
217 kfree_rcu(p_old
, rcu
);
219 if (ret
== ACT_P_CREATED
)
220 tcf_idr_insert(tn
, *a
);
224 static void tcf_vlan_cleanup(struct tc_action
*a
, int bind
)
226 struct tcf_vlan
*v
= to_vlan(a
);
227 struct tcf_vlan_params
*p
;
229 p
= rcu_dereference_protected(v
->vlan_p
, 1);
234 static int tcf_vlan_dump(struct sk_buff
*skb
, struct tc_action
*a
,
237 unsigned char *b
= skb_tail_pointer(skb
);
238 struct tcf_vlan
*v
= to_vlan(a
);
239 struct tcf_vlan_params
*p
= rtnl_dereference(v
->vlan_p
);
240 struct tc_vlan opt
= {
241 .index
= v
->tcf_index
,
242 .refcnt
= v
->tcf_refcnt
- ref
,
243 .bindcnt
= v
->tcf_bindcnt
- bind
,
244 .action
= v
->tcf_action
,
245 .v_action
= p
->tcfv_action
,
249 if (nla_put(skb
, TCA_VLAN_PARMS
, sizeof(opt
), &opt
))
250 goto nla_put_failure
;
252 if ((p
->tcfv_action
== TCA_VLAN_ACT_PUSH
||
253 p
->tcfv_action
== TCA_VLAN_ACT_MODIFY
) &&
254 (nla_put_u16(skb
, TCA_VLAN_PUSH_VLAN_ID
, p
->tcfv_push_vid
) ||
255 nla_put_be16(skb
, TCA_VLAN_PUSH_VLAN_PROTOCOL
,
256 p
->tcfv_push_proto
) ||
257 (nla_put_u8(skb
, TCA_VLAN_PUSH_VLAN_PRIORITY
,
258 p
->tcfv_push_prio
))))
259 goto nla_put_failure
;
261 tcf_tm_dump(&t
, &v
->tcf_tm
);
262 if (nla_put_64bit(skb
, TCA_VLAN_TM
, sizeof(t
), &t
, TCA_VLAN_PAD
))
263 goto nla_put_failure
;
271 static int tcf_vlan_walker(struct net
*net
, struct sk_buff
*skb
,
272 struct netlink_callback
*cb
, int type
,
273 const struct tc_action_ops
*ops
)
275 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
277 return tcf_generic_walker(tn
, skb
, cb
, type
, ops
);
280 static int tcf_vlan_search(struct net
*net
, struct tc_action
**a
, u32 index
)
282 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
284 return tcf_idr_search(tn
, a
, index
);
287 static struct tc_action_ops act_vlan_ops
= {
289 .type
= TCA_ACT_VLAN
,
290 .owner
= THIS_MODULE
,
292 .dump
= tcf_vlan_dump
,
293 .init
= tcf_vlan_init
,
294 .cleanup
= tcf_vlan_cleanup
,
295 .walk
= tcf_vlan_walker
,
296 .lookup
= tcf_vlan_search
,
297 .size
= sizeof(struct tcf_vlan
),
300 static __net_init
int vlan_init_net(struct net
*net
)
302 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
304 return tc_action_net_init(tn
, &act_vlan_ops
);
307 static void __net_exit
vlan_exit_net(struct net
*net
)
309 struct tc_action_net
*tn
= net_generic(net
, vlan_net_id
);
311 tc_action_net_exit(tn
);
314 static struct pernet_operations vlan_net_ops
= {
315 .init
= vlan_init_net
,
316 .exit
= vlan_exit_net
,
318 .size
= sizeof(struct tc_action_net
),
321 static int __init
vlan_init_module(void)
323 return tcf_register_action(&act_vlan_ops
, &vlan_net_ops
);
326 static void __exit
vlan_cleanup_module(void)
328 tcf_unregister_action(&act_vlan_ops
, &vlan_net_ops
);
331 module_init(vlan_init_module
);
332 module_exit(vlan_cleanup_module
);
334 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
335 MODULE_DESCRIPTION("vlan manipulation actions");
336 MODULE_LICENSE("GPL v2");