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 */ |