Linux Audio

Check our new training course

Loading...
v4.17
 
  1/*
  2 * Copyright (c) 2008, Intel Corporation.
  3 *
  4 * This program is free software; you can redistribute it and/or modify it
  5 * under the terms and conditions of the GNU General Public License,
  6 * version 2, as published by the Free Software Foundation.
  7 *
  8 * This program is distributed in the hope it will be useful, but WITHOUT
  9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 10 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 11 * more details.
 12 *
 13 * You should have received a copy of the GNU General Public License along with
 14 * this program; if not, see <http://www.gnu.org/licenses/>.
 15 *
 16 * Author: Alexander Duyck <alexander.h.duyck@intel.com>
 17 */
 18
 19#include <linux/module.h>
 20#include <linux/init.h>
 21#include <linux/kernel.h>
 22#include <linux/skbuff.h>
 23#include <linux/rtnetlink.h>
 24#include <net/netlink.h>
 25#include <net/pkt_sched.h>
 
 
 
 
 26
 27#include <linux/tc_act/tc_skbedit.h>
 28#include <net/tc_act/tc_skbedit.h>
 29
 30static unsigned int skbedit_net_id;
 31static struct tc_action_ops act_skbedit_ops;
 32
 33static int tcf_skbedit(struct sk_buff *skb, const struct tc_action *a,
 34		       struct tcf_result *res)
 35{
 36	struct tcf_skbedit *d = to_skbedit(a);
 
 
 37
 38	spin_lock(&d->tcf_lock);
 39	tcf_lastuse_update(&d->tcf_tm);
 40	bstats_update(&d->tcf_bstats, skb);
 41
 42	if (d->flags & SKBEDIT_F_PRIORITY)
 43		skb->priority = d->priority;
 44	if (d->flags & SKBEDIT_F_QUEUE_MAPPING &&
 45	    skb->dev->real_num_tx_queues > d->queue_mapping)
 46		skb_set_queue_mapping(skb, d->queue_mapping);
 47	if (d->flags & SKBEDIT_F_MARK) {
 48		skb->mark &= ~d->mask;
 49		skb->mark |= d->mark & d->mask;
 50	}
 51	if (d->flags & SKBEDIT_F_PTYPE)
 52		skb->pkt_type = d->ptype;
 53
 54	spin_unlock(&d->tcf_lock);
 55	return d->tcf_action;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 56}
 57
 58static const struct nla_policy skbedit_policy[TCA_SKBEDIT_MAX + 1] = {
 59	[TCA_SKBEDIT_PARMS]		= { .len = sizeof(struct tc_skbedit) },
 60	[TCA_SKBEDIT_PRIORITY]		= { .len = sizeof(u32) },
 61	[TCA_SKBEDIT_QUEUE_MAPPING]	= { .len = sizeof(u16) },
 62	[TCA_SKBEDIT_MARK]		= { .len = sizeof(u32) },
 63	[TCA_SKBEDIT_PTYPE]		= { .len = sizeof(u16) },
 64	[TCA_SKBEDIT_MASK]		= { .len = sizeof(u32) },
 
 65};
 66
 67static int tcf_skbedit_init(struct net *net, struct nlattr *nla,
 68			    struct nlattr *est, struct tc_action **a,
 69			    int ovr, int bind, struct netlink_ext_ack *extack)
 
 
 70{
 71	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
 
 72	struct nlattr *tb[TCA_SKBEDIT_MAX + 1];
 
 73	struct tc_skbedit *parm;
 74	struct tcf_skbedit *d;
 75	u32 flags = 0, *priority = NULL, *mark = NULL, *mask = NULL;
 76	u16 *queue_mapping = NULL, *ptype = NULL;
 77	bool exists = false;
 78	int ret = 0, err;
 
 79
 80	if (nla == NULL)
 81		return -EINVAL;
 82
 83	err = nla_parse_nested(tb, TCA_SKBEDIT_MAX, nla, skbedit_policy, NULL);
 
 84	if (err < 0)
 85		return err;
 86
 87	if (tb[TCA_SKBEDIT_PARMS] == NULL)
 88		return -EINVAL;
 89
 90	if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
 91		flags |= SKBEDIT_F_PRIORITY;
 92		priority = nla_data(tb[TCA_SKBEDIT_PRIORITY]);
 93	}
 94
 95	if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
 96		flags |= SKBEDIT_F_QUEUE_MAPPING;
 97		queue_mapping = nla_data(tb[TCA_SKBEDIT_QUEUE_MAPPING]);
 98	}
 99
