Commit 0190c1d4 authored by Vlad Buslov's avatar Vlad Buslov Committed by David S. Miller
Browse files

net: sched: atomically check-allocate action 



Implement function that atomically checks if action exists and either takes
reference to it, or allocates idr slot for action index to prevent
concurrent allocations of actions with same index. Use EBUSY error pointer
to indicate that idr slot is reserved.

Implement cleanup helper function that removes temporary error pointer from
idr. (in case of error between idr allocation and insertion of newly
created action to specified index)

Refactor all action init functions to insert new action to idr using this
API.

Reviewed-by: default avatarMarcelo Ricardo Leitner <marcelo.leitner@gmail.com>
Signed-off-by: default avatarVlad Buslov <vladbu@mellanox.com>
Signed-off-by: default avatarJiri Pirko <jiri@mellanox.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent cae422f3

include/net/act_api.h

+3 −0
Original line number Diff line number Diff line
@@ -154,6 +154,9 @@ int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, 
		   int bind, bool cpustats);

void tcf_idr_insert(struct tc_action_net *tn, struct tc_action *a);



void tcf_idr_cleanup(struct tc_action_net *tn, u32 index);

int tcf_idr_check_alloc(struct tc_action_net *tn, u32 *index,

			struct tc_action **a, int bind);

int tcf_idr_delete_index(struct tc_action_net *tn, u32 index);

int __tcf_idr_release(struct tc_action *a, bool bind, bool strict);

net/sched/act_api.c

+72 −20
Original line number Diff line number Diff line
@@ -303,7 +303,9 @@ static bool __tcf_idr_check(struct tc_action_net *tn, u32 index, 


	spin_lock(&idrinfo->lock);

	p = idr_find(&idrinfo->action_idr, index);

	if (p) {

	if (IS_ERR(p)) {

		p = NULL;

	} else if (p) {

		refcount_inc(&p->tcfa_refcnt);

		if (bind)

			atomic_inc(&p->tcfa_bindcnt);
@@ -371,7 +373,6 @@ int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, 
{

	struct tc_action *p = kzalloc(ops->size, GFP_KERNEL);

	struct tcf_idrinfo *idrinfo = tn->idrinfo;

	struct idr *idr = &idrinfo->action_idr;

	int err = -ENOMEM;



	if (unlikely(!p))
@@ -389,20 +390,6 @@ int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, 
			goto err2;

	}

	spin_lock_init(&p->tcfa_lock);

	idr_preload(GFP_KERNEL);

	spin_lock(&idrinfo->lock);

	/* user doesn't specify an index */

	if (!index) {

		index = 1;

		err = idr_alloc_u32(idr, NULL, &index, UINT_MAX, GFP_ATOMIC);

	} else {

		err = idr_alloc_u32(idr, NULL, &index, index, GFP_ATOMIC);

	}

	spin_unlock(&idrinfo->lock);

	idr_preload_end();

	if (err)

		goto err3;



	p->tcfa_index = index;

	p->tcfa_tm.install = jiffies;

	p->tcfa_tm.lastuse = jiffies;
@@ -412,7 +399,7 @@ int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, 
					&p->tcfa_rate_est,

					&p->tcfa_lock, NULL, est);

		if (err)

			goto err4;

			goto err3;

	}



	p->idrinfo = idrinfo;
@@ -420,8 +407,6 @@ int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, 
	INIT_LIST_HEAD(&p->list);

	*a = p;

	return 0;

err4:

	idr_remove(idr, index);

err3:

	free_percpu(p->cpu_qstats);

err2:
@@ -437,11 +422,78 @@ void tcf_idr_insert(struct tc_action_net *tn, struct tc_action *a) 
	struct tcf_idrinfo *idrinfo = tn->idrinfo;



	spin_lock(&idrinfo->lock);

	idr_replace(&idrinfo->action_idr, a, a->tcfa_index);

	/* Replace ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc */

	WARN_ON(!IS_ERR(idr_replace(&idrinfo->action_idr, a, a->tcfa_index)));

	spin_unlock(&idrinfo->lock);

}

EXPORT_SYMBOL(tcf_idr_insert);



/* Cleanup idr index that was allocated but not initialized. */



void tcf_idr_cleanup(struct tc_action_net *tn, u32 index)

{

	struct tcf_idrinfo *idrinfo = tn->idrinfo;



	spin_lock(&idrinfo->lock);

	/* Remove ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc */

	WARN_ON(!IS_ERR(idr_remove(&idrinfo->action_idr, index)));

	spin_unlock(&idrinfo->lock);

}

EXPORT_SYMBOL(tcf_idr_cleanup);



/* Check if action with specified index exists. If actions is found, increments

 * its reference and bind counters, and return 1. Otherwise insert temporary

 * error pointer (to prevent concurrent users from inserting actions with same

 * index) and return 0.

 */



int tcf_idr_check_alloc(struct tc_action_net *tn, u32 *index,

			struct tc_action **a, int bind)

