[net-next-2.6.git] / net / ipv4 / netfilter / arptable_filter.c

/*
 * Filtering ARP tables module.
 *
 * Copyright (C) 2002 David S. Miller (davem@redhat.com)
 *
 */

#include <linux/module.h>
#include <linux/netfilter/x_tables.h>
#include <linux/netfilter_arp/arp_tables.h>
#include <linux/slab.h>

MODULE_LICENSE("GPL");
MODULE_AUTHOR("David S. Miller <davem@redhat.com>");
MODULE_DESCRIPTION("arptables filter table");

#define FILTER_VALID_HOOKS ((1 << NF_ARP_IN) | (1 << NF_ARP_OUT) | \
			   (1 << NF_ARP_FORWARD))

static const struct xt_table packet_filter = {
	.name		= "filter",
	.valid_hooks	= FILTER_VALID_HOOKS,
	.me		= THIS_MODULE,
	.af		= NFPROTO_ARP,
	.priority	= NF_IP_PRI_FILTER,
};

/* The work comes in here from netfilter.c */
static unsigned int
arptable_filter_hook(unsigned int hook, struct sk_buff *skb,
		     const struct net_device *in, const struct net_device *out,
		     int (*okfn)(struct sk_buff *))
{
	const struct net *net = dev_net((in != NULL) ? in : out);

	return arpt_do_table(skb, hook, in, out, net->ipv4.arptable_filter);
}

static struct nf_hook_ops *arpfilter_ops __read_mostly;

static int __net_init arptable_filter_net_init(struct net *net)
{
	struct arpt_replace *repl;
	
	repl = arpt_alloc_initial_table(&packet_filter);
	if (repl == NULL)
		return -ENOMEM;
	net->ipv4.arptable_filter =
		arpt_register_table(net, &packet_filter, repl);
	kfree(repl);
	if (IS_ERR(net->ipv4.arptable_filter))
		return PTR_ERR(net->ipv4.arptable_filter);
	return 0;
}

static void __net_exit arptable_filter_net_exit(struct net *net)
{
	arpt_unregister_table(net->ipv4.arptable_filter);
}

static struct pernet_operations arptable_filter_net_ops = {
	.init = arptable_filter_net_init,
	.exit = arptable_filter_net_exit,
};

static int __init arptable_filter_init(void)
{
	int ret;

	ret = register_pernet_subsys(&arptable_filter_net_ops);
	if (ret < 0)
		return ret;

	arpfilter_ops = xt_hook_link(&packet_filter, arptable_filter_hook);
	if (IS_ERR(arpfilter_ops)) {
		ret = PTR_ERR(arpfilter_ops);
		goto cleanup_table;
	}
	return ret;

cleanup_table:
	unregister_pernet_subsys(&arptable_filter_net_ops);
	return ret;
}

static void __exit arptable_filter_fini(void)
{
	xt_hook_unlink(&packet_filter, arpfilter_ops);
	unregister_pernet_subsys(&arptable_filter_net_ops);
}

module_init(arptable_filter_init);
module_exit(arptable_filter_fini);
Commit	Line	Data
1da177e4 LT	1	/*
	2	* Filtering ARP tables module.
	3	*
	4	* Copyright (C) 2002 David S. Miller (davem@redhat.com)
	5	*
	6	*/
	7
	8	#include <linux/module.h>
e3eaa991	9	#include <linux/netfilter/x_tables.h>
1da177e4	10	#include <linux/netfilter_arp/arp_tables.h>
5a0e3ad6	11	#include <linux/slab.h>
1da177e4 LT	12
	13	MODULE_LICENSE("GPL");
	14	MODULE_AUTHOR("David S. Miller <davem@redhat.com>");
	15	MODULE_DESCRIPTION("arptables filter table");
	16
	17	#define FILTER_VALID_HOOKS ((1 << NF_ARP_IN) \| (1 << NF_ARP_OUT) \| \
	18	(1 << NF_ARP_FORWARD))
	19
35aad0ff	20	static const struct xt_table packet_filter = {
1da177e4 LT	21	.name = "filter",
1da177e4 LT	22	.valid_hooks = FILTER_VALID_HOOKS,
1da177e4	23	.me = THIS_MODULE,
ee999d8b	24	.af = NFPROTO_ARP,
2b95efe7	25	.priority = NF_IP_PRI_FILTER,
1da177e4 LT	26	};
	27
	28	/* The work comes in here from netfilter.c */
737535c5 JE	29	static unsigned int
	30	arptable_filter_hook(unsigned int hook, struct sk_buff *skb,
	31	const struct net_device in, const struct net_device out,
	32	int (okfn)(struct sk_buff ))
1da177e4	33	{
2b21e051	34	const struct net *net = dev_net((in != NULL) ? in : out);
3918fed5	35
2b21e051	36	return arpt_do_table(skb, hook, in, out, net->ipv4.arptable_filter);
3918fed5 AD	37	}
3918fed5 AD	38
2b95efe7	39	static struct nf_hook_ops *arpfilter_ops __read_mostly;
1da177e4	40
9ea0cb26 AD	41	static int __net_init arptable_filter_net_init(struct net *net)
9ea0cb26 AD	42	{
e3eaa991 JE	43	struct arpt_replace *repl;
	44
	45	repl = arpt_alloc_initial_table(&packet_filter);
	46	if (repl == NULL)
	47	return -ENOMEM;
9ea0cb26	48	net->ipv4.arptable_filter =
e3eaa991 JE	49	arpt_register_table(net, &packet_filter, repl);
e3eaa991 JE	50	kfree(repl);
9ea0cb26 AD	51	if (IS_ERR(net->ipv4.arptable_filter))
	52	return PTR_ERR(net->ipv4.arptable_filter);
	53	return 0;
	54	}
	55
	56	static void __net_exit arptable_filter_net_exit(struct net *net)
	57	{
	58	arpt_unregister_table(net->ipv4.arptable_filter);
	59	}
	60
	61	static struct pernet_operations arptable_filter_net_ops = {
	62	.init = arptable_filter_net_init,
	63	.exit = arptable_filter_net_exit,
	64	};
	65
65b4b4e8	66	static int __init arptable_filter_init(void)
1da177e4	67	{
964ddaa1	68	int ret;
1da177e4	69
9ea0cb26 AD	70	ret = register_pernet_subsys(&arptable_filter_net_ops);
	71	if (ret < 0)
	72	return ret;
1da177e4	73
2b95efe7 JE	74	arpfilter_ops = xt_hook_link(&packet_filter, arptable_filter_hook);
	75	if (IS_ERR(arpfilter_ops)) {
	76	ret = PTR_ERR(arpfilter_ops);
964ddaa1	77	goto cleanup_table;
2b95efe7	78	}
1da177e4 LT	79	return ret;
1da177e4 LT	80
964ddaa1	81	cleanup_table:
9ea0cb26	82	unregister_pernet_subsys(&arptable_filter_net_ops);
1da177e4 LT	83	return ret;
	84	}
	85
65b4b4e8	86	static void __exit arptable_filter_fini(void)
1da177e4	87	{
2b95efe7	88	xt_hook_unlink(&packet_filter, arpfilter_ops);
9ea0cb26	89	unregister_pernet_subsys(&arptable_filter_net_ops);
1da177e4 LT	90	}
1da177e4 LT	91
65b4b4e8 AM	92	module_init(arptable_filter_init);
65b4b4e8 AM	93	module_exit(arptable_filter_fini);