100	if (tb[TCA_SKBEDIT_PTYPE] != NULL) {
101		ptype = nla_data(tb[TCA_SKBEDIT_PTYPE]);
102		if (!skb_pkt_type_ok(*ptype))
103			return -EINVAL;
104		flags |= SKBEDIT_F_PTYPE;
105	}
106
107	if (tb[TCA_SKBEDIT_MARK] != NULL) {
108		flags |= SKBEDIT_F_MARK;
109		mark = nla_data(tb[TCA_SKBEDIT_MARK]);
110	}
111
112	if (tb[TCA_SKBEDIT_MASK] != NULL) {
113		flags |= SKBEDIT_F_MASK;
114		mask = nla_data(tb[TCA_SKBEDIT_MASK]);
115	}
116
117	parm = nla_data(tb[TCA_SKBEDIT_PARMS]);
 
 
 
 
 
118
119	exists = tcf_idr_check(tn, parm->index, a, bind);
 
 
 
 
 
120	if (exists && bind)
121		return 0;
122
123	if (!flags) {
124		if (exists)
125			tcf_idr_release(*a, bind);
 
 
126		return -EINVAL;
127	}
128
129	if (!exists) {
130		ret = tcf_idr_create(tn, parm->index, est, a,
131				     &act_skbedit_ops, bind, false);
132		if (ret)
 
133			return ret;
 
134
135		d = to_skbedit(*a);
136		ret = ACT_P_CREATED;
137	} else {
138		d = to_skbedit(*a);
139		tcf_idr_release(*a, bind);
140		if (!ovr)
141			return -EEXIST;
 
142	}
 
 
 