{

	struct tcf_idrinfo *idrinfo = tn->idrinfo;

	struct tc_action *p;

	int ret;



again:

	spin_lock(&idrinfo->lock);

	if (*index) {

		p = idr_find(&idrinfo->action_idr, *index);

		if (IS_ERR(p)) {

			/* This means that another process allocated

			 * index but did not assign the pointer yet.

			 */

			spin_unlock(&idrinfo->lock);

			goto again;

		}



		if (p) {

			refcount_inc(&p->tcfa_refcnt);

			if (bind)

				atomic_inc(&p->tcfa_bindcnt);

			*a = p;

			ret = 1;

		} else {

			*a = NULL;

			ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index,

					    *index, GFP_ATOMIC);

			if (!ret)

				idr_replace(&idrinfo->action_idr,

					    ERR_PTR(-EBUSY), *index);

		}

	} else {

		*index = 1;

		*a = NULL;

		ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index,

				    UINT_MAX, GFP_ATOMIC);

		if (!ret)

			idr_replace(&idrinfo->action_idr, ERR_PTR(-EBUSY),

				    *index);

	}

	spin_unlock(&idrinfo->lock);

	return ret;

}

EXPORT_SYMBOL(tcf_idr_check_alloc);



void tcf_idrinfo_destroy(const struct tc_action_ops *ops,

			 struct tcf_idrinfo *idrinfo)

{

net/sched/act_bpf.c

+8 −3
Original line number Diff line number Diff line
@@ -299,14 +299,17 @@ static int tcf_bpf_init(struct net *net, struct nlattr *nla, 


	parm = nla_data(tb[TCA_ACT_BPF_PARMS]);



	if (!tcf_idr_check(tn, parm->index, act, bind)) {

	ret = tcf_idr_check_alloc(tn, &parm->index, act, bind);

	if (!ret) {

		ret = tcf_idr_create(tn, parm->index, est, act,

				     &act_bpf_ops, bind, true);

		if (ret < 0)

		if (ret < 0) {

			tcf_idr_cleanup(tn, parm->index);

			return ret;

		}



		res = ACT_P_CREATED;

	} else {

	} else if (ret > 0) {

		/* Don't override defaults. */

		if (bind)

			return 0;
@@ -315,6 +318,8 @@ static int tcf_bpf_init(struct net *net, struct nlattr *nla, 
			tcf_idr_release(*act, bind);

			return -EEXIST;

		}

	} else {

		return ret;

	}



	is_bpf = tb[TCA_ACT_BPF_OPS_LEN] && tb[TCA_ACT_BPF_OPS];

net/sched/act_connmark.c

+7 −3
Original line number Diff line number Diff line
@@ -118,11 +118,14 @@ static int tcf_connmark_init(struct net *net, struct nlattr *nla, 


	parm = nla_data(tb[TCA_CONNMARK_PARMS]);



	if (!tcf_idr_check(tn, parm->index, a, bind)) {

	ret = tcf_idr_check_alloc(tn, &parm->index, a, bind);

	if (!ret) {

		ret = tcf_idr_create(tn, parm->index, est, a,

				     &act_connmark_ops, bind, false);

		if (ret)

		if (ret) {

			tcf_idr_cleanup(tn, parm->index);

			return ret;

		}



		ci = to_connmark(*a);

		ci->tcf_action = parm->action;
@@ -131,7 +134,7 @@ static int tcf_connmark_init(struct net *net, struct nlattr *nla, 


		tcf_idr_insert(tn, *a);

		ret = ACT_P_CREATED;

	} else {

	} else if (ret > 0) {

		ci = to_connmark(*a);

		if (bind)

			return 0;
@@ -142,6 +145,7 @@ static int tcf_connmark_init(struct net *net, struct nlattr *nla, 
		/* replacing action and zone */

		ci->tcf_action = parm->action;

		ci->zone = parm->zone;

		ret = 0;

	}



	return ret;

net/sched/act_csum.c

+8 −3
Original line number Diff line number Diff line
@@ -67,19 +67,24 @@ static int tcf_csum_init(struct net *net, struct nlattr *nla, 
		return -EINVAL;

	parm = nla_data(tb[TCA_CSUM_PARMS]);



	if (!tcf_idr_check(tn, parm->index, a, bind)) {

	err = tcf_idr_check_alloc(tn, &parm->index, a, bind);

	if (!err) {

		ret = tcf_idr_create(tn, parm->index, est, a,

				     &act_csum_ops, bind, true);

		if (ret)

		if (ret) {

			tcf_idr_cleanup(tn, parm->index);

			return ret;

		}

		ret = ACT_P_CREATED;

	} else {

	} else if (err > 0) {

		if (bind)/* dont override defaults */

			return 0;

		if (!ovr) {

			tcf_idr_release(*a, bind);

			return -EEXIST;

		}

	} else {

		return err;

	}



	p = to_tcf_csum(*a);

CRA Git | Maintained and supported by SUSTech CRA and CCSE