| Jamal Hadi Salim | db75307 | 2005-04-24 20:10:16 -0700 | [diff] [blame] | 1 | /* | 
 | 2 |  * include/net/act_generic.h | 
 | 3 |  * | 
 | 4 | */ | 
| Jamal Hadi Salim | 1eda339 | 2005-05-19 12:42:39 -0700 | [diff] [blame] | 5 | #ifndef _NET_ACT_GENERIC_H | 
 | 6 | #define _NET_ACT_GENERIC_H | 
| Jamal Hadi Salim | db75307 | 2005-04-24 20:10:16 -0700 | [diff] [blame] | 7 | static inline int tcf_defact_release(struct tcf_defact *p, int bind) | 
 | 8 | { | 
 | 9 | 	int ret = 0; | 
 | 10 | 	if (p) { | 
 | 11 | 		if (bind) { | 
 | 12 | 			p->bindcnt--; | 
 | 13 | 		} | 
 | 14 | 		p->refcnt--; | 
 | 15 | 		if (p->bindcnt <= 0 && p->refcnt <= 0) { | 
 | 16 | 			kfree(p->defdata); | 
 | 17 | 			tcf_hash_destroy(p); | 
 | 18 | 			ret = 1; | 
 | 19 | 		} | 
 | 20 | 	} | 
 | 21 | 	return ret; | 
 | 22 | } | 
 | 23 |  | 
 | 24 | static inline int | 
 | 25 | alloc_defdata(struct tcf_defact *p, u32 datalen, void *defdata) | 
 | 26 | { | 
 | 27 | 	p->defdata = kmalloc(datalen, GFP_KERNEL); | 
 | 28 | 	if (p->defdata == NULL) | 
 | 29 | 		return -ENOMEM; | 
 | 30 | 	p->datalen = datalen; | 
 | 31 | 	memcpy(p->defdata, defdata, datalen); | 
 | 32 | 	return 0; | 
 | 33 | } | 
 | 34 |  | 
 | 35 | static inline int | 
 | 36 | realloc_defdata(struct tcf_defact *p, u32 datalen, void *defdata) | 
 | 37 | { | 
 | 38 | 	/* safer to be just brute force for now */ | 
 | 39 | 	kfree(p->defdata); | 
 | 40 | 	return alloc_defdata(p, datalen, defdata); | 
 | 41 | } | 
 | 42 |  | 
 | 43 | static inline int | 
 | 44 | tcf_defact_init(struct rtattr *rta, struct rtattr *est, | 
 | 45 | 		struct tc_action *a, int ovr, int bind) | 
 | 46 | { | 
 | 47 | 	struct rtattr *tb[TCA_DEF_MAX]; | 
 | 48 | 	struct tc_defact *parm; | 
 | 49 | 	struct tcf_defact *p; | 
 | 50 | 	void *defdata; | 
 | 51 | 	u32 datalen = 0; | 
 | 52 | 	int ret = 0; | 
 | 53 |  | 
 | 54 | 	if (rta == NULL || rtattr_parse_nested(tb, TCA_DEF_MAX, rta) < 0) | 
 | 55 | 		return -EINVAL; | 
 | 56 |  | 
 | 57 | 	if (tb[TCA_DEF_PARMS - 1] == NULL ||  | 
 | 58 | 	    RTA_PAYLOAD(tb[TCA_DEF_PARMS - 1]) < sizeof(*parm)) | 
 | 59 | 		return -EINVAL; | 
 | 60 |  | 
 | 61 | 	parm = RTA_DATA(tb[TCA_DEF_PARMS - 1]); | 
 | 62 | 	defdata = RTA_DATA(tb[TCA_DEF_DATA - 1]); | 
 | 63 | 	if (defdata == NULL) | 
 | 64 | 		return -EINVAL; | 
 | 65 |  | 
 | 66 | 	datalen = RTA_PAYLOAD(tb[TCA_DEF_DATA - 1]); | 
 | 67 | 	if (datalen <= 0) | 
 | 68 | 		return -EINVAL; | 
 | 69 |  | 
 | 70 | 	p = tcf_hash_check(parm->index, a, ovr, bind); | 
 | 71 | 	if (p == NULL) { | 
 | 72 | 		p = tcf_hash_create(parm->index, est, a, sizeof(*p), ovr, bind); | 
 | 73 | 		if (p == NULL) | 
 | 74 | 			return -ENOMEM; | 
 | 75 |  | 
 | 76 | 		ret = alloc_defdata(p, datalen, defdata); | 
 | 77 | 		if (ret < 0) { | 
 | 78 | 			kfree(p); | 
 | 79 | 			return ret; | 
 | 80 | 		} | 
 | 81 | 		ret = ACT_P_CREATED; | 
 | 82 | 	} else { | 
 | 83 | 		if (!ovr) { | 
 | 84 | 			tcf_defact_release(p, bind); | 
 | 85 | 			return -EEXIST; | 
 | 86 | 		} | 
 | 87 | 		realloc_defdata(p, datalen, defdata); | 
 | 88 | 	} | 
 | 89 |  | 
 | 90 | 	spin_lock_bh(&p->lock); | 
 | 91 | 	p->action = parm->action; | 
 | 92 | 	spin_unlock_bh(&p->lock); | 
 | 93 | 	if (ret == ACT_P_CREATED) | 
 | 94 | 		tcf_hash_insert(p); | 
 | 95 | 	return ret; | 
 | 96 | } | 
 | 97 |  | 
 | 98 | static inline int tcf_defact_cleanup(struct tc_action *a, int bind) | 
 | 99 | { | 
 | 100 | 	struct tcf_defact *p = PRIV(a, defact); | 
 | 101 |  | 
 | 102 | 	if (p != NULL) | 
 | 103 | 		return tcf_defact_release(p, bind); | 
 | 104 | 	return 0; | 
 | 105 | } | 
 | 106 |  | 
 | 107 | static inline int | 
 | 108 | tcf_defact_dump(struct sk_buff *skb, struct tc_action *a, int bind, int ref) | 
 | 109 | { | 
 | 110 | 	unsigned char *b = skb->tail; | 
 | 111 | 	struct tc_defact opt; | 
 | 112 | 	struct tcf_defact *p = PRIV(a, defact); | 
 | 113 | 	struct tcf_t t; | 
 | 114 |  | 
 | 115 | 	opt.index = p->index; | 
 | 116 | 	opt.refcnt = p->refcnt - ref; | 
 | 117 | 	opt.bindcnt = p->bindcnt - bind; | 
 | 118 | 	opt.action = p->action; | 
 | 119 | 	RTA_PUT(skb, TCA_DEF_PARMS, sizeof(opt), &opt); | 
 | 120 | 	RTA_PUT(skb, TCA_DEF_DATA, p->datalen, p->defdata); | 
 | 121 | 	t.install = jiffies_to_clock_t(jiffies - p->tm.install); | 
 | 122 | 	t.lastuse = jiffies_to_clock_t(jiffies - p->tm.lastuse); | 
 | 123 | 	t.expires = jiffies_to_clock_t(p->tm.expires); | 
 | 124 | 	RTA_PUT(skb, TCA_DEF_TM, sizeof(t), &t); | 
 | 125 | 	return skb->len; | 
 | 126 |  | 
 | 127 | rtattr_failure: | 
 | 128 | 	skb_trim(skb, b - skb->data); | 
 | 129 | 	return -1; | 
 | 130 | } | 
 | 131 |  | 
 | 132 | #define tca_use_default_ops \ | 
 | 133 | 	.dump           =       tcf_defact_dump, \ | 
 | 134 | 	.cleanup        =       tcf_defact_cleanup, \ | 
 | 135 | 	.init           =       tcf_defact_init, \ | 
 | 136 | 	.walk           =       tcf_generic_walker, \ | 
 | 137 |  | 
 | 138 | #define tca_use_default_defines(name) \ | 
 | 139 | 	static u32 idx_gen; \ | 
 | 140 | 	static struct tcf_defact *tcf_##name_ht[MY_TAB_SIZE]; \ | 
 | 141 | 	static DEFINE_RWLOCK(##name_lock); | 
 | 142 | #endif /* _NET_ACT_GENERIC_H */ |