143
144	spin_lock_bh(&d->tcf_lock);
 
 
 
 
145
146	d->flags = flags;
147	if (flags & SKBEDIT_F_PRIORITY)
148		d->priority = *priority;
149	if (flags & SKBEDIT_F_QUEUE_MAPPING)
150		d->queue_mapping = *queue_mapping;
151	if (flags & SKBEDIT_F_MARK)
152		d->mark = *mark;
153	if (flags & SKBEDIT_F_PTYPE)
154		d->ptype = *ptype;
155	/* default behaviour is to use all the bits */
156	d->mask = 0xffffffff;
157	if (flags & SKBEDIT_F_MASK)
158		d->mask = *mask;
159
160	d->tcf_action = parm->action;
161
 
 
 
 
162	spin_unlock_bh(&d->tcf_lock);
 
 
 
 
163
164	if (ret == ACT_P_CREATED)
165		tcf_idr_insert(tn, *a);
166	return ret;
 
 
 
 
 
 
167}
168
169static int tcf_skbedit_dump(struct sk_buff *skb, struct tc_action *a,
170			    int bind, int ref)
171{
172	unsigned char *b = skb_tail_pointer(skb);
173	struct tcf_skbedit *d = to_skbedit(a);
 
174	struct tc_skbedit opt = {
175		.index   = d->tcf_index,
176		.refcnt  = d->tcf_refcnt - ref,
177		.bindcnt = d->tcf_bindcnt - bind,
178		.action  = d->tcf_action,
179	};
 
180	struct tcf_t t;
181
 
 
 
 
 
182	if (nla_put(skb, TCA_SKBEDIT_PARMS, sizeof(opt), &opt))
183		goto nla_put_failure;
184	if ((d->flags & SKBEDIT_F_PRIORITY) &&
185	    nla_put_u32(skb, TCA_SKBEDIT_PRIORITY, d->priority))
 
 
 
186		goto nla_put_failure;
187	if ((d->flags & SKBEDIT_F_QUEUE_MAPPING) &&
188	    nla_put_u16(skb, TCA_SKBEDIT_QUEUE_MAPPING, d->queue_mapping))
189		goto nla_put_failure;
190	if ((d->flags & SKBEDIT_F_MARK) &&
191	    nla_put_u32(skb, TCA_SKBEDIT_MARK, d->mark))
192		goto nla_put_failure;
193	if ((d->flags & SKBEDIT_F_PTYPE) &&
194	    nla_put_u16(skb, TCA_SKBEDIT_PTYPE, d->ptype))
195		goto nla_put_failure;
196	if ((d->flags & SKBEDIT_F_MASK) &&
197	    nla_put_u32(skb, TCA_SKBEDIT_MASK, d->mask))
 
 
198		goto nla_put_failure;
199
200	tcf_tm_dump(&t, &d->tcf_tm);
201	if (nla_put_64bit(skb, TCA_SKBEDIT_TM, sizeof(t), &t, TCA_SKBEDIT_PAD))
202		goto nla_put_failure;
 
 
203	return skb->len;
204
205nla_put_failure:
 
206	nlmsg_trim(skb, b);
207	return -1;
208}
209
 
 
 
 
 
 
 
 
 
 
210static int tcf_skbedit_walker(struct net *net, struct sk_buff *skb,
211			      struct netlink_callback *cb, int type,
212			      const struct tc_action_ops *ops,
213			      struct netlink_ext_ack *extack)
214{
215	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
216
217	return tcf_generic_walker(tn, skb, cb, type, ops, extack);
218}
219
220static int tcf_skbedit_search(struct net *net, struct tc_action **a, u32 index,
221			      struct netlink_ext_ack *extack)
222{
223	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
224
225	return tcf_idr_search(tn, a, index);
226}
227
 
 
 
 
 
 
 
 
 
 
 
