[deliverable/linux.git] / net / sched / simple.c

/*
 * net/sched/simp.c	Simple example of an action
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 *
 * Authors:	Jamal Hadi Salim (2005)
 *
 */

#include <linux/config.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/netdevice.h>
#include <linux/skbuff.h>
#include <linux/rtnetlink.h>
#include <net/pkt_sched.h>

#define TCA_ACT_SIMP 22

/* XXX: Hide all these common elements under some macro 
 * probably
*/
#include <linux/tc_act/tc_defact.h>
#include <net/tc_act/tc_defact.h>

/* use generic hash table with 8 buckets */
#define MY_TAB_SIZE     8
#define MY_TAB_MASK     (MY_TAB_SIZE - 1)
static u32 idx_gen;
static struct tcf_defact *tcf_simp_ht[MY_TAB_SIZE];
static DEFINE_RWLOCK(simp_lock);

/* override the defaults */
#define tcf_st		tcf_defact
#define tc_st		tc_defact
#define tcf_t_lock	simp_lock
#define tcf_ht		tcf_simp_ht

#define CONFIG_NET_ACT_INIT 1
#include <net/pkt_act.h>
#include <net/act_generic.h>

static int tcf_simp(struct sk_buff **pskb, struct tc_action *a, struct tcf_result *res)
{
	struct sk_buff *skb = *pskb;
	struct tcf_defact *p = PRIV(a, defact);

	spin_lock(&p->lock);
	p->tm.lastuse = jiffies;
	p->bstats.bytes += skb->len;
	p->bstats.packets++;

	/* print policy string followed by _ then packet count 
	 * Example if this was the 3rd packet and the string was "hello" 
	 * then it would look like "hello_3" (without quotes) 
	 **/
	printk("simple: %s_%d\n", (char *)p->defdata, p->bstats.packets);
	spin_unlock(&p->lock);
	return p->action;
}

static struct tc_action_ops act_simp_ops = {
	.kind = "simple",
	.type = TCA_ACT_SIMP,
	.capab = TCA_CAP_NONE,
	.owner = THIS_MODULE,
	.act = tcf_simp,
	tca_use_default_ops
};

MODULE_AUTHOR("Jamal Hadi Salim(2005)");
MODULE_DESCRIPTION("Simple example action");
MODULE_LICENSE("GPL");

static int __init simp_init_module(void)
{
	int ret = tcf_register_action(&act_simp_ops);
	if (!ret)
		printk("Simple TC action Loaded\n");
	return ret;
}

static void __exit simp_cleanup_module(void)
{
	tcf_unregister_action(&act_simp_ops);
}

module_init(simp_init_module);
module_exit(simp_cleanup_module);
Commit	Line	Data
db753079 JHS	1	/*
	2	* net/sched/simp.c Simple example of an action
	3	*
	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.
	8	*
	9	* Authors: Jamal Hadi Salim (2005)
	10	*
	11	*/
	12
db753079	13	#include <linux/config.h>
cbdbf00a DM	14	#include <linux/module.h>
cbdbf00a DM	15	#include <linux/init.h>
db753079	16	#include <linux/kernel.h>
db753079 JHS	17	#include <linux/netdevice.h>
	18	#include <linux/skbuff.h>
	19	#include <linux/rtnetlink.h>
db753079 JHS	20	#include <net/pkt_sched.h>
	21
	22	#define TCA_ACT_SIMP 22
	23
	24	/* XXX: Hide all these common elements under some macro
	25	* probably
	26	*/
	27	#include <linux/tc_act/tc_defact.h>
	28	#include <net/tc_act/tc_defact.h>
	29
	30	/* use generic hash table with 8 buckets */
	31	#define MY_TAB_SIZE 8
	32	#define MY_TAB_MASK (MY_TAB_SIZE - 1)
	33	static u32 idx_gen;
	34	static struct tcf_defact *tcf_simp_ht[MY_TAB_SIZE];
	35	static DEFINE_RWLOCK(simp_lock);
	36
	37	/* override the defaults */
	38	#define tcf_st tcf_defact
	39	#define tc_st tc_defact
	40	#define tcf_t_lock simp_lock
	41	#define tcf_ht tcf_simp_ht
	42
	43	#define CONFIG_NET_ACT_INIT 1
	44	#include <net/pkt_act.h>
	45	#include <net/act_generic.h>
	46
abc3bc58	47	static int tcf_simp(struct sk_buff *pskb, struct tc_action a, struct tcf_result *res)
db753079 JHS	48	{
	49	struct sk_buff skb = pskb;
	50	struct tcf_defact *p = PRIV(a, defact);
	51
	52	spin_lock(&p->lock);
	53	p->tm.lastuse = jiffies;
	54	p->bstats.bytes += skb->len;
	55	p->bstats.packets++;
	56
	57	/* print policy string followed by _ then packet count
	58	* Example if this was the 3rd packet and the string was "hello"
	59	* then it would look like "hello_3" (without quotes)
	60	**/
	61	printk("simple: %s_%d\n", (char *)p->defdata, p->bstats.packets);
	62	spin_unlock(&p->lock);
	63	return p->action;
	64	}
	65
	66	static struct tc_action_ops act_simp_ops = {
	67	.kind = "simple",
	68	.type = TCA_ACT_SIMP,
	69	.capab = TCA_CAP_NONE,
	70	.owner = THIS_MODULE,
	71	.act = tcf_simp,
	72	tca_use_default_ops
	73	};
	74
	75	MODULE_AUTHOR("Jamal Hadi Salim(2005)");
	76	MODULE_DESCRIPTION("Simple example action");
	77	MODULE_LICENSE("GPL");
	78
	79	static int __init simp_init_module(void)
	80	{
	81	int ret = tcf_register_action(&act_simp_ops);
	82	if (!ret)
	83	printk("Simple TC action Loaded\n");
	84	return ret;
	85	}
	86
	87	static void __exit simp_cleanup_module(void)
	88	{
	89	tcf_unregister_action(&act_simp_ops);
	90	}
	91
	92	module_init(simp_init_module);
	93	module_exit(simp_cleanup_module);