2 * net/sched/gact.c Generic actions
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
9 * copyright Jamal Hadi Salim (2002-4)
13 #include <linux/types.h>
14 #include <linux/kernel.h>
15 #include <linux/string.h>
16 #include <linux/errno.h>
17 #include <linux/skbuff.h>
18 #include <linux/rtnetlink.h>
19 #include <linux/module.h>
20 #include <linux/init.h>
21 #include <net/netlink.h>
22 #include <net/pkt_sched.h>
23 #include <linux/tc_act/tc_gact.h>
24 #include <net/tc_act/tc_gact.h>
26 #define GACT_TAB_MASK 15
27 static struct tcf_common
*tcf_gact_ht
[GACT_TAB_MASK
+ 1];
28 static u32 gact_idx_gen
;
29 static DEFINE_RWLOCK(gact_lock
);
31 static struct tcf_hashinfo gact_hash_info
= {
33 .hmask
= GACT_TAB_MASK
,
37 #ifdef CONFIG_GACT_PROB
38 static int gact_net_rand(struct tcf_gact
*gact
)
40 if (!gact
->tcfg_pval
|| net_random() % gact
->tcfg_pval
)
41 return gact
->tcf_action
;
42 return gact
->tcfg_paction
;
45 static int gact_determ(struct tcf_gact
*gact
)
47 if (!gact
->tcfg_pval
|| gact
->tcf_bstats
.packets
% gact
->tcfg_pval
)
48 return gact
->tcf_action
;
49 return gact
->tcfg_paction
;
52 typedef int (*g_rand
)(struct tcf_gact
*gact
);
53 static g_rand gact_rand
[MAX_RAND
]= { NULL
, gact_net_rand
, gact_determ
};
54 #endif /* CONFIG_GACT_PROB */
56 static int tcf_gact_init(struct nlattr
*nla
, struct nlattr
*est
,
57 struct tc_action
*a
, int ovr
, int bind
)
59 struct nlattr
*tb
[TCA_GACT_MAX
+ 1];
61 struct tcf_gact
*gact
;
62 struct tcf_common
*pc
;
69 err
= nla_parse_nested(tb
, TCA_GACT_MAX
, nla
, NULL
);
73 if (tb
[TCA_GACT_PARMS
] == NULL
||
74 nla_len(tb
[TCA_GACT_PARMS
]) < sizeof(*parm
))
76 parm
= nla_data(tb
[TCA_GACT_PARMS
]);
78 if (tb
[TCA_GACT_PROB
] != NULL
)
79 #ifdef CONFIG_GACT_PROB
80 if (nla_len(tb
[TCA_GACT_PROB
]) < sizeof(struct tc_gact_p
))
86 pc
= tcf_hash_check(parm
->index
, a
, bind
, &gact_hash_info
);
88 pc
= tcf_hash_create(parm
->index
, est
, a
, sizeof(*gact
),
89 bind
, &gact_idx_gen
, &gact_hash_info
);
95 tcf_hash_release(pc
, bind
, &gact_hash_info
);
102 spin_lock_bh(&gact
->tcf_lock
);
103 gact
->tcf_action
= parm
->action
;
104 #ifdef CONFIG_GACT_PROB
105 if (tb
[TCA_GACT_PROB
] != NULL
) {
106 struct tc_gact_p
*p_parm
= nla_data(tb
[TCA_GACT_PROB
]);
107 gact
->tcfg_paction
= p_parm
->paction
;
108 gact
->tcfg_pval
= p_parm
->pval
;
109 gact
->tcfg_ptype
= p_parm
->ptype
;
112 spin_unlock_bh(&gact
->tcf_lock
);
113 if (ret
== ACT_P_CREATED
)
114 tcf_hash_insert(pc
, &gact_hash_info
);
118 static int tcf_gact_cleanup(struct tc_action
*a
, int bind
)
120 struct tcf_gact
*gact
= a
->priv
;
123 return tcf_hash_release(&gact
->common
, bind
, &gact_hash_info
);
127 static int tcf_gact(struct sk_buff
*skb
, struct tc_action
*a
, struct tcf_result
*res
)
129 struct tcf_gact
*gact
= a
->priv
;
130 int action
= TC_ACT_SHOT
;
132 spin_lock(&gact
->tcf_lock
);
133 #ifdef CONFIG_GACT_PROB
134 if (gact
->tcfg_ptype
&& gact_rand
[gact
->tcfg_ptype
] != NULL
)
135 action
= gact_rand
[gact
->tcfg_ptype
](gact
);
137 action
= gact
->tcf_action
;
139 action
= gact
->tcf_action
;
141 gact
->tcf_bstats
.bytes
+= skb
->len
;
142 gact
->tcf_bstats
.packets
++;
143 if (action
== TC_ACT_SHOT
)
144 gact
->tcf_qstats
.drops
++;
145 gact
->tcf_tm
.lastuse
= jiffies
;
146 spin_unlock(&gact
->tcf_lock
);
151 static int tcf_gact_dump(struct sk_buff
*skb
, struct tc_action
*a
, int bind
, int ref
)
153 unsigned char *b
= skb_tail_pointer(skb
);
155 struct tcf_gact
*gact
= a
->priv
;
158 opt
.index
= gact
->tcf_index
;
159 opt
.refcnt
= gact
->tcf_refcnt
- ref
;
160 opt
.bindcnt
= gact
->tcf_bindcnt
- bind
;
161 opt
.action
= gact
->tcf_action
;
162 NLA_PUT(skb
, TCA_GACT_PARMS
, sizeof(opt
), &opt
);
163 #ifdef CONFIG_GACT_PROB
164 if (gact
->tcfg_ptype
) {
165 struct tc_gact_p p_opt
;
166 p_opt
.paction
= gact
->tcfg_paction
;
167 p_opt
.pval
= gact
->tcfg_pval
;
168 p_opt
.ptype
= gact
->tcfg_ptype
;
169 NLA_PUT(skb
, TCA_GACT_PROB
, sizeof(p_opt
), &p_opt
);
172 t
.install
= jiffies_to_clock_t(jiffies
- gact
->tcf_tm
.install
);
173 t
.lastuse
= jiffies_to_clock_t(jiffies
- gact
->tcf_tm
.lastuse
);
174 t
.expires
= jiffies_to_clock_t(gact
->tcf_tm
.expires
);
175 NLA_PUT(skb
, TCA_GACT_TM
, sizeof(t
), &t
);
183 static struct tc_action_ops act_gact_ops
= {
185 .hinfo
= &gact_hash_info
,
186 .type
= TCA_ACT_GACT
,
187 .capab
= TCA_CAP_NONE
,
188 .owner
= THIS_MODULE
,
190 .dump
= tcf_gact_dump
,
191 .cleanup
= tcf_gact_cleanup
,
192 .lookup
= tcf_hash_search
,
193 .init
= tcf_gact_init
,
194 .walk
= tcf_generic_walker
197 MODULE_AUTHOR("Jamal Hadi Salim(2002-4)");
198 MODULE_DESCRIPTION("Generic Classifier actions");
199 MODULE_LICENSE("GPL");
201 static int __init
gact_init_module(void)
203 #ifdef CONFIG_GACT_PROB
204 printk("GACT probability on\n");
206 printk("GACT probability NOT on\n");
208 return tcf_register_action(&act_gact_ops
);
211 static void __exit
gact_cleanup_module(void)
213 tcf_unregister_action(&act_gact_ops
);
216 module_init(gact_init_module
);
217 module_exit(gact_cleanup_module
);