228static struct tc_action_ops act_skbedit_ops = {
229	.kind		=	"skbedit",
230	.type		=	TCA_ACT_SKBEDIT,
231	.owner		=	THIS_MODULE,
232	.act		=	tcf_skbedit,
233	.dump		=	tcf_skbedit_dump,
234	.init		=	tcf_skbedit_init,
 
235	.walk		=	tcf_skbedit_walker,
 
236	.lookup		=	tcf_skbedit_search,
237	.size		=	sizeof(struct tcf_skbedit),
238};
239
240static __net_init int skbedit_init_net(struct net *net)
241{
242	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
243
244	return tc_action_net_init(tn, &act_skbedit_ops);
245}
246
247static void __net_exit skbedit_exit_net(struct list_head *net_list)
248{
249	tc_action_net_exit(net_list, skbedit_net_id);
250}
251
252static struct pernet_operations skbedit_net_ops = {
253	.init = skbedit_init_net,
254	.exit_batch = skbedit_exit_net,
255	.id   = &skbedit_net_id,
256	.size = sizeof(struct tc_action_net),
257};
258
259MODULE_AUTHOR("Alexander Duyck, <alexander.h.duyck@intel.com>");
260MODULE_DESCRIPTION("SKB Editing");
261MODULE_LICENSE("GPL");
262
263static int __init skbedit_init_module(void)
264{
265	return tcf_register_action(&act_skbedit_ops, &skbedit_net_ops);
266}
267
268static void __exit skbedit_cleanup_module(void)
269{
270	tcf_unregister_action(&act_skbedit_ops, &skbedit_net_ops);
271}
272
273module_init(skbedit_init_module);
274module_exit(skbedit_cleanup_module);
v5.4
  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * Copyright (c) 2008, Intel Corporation.
  4 *
 
 
 
 
 
 
 
 
 
 
 
 
  5 * Author: Alexander Duyck <alexander.h.duyck@intel.com>
  6 */
  7
  8#include <linux/module.h>
  9#include <linux/init.h>
 10#include <linux/kernel.h>
 11#include <linux/skbuff.h>
 12#include <linux/rtnetlink.h>
 13#include <net/netlink.h>
 14#include <net/pkt_sched.h>
 15#include <net/ip.h>
 16#include <net/ipv6.h>
 17#include <net/dsfield.h>
 18#include <net/pkt_cls.h>
 19
 20#include <linux/tc_act/tc_skbedit.h>
 21#include <net/tc_act/tc_skbedit.h>
 22
 23static unsigned int skbedit_net_id;
 24static struct tc_action_ops act_skbedit_ops;
 25
 26static int tcf_skbedit_act(struct sk_buff *skb, const struct tc_action *a,
 27			   struct tcf_result *res)
 28{
 29	struct tcf_skbedit *d = to_skbedit(a);
 30	struct tcf_skbedit_params *params;
 31	int action;
 32
 
 33	tcf_lastuse_update(&d->tcf_tm);
 34	bstats_cpu_update(this_cpu_ptr(d->common.cpu_bstats), skb);
 35
 36	params = rcu_dereference_bh(d->params);
 37	action = READ_ONCE(d->tcf_action);
 
 
 
 
 
 
 
 
 
 38
 39	if (params->flags & SKBEDIT_F_PRIORITY)
 40		skb->priority = params->priority;
 41	if (params->flags & SKBEDIT_F_INHERITDSFIELD) {
 42		int wlen = skb_network_offset(skb);
 43
 44		switch (tc_skb_protocol(skb)) {
 45		case htons(ETH_P_IP):
 46			wlen += sizeof(struct iphdr);
 47			if (!pskb_may_pull(skb, wlen))
 48				goto err;
 49			skb->priority = ipv4_get_dsfield(ip_hdr(skb)) >> 2;
 50			break;
 51
 52		case htons(ETH_P_IPV6):
 53			wlen += sizeof(struct ipv6hdr);
 54			if (!pskb_may_pull(skb, wlen))
 55				goto err;
 56			skb->priority = ipv6_get_dsfield(ipv6_hdr(skb)) >> 2;
 57			break;
 58		}
 59	}
 60	if (params->flags & SKBEDIT_F_QUEUE_MAPPING &&
 61	    skb->dev->real_num_tx_queues > params->queue_mapping)
 62		skb_set_queue_mapping(skb, params->queue_mapping);
 63	if (params->flags & SKBEDIT_F_MARK) {
 64		skb->mark &= ~params->mask;
 65		skb->mark |= params->mark & params->mask;
 66	}
 67	if (params->flags & SKBEDIT_F_PTYPE)
 68		skb->pkt_type = params->ptype;
 69	return action;
 70
 71err:
 72	qstats_drop_inc(this_cpu_ptr(d->common.cpu_qstats));
 73	return TC_ACT_SHOT;
 74}
 75
 76static const struct nla_policy skbedit_policy[TCA_SKBEDIT_MAX + 1] = {
 77	[TCA_SKBEDIT_PARMS]		= { .len = sizeof(struct tc_skbedit) },
 78	[TCA_SKBEDIT_PRIORITY]		= { .len = sizeof(u32) },
 79	[TCA_SKBEDIT_QUEUE_MAPPING]	= { .len = sizeof(u16) },
 80	[TCA_SKBEDIT_MARK]		= { .len = sizeof(u32) },
 81	[TCA_SKBEDIT_PTYPE]		= { .len = sizeof(u16) },
 82	[TCA_SKBEDIT_MASK]		= { .len = sizeof(u32) },
 83	[TCA_SKBEDIT_FLAGS]		= { .len = sizeof(u64) },
 84};
 85
 86static int tcf_skbedit_init(struct net *net, struct nlattr *nla,
 87			    struct nlattr *est, struct tc_action **a,
 88			    int ovr, int bind, bool rtnl_held,
 89			    struct tcf_proto *tp,
 90			    struct netlink_ext_ack *extack)
 91{
 92	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
 93	struct tcf_skbedit_params *params_new;
 94	struct nlattr *tb[TCA_SKBEDIT_MAX + 1];
 95	struct tcf_chain *goto_ch = NULL;
 96	struct tc_skbedit *parm;
 97	struct tcf_skbedit *d;
 98	u32 flags = 0, *priority = NULL, *mark = NULL, *mask = NULL;
 99	u16 *queue_mapping = NULL, *ptype = NULL;
100	bool exists = false;
101	int ret = 0, err;
102	u32 index;
103
104	if (nla == NULL)
105		return -EINVAL;
106
107	err = nla_parse_nested_deprecated(tb, TCA_SKBEDIT_MAX, nla,
108					  skbedit_policy, NULL);
109	if (err < 0)
110		return err;
111
112	if (tb[TCA_SKBEDIT_PARMS] == NULL)
113		return -EINVAL;
114
115	if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
116		flags |= SKBEDIT_F_PRIORITY;
117		priority = nla_data(tb[TCA_SKBEDIT_PRIORITY]);
118	}
119
120	if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
121		flags |= SKBEDIT_F_QUEUE_MAPPING;
122		queue_mapping = nla_data(tb[TCA_SKBEDIT_QUEUE_MAPPING]);
123	}
124
125	if (tb[TCA_SKBEDIT_PTYPE] != NULL) {
126		ptype = nla_data(tb[TCA_SKBEDIT_PTYPE]);
127		if (!skb_pkt_type_ok(*ptype))
128			return -EINVAL;
129		flags |= SKBEDIT_F_PTYPE;
130	}
131
132	if (tb[TCA_SKBEDIT_MARK] != NULL) {
133		flags |= SKBEDIT_F_MARK;
134		mark = nla_data(tb[TCA_SKBEDIT_MARK]);
135	}
136
137	if (tb[TCA_SKBEDIT_MASK] != NULL) {
138		flags |= SKBEDIT_F_MASK;
139		mask = nla_data(tb[TCA_SKBEDIT_MASK]);
140	}
141
142	if (tb[TCA_SKBEDIT_FLAGS] != NULL) {
143		u64 *pure_flags = nla_data(tb[TCA_SKBEDIT_FLAGS]);
144
145		if (*pure_flags & SKBEDIT_F_INHERITDSFIELD)
146			flags |= SKBEDIT_F_INHERITDSFIELD;
147	}
148
149	parm = nla_data(tb[TCA_SKBEDIT_PARMS]);
150	index = parm->index;
151	err = tcf_idr_check_alloc(tn, &index, a, bind);
152	if (err < 0)
153		return err;
154	exists = err;
155	if (exists && bind)
156		return 0;
157
158	if (!flags) {
159		if (exists)
160			tcf_idr_release(*a, bind);
161		else
162			tcf_idr_cleanup(tn, index);
163		return -EINVAL;
164	}
165
166	if (!exists) {
167		ret = tcf_idr_create(tn, index, est, a,
168				     &act_skbedit_ops, bind, true);
169		if (ret) {
170			tcf_idr_cleanup(tn, index);
171			return ret;
172		}
173
174		d = to_skbedit(*a);
175		ret = ACT_P_CREATED;
176	} else {
177		d = to_skbedit(*a);
178		if (!ovr) {
179			tcf_idr_release(*a, bind);
180			return -EEXIST;
181		}
182	}
183	err = tcf_action_check_ctrlact(parm->action, tp, &goto_ch, extack);
184	if (err < 0)
185		goto release_idr;
186
187	params_new = kzalloc(sizeof(*params_new), GFP_KERNEL);
188	if (unlikely(!params_new)) {
189		err = -ENOMEM;
190		goto put_chain;
191	}
192
193	params_new->flags = flags;
194	if (flags & SKBEDIT_F_PRIORITY)
195		params_new->priority = *priority;
196	if (flags & SKBEDIT_F_QUEUE_MAPPING)
197		params_new->queue_mapping = *queue_mapping;
198	if (flags & SKBEDIT_F_MARK)
199		params_new->mark = *mark;
200	if (flags & SKBEDIT_F_PTYPE)
201		params_new->ptype = *ptype;
202	/* default behaviour is to use all the bits */
203	params_new->mask = 0xffffffff;
204	if (flags & SKBEDIT_F_MASK)
205		params_new->mask = *mask;
 
 
206
207	spin_lock_bh(&d->tcf_lock);
208	goto_ch = tcf_action_set_ctrlact(*a, parm->action, goto_ch);
209	rcu_swap_protected(d->params, params_new,
210			   lockdep_is_held(&d->tcf_lock));
211	spin_unlock_bh(&d->tcf_lock);
212	if (params_new)
213		kfree_rcu(params_new, rcu);
214	if (goto_ch)
215		tcf_chain_put_by_act(goto_ch);
216
217	if (ret == ACT_P_CREATED)
218		tcf_idr_insert(tn, *a);
219	return ret;
220put_chain:
221	if (goto_ch)
222		tcf_chain_put_by_act(goto_ch);
223release_idr:
224	tcf_idr_release(*a, bind);
225	return err;
226}
227
228static int tcf_skbedit_dump(struct sk_buff *skb, struct tc_action *a,
229			    int bind, int ref)
230{
231	unsigned char *b = skb_tail_pointer(skb);
232	struct tcf_skbedit *d = to_skbedit(a);
233	struct tcf_skbedit_params *params;
234	struct tc_skbedit opt = {
235		.index   = d->tcf_index,
236		.refcnt  = refcount_read(&d->tcf_refcnt) - ref,
237		.bindcnt = atomic_read(&d->tcf_bindcnt) - bind,
 
238	};
239	u64 pure_flags = 0;
240	struct tcf_t t;
241
242	spin_lock_bh(&d->tcf_lock);
243	params = rcu_dereference_protected(d->params,
244					   lockdep_is_held(&d->tcf_lock));
245	opt.action = d->tcf_action;
246
247	if (nla_put(skb, TCA_SKBEDIT_PARMS, sizeof(opt), &opt))
248		goto nla_put_failure;
249	if ((params->flags & SKBEDIT_F_PRIORITY) &&
250	    nla_put_u32(skb, TCA_SKBEDIT_PRIORITY, params->priority))
251		goto nla_put_failure;
252	if ((params->flags & SKBEDIT_F_QUEUE_MAPPING) &&
253	    nla_put_u16(skb, TCA_SKBEDIT_QUEUE_MAPPING, params->queue_mapping))
254		goto nla_put_failure;
255	if ((params->flags & SKBEDIT_F_MARK) &&
256	    nla_put_u32(skb, TCA_SKBEDIT_MARK, params->mark))
257		goto nla_put_failure;
258	if ((params->flags & SKBEDIT_F_PTYPE) &&
259	    nla_put_u16(skb, TCA_SKBEDIT_PTYPE, params->ptype))
260		goto nla_put_failure;
261	if ((params->flags & SKBEDIT_F_MASK) &&
262	    nla_put_u32(skb, TCA_SKBEDIT_MASK, params->mask))
263		goto nla_put_failure;
264	if (params->flags & SKBEDIT_F_INHERITDSFIELD)
265		pure_flags |= SKBEDIT_F_INHERITDSFIELD;
266	if (pure_flags != 0 &&
267	    nla_put(skb, TCA_SKBEDIT_FLAGS, sizeof(pure_flags), &pure_flags))
268		goto nla_put_failure;
269
270	tcf_tm_dump(&t, &d->tcf_tm);
271	if (nla_put_64bit(skb, TCA_SKBEDIT_TM, sizeof(t), &t, TCA_SKBEDIT_PAD))
272		goto nla_put_failure;
273	spin_unlock_bh(&d->tcf_lock);
274
275	return skb->len;
276
277nla_put_failure:
278	spin_unlock_bh(&d->tcf_lock);
279	nlmsg_trim(skb, b);
280	return -1;
281}
282
283static void tcf_skbedit_cleanup(struct tc_action *a)
284{
285	struct tcf_skbedit *d = to_skbedit(a);
286	struct tcf_skbedit_params *params;
287
288	params = rcu_dereference_protected(d->params, 1);
289	if (params)
290		kfree_rcu(params, rcu);
291}
292
293static int tcf_skbedit_walker(struct net *net, struct sk_buff *skb,
294			      struct netlink_callback *cb, int type,
295			      const struct tc_action_ops *ops,
296			      struct netlink_ext_ack *extack)
297{
298	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
299
300	return tcf_generic_walker(tn, skb, cb, type, ops, extack);
301}
302
303static int tcf_skbedit_search(struct net *net, struct tc_action **a, u32 index)
 
304{
305	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
306
307	return tcf_idr_search(tn, a, index);
308}
309
310static size_t tcf_skbedit_get_fill_size(const struct tc_action *act)
311{
312	return nla_total_size(sizeof(struct tc_skbedit))
313		+ nla_total_size(sizeof(u32)) /* TCA_SKBEDIT_PRIORITY */
314		+ nla_total_size(sizeof(u16)) /* TCA_SKBEDIT_QUEUE_MAPPING */
315		+ nla_total_size(sizeof(u32)) /* TCA_SKBEDIT_MARK */
316		+ nla_total_size(sizeof(u16)) /* TCA_SKBEDIT_PTYPE */
317		+ nla_total_size(sizeof(u32)) /* TCA_SKBEDIT_MASK */
318		+ nla_total_size_64bit(sizeof(u64)); /* TCA_SKBEDIT_FLAGS */
319}
320
321static struct tc_action_ops act_skbedit_ops = {
322	.kind		=	"skbedit",
323	.id		=	TCA_ID_SKBEDIT,
324	.owner		=	THIS_MODULE,
325	.act		=	tcf_skbedit_act,
326	.dump		=	tcf_skbedit_dump,
327	.init		=	tcf_skbedit_init,
328	.cleanup	=	tcf_skbedit_cleanup,
329	.walk		=	tcf_skbedit_walker,
330	.get_fill_size	=	tcf_skbedit_get_fill_size,
331	.lookup		=	tcf_skbedit_search,
332	.size		=	sizeof(struct tcf_skbedit),
333};
334
335static __net_init int skbedit_init_net(struct net *net)
336{
337	struct tc_action_net *tn = net_generic(net, skbedit_net_id);
338
339	return tc_action_net_init(net, tn, &act_skbedit_ops);
340}
341
342static void __net_exit skbedit_exit_net(struct list_head *net_list)
343{
344	tc_action_net_exit(net_list, skbedit_net_id);
345}
346
347static struct pernet_operations skbedit_net_ops = {
348	.init = skbedit_init_net,
349	.exit_batch = skbedit_exit_net,
350	.id   = &skbedit_net_id,
351	.size = sizeof(struct tc_action_net),
352};
353
354MODULE_AUTHOR("Alexander Duyck, <alexander.h.duyck@intel.com>");
355MODULE_DESCRIPTION("SKB Editing");
356MODULE_LICENSE("GPL");
357
358static int __init skbedit_init_module(void)
359{
360	return tcf_register_action(&act_skbedit_ops, &skbedit_net_ops);
361}
362
363static void __exit skbedit_cleanup_module(void)
364{
365	tcf_unregister_action(&act_skbedit_ops, &skbedit_net_ops);
366}
367
368module_init(skbedit_init_module);
369module_exit(skbedit_cleanup_module);