[net-next-2.6.git] / net / bridge / netfilter / ebt_ulog.c

/*
 * netfilter module for userspace bridged Ethernet frames logging daemons
 *
 *	Authors:
 *	Bart De Schuymer <bdschuym@pandora.be>
 *	Harald Welte <laforge@netfilter.org>
 *
 *  November, 2004
 *
 * Based on ipt_ULOG.c, which is
 * (C) 2000-2002 by Harald Welte <laforge@netfilter.org>
 *
 * This module accepts two parameters:
 *
 * nlbufsiz:
 *   The parameter specifies how big the buffer for each netlink multicast
 * group is. e.g. If you say nlbufsiz=8192, up to eight kb of packets will
 * get accumulated in the kernel until they are sent to userspace. It is
 * NOT possible to allocate more than 128kB, and it is strongly discouraged,
 * because atomically allocating 128kB inside the network rx softirq is not
 * reliable. Please also keep in mind that this buffer size is allocated for
 * each nlgroup you are using, so the total kernel memory usage increases
 * by that factor.
 *
 * flushtimeout:
 *   Specify, after how many hundredths of a second the queue should be
 *   flushed even if it is not full yet.
 *
 */

#include <linux/module.h>
#include <linux/spinlock.h>
#include <linux/socket.h>
#include <linux/skbuff.h>
#include <linux/kernel.h>
#include <linux/timer.h>
#include <linux/netlink.h>
#include <linux/netdevice.h>
#include <linux/netfilter_bridge/ebtables.h>
#include <linux/netfilter_bridge/ebt_ulog.h>
#include <net/netfilter/nf_log.h>
#include <net/sock.h>
#include "../br_private.h"

#define PRINTR(format, args...) do { if (net_ratelimit()) \
				printk(format , ## args); } while (0)

static unsigned int nlbufsiz = NLMSG_GOODSIZE;
module_param(nlbufsiz, uint, 0600);
MODULE_PARM_DESC(nlbufsiz, "netlink buffer size (number of bytes) "
			   "(defaults to 4096)");

static unsigned int flushtimeout = 10;
module_param(flushtimeout, uint, 0600);
MODULE_PARM_DESC(flushtimeout, "buffer flush timeout (hundredths ofa second) "
			       "(defaults to 10)");

typedef struct {
	unsigned int qlen;		/* number of nlmsgs' in the skb */
	struct nlmsghdr *lastnlh;	/* netlink header of last msg in skb */
	struct sk_buff *skb;		/* the pre-allocated skb */
	struct timer_list timer;	/* the timer function */
	spinlock_t lock;		/* the per-queue lock */
} ebt_ulog_buff_t;

static ebt_ulog_buff_t ulog_buffers[EBT_ULOG_MAXNLGROUPS];
static struct sock *ebtulognl;

/* send one ulog_buff_t to userspace */
static void ulog_send(unsigned int nlgroup)
{
	ebt_ulog_buff_t *ub = &ulog_buffers[nlgroup];

	if (timer_pending(&ub->timer))
		del_timer(&ub->timer);

	if (!ub->skb)
		return;

	/* last nlmsg needs NLMSG_DONE */
	if (ub->qlen > 1)
		ub->lastnlh->nlmsg_type = NLMSG_DONE;

	NETLINK_CB(ub->skb).dst_group = nlgroup + 1;
	netlink_broadcast(ebtulognl, ub->skb, 0, nlgroup + 1, GFP_ATOMIC);

	ub->qlen = 0;
	ub->skb = NULL;
}

/* timer function to flush queue in flushtimeout time */
static void ulog_timer(unsigned long data)
{
	spin_lock_bh(&ulog_buffers[data].lock);
	if (ulog_buffers[data].skb)
		ulog_send(data);
	spin_unlock_bh(&ulog_buffers[data].lock);
}

static struct sk_buff *ulog_alloc_skb(unsigned int size)
{
	struct sk_buff *skb;
	unsigned int n;

	n = max(size, nlbufsiz);
	skb = alloc_skb(n, GFP_ATOMIC);
	if (!skb) {
		PRINTR(KERN_ERR "ebt_ulog: can't alloc whole buffer "
		       "of size %ub!\n", n);
		if (n > size) {
			/* try to allocate only as much as we need for
			 * current packet */
			skb = alloc_skb(size, GFP_ATOMIC);
			if (!skb)
				PRINTR(KERN_ERR "ebt_ulog: can't even allocate "
				       "buffer of size %ub\n", size);
		}
	}

	return skb;
}

static void ebt_ulog_packet(unsigned int hooknr, const struct sk_buff *skb,
   const struct net_device *in, const struct net_device *out,
   const struct ebt_ulog_info *uloginfo, const char *prefix)
{
	ebt_ulog_packet_msg_t *pm;
	size_t size, copy_len;
	struct nlmsghdr *nlh;
	unsigned int group = uloginfo->nlgroup;
	ebt_ulog_buff_t *ub = &ulog_buffers[group];
	spinlock_t *lock = &ub->lock;
	ktime_t kt;

	if ((uloginfo->cprange == 0) ||
	    (uloginfo->cprange > skb->len + ETH_HLEN))
		copy_len = skb->len + ETH_HLEN;
	else
		copy_len = uloginfo->cprange;

	size = NLMSG_SPACE(sizeof(*pm) + copy_len);
	if (size > nlbufsiz) {
		PRINTR("ebt_ulog: Size %Zd needed, but nlbufsiz=%d\n",
		       size, nlbufsiz);
		return;
	}

	spin_lock_bh(lock);

	if (!ub->skb) {
		if (!(ub->skb = ulog_alloc_skb(size)))
			goto alloc_failure;
	} else if (size > skb_tailroom(ub->skb)) {
		ulog_send(group);

		if (!(ub->skb = ulog_alloc_skb(size)))
			goto alloc_failure;
	}

	nlh = NLMSG_PUT(ub->skb, 0, ub->qlen, 0,
			size - NLMSG_ALIGN(sizeof(*nlh)));
	ub->qlen++;

	pm = NLMSG_DATA(nlh);

	/* Fill in the ulog data */
	pm->version = EBT_ULOG_VERSION;
	kt = ktime_get_real();
	pm->stamp = ktime_to_timeval(kt);
	if (ub->qlen == 1)
		ub->skb->tstamp = kt;
	pm->data_len = copy_len;
	pm->mark = skb->mark;
	pm->hook = hooknr;
	if (uloginfo->prefix != NULL)
		strcpy(pm->prefix, uloginfo->prefix);
	else
		*(pm->prefix) = '\0';

	if (in) {
		strcpy(pm->physindev, in->name);
		/* If in isn't a bridge, then physindev==indev */
		if (in->br_port)
			strcpy(pm->indev, in->br_port->br->dev->name);
		else
			strcpy(pm->indev, in->name);
	} else
		pm->indev[0] = pm->physindev[0] = '\0';

	if (out) {
		/* If out exists, then out is a bridge port */
		strcpy(pm->physoutdev, out->name);
		strcpy(pm->outdev, out->br_port->br->dev->name);
	} else
		pm->outdev[0] = pm->physoutdev[0] = '\0';

	if (skb_copy_bits(skb, -ETH_HLEN, pm->data, copy_len) < 0)
		BUG();

	if (ub->qlen > 1)
		ub->lastnlh->nlmsg_flags |= NLM_F_MULTI;

	ub->lastnlh = nlh;

	if (ub->qlen >= uloginfo->qthreshold)
		ulog_send(group);
	else if (!timer_pending(&ub->timer)) {
		ub->timer.expires = jiffies + flushtimeout * HZ / 100;
		add_timer(&ub->timer);
	}

unlock:
	spin_unlock_bh(lock);

	return;

nlmsg_failure:
	printk(KERN_CRIT "ebt_ulog: error during NLMSG_PUT. This should "
	       "not happen, please report to author.\n");
	goto unlock;
alloc_failure:
	goto unlock;
}

/* this function is registered with the netfilter core */
static void ebt_log_packet(u_int8_t pf, unsigned int hooknum,
   const struct sk_buff *skb, const struct net_device *in,
   const struct net_device *out, const struct nf_loginfo *li,
   const char *prefix)
{
	struct ebt_ulog_info loginfo;

	if (!li || li->type != NF_LOG_TYPE_ULOG) {
		loginfo.nlgroup = EBT_ULOG_DEFAULT_NLGROUP;
		loginfo.cprange = 0;
		loginfo.qthreshold = EBT_ULOG_DEFAULT_QTHRESHOLD;
		loginfo.prefix[0] = '\0';
	} else {
		loginfo.nlgroup = li->u.ulog.group;
		loginfo.cprange = li->u.ulog.copy_len;
		loginfo.qthreshold = li->u.ulog.qthreshold;
		strlcpy(loginfo.prefix, prefix, sizeof(loginfo.prefix));
	}

	ebt_ulog_packet(hooknum, skb, in, out, &loginfo, prefix);
}

static void ebt_ulog(const struct sk_buff *skb, unsigned int hooknr,
   const struct net_device *in, const struct net_device *out,
   const void *data, unsigned int datalen)
{
	const struct ebt_ulog_info *uloginfo = data;

	ebt_ulog_packet(hooknr, skb, in, out, uloginfo, NULL);
}


static int ebt_ulog_check(const char *tablename, unsigned int hookmask,
   const struct ebt_entry *e, void *data, unsigned int datalen)
{
	struct ebt_ulog_info *uloginfo = data;

	if (datalen != EBT_ALIGN(sizeof(struct ebt_ulog_info)) ||
	    uloginfo->nlgroup > 31)
		return -EINVAL;

	uloginfo->prefix[EBT_ULOG_PREFIX_LEN - 1] = '\0';

	if (uloginfo->qthreshold > EBT_ULOG_MAX_QLEN)
		uloginfo->qthreshold = EBT_ULOG_MAX_QLEN;

	return 0;
}

static struct ebt_watcher ulog __read_mostly = {
	.name		= EBT_ULOG_WATCHER,
	.watcher	= ebt_ulog,
	.check		= ebt_ulog_check,
	.me		= THIS_MODULE,
};

static const struct nf_logger ebt_ulog_logger = {
	.name		= EBT_ULOG_WATCHER,
	.logfn		= &ebt_log_packet,
	.me		= THIS_MODULE,
};

static int __init ebt_ulog_init(void)
{
	int i, ret = 0;

	if (nlbufsiz >= 128*1024) {
		printk(KERN_NOTICE "ebt_ulog: Netlink buffer has to be <= 128kB,"
		       " please try a smaller nlbufsiz parameter.\n");
		return -EINVAL;
	}

	/* initialize ulog_buffers */
	for (i = 0; i < EBT_ULOG_MAXNLGROUPS; i++) {
		setup_timer(&ulog_buffers[i].timer, ulog_timer, i);
		spin_lock_init(&ulog_buffers[i].lock);
	}

	ebtulognl = netlink_kernel_create(&init_net, NETLINK_NFLOG,
					  EBT_ULOG_MAXNLGROUPS, NULL, NULL,
					  THIS_MODULE);
	if (!ebtulognl)
		ret = -ENOMEM;
	else if ((ret = ebt_register_watcher(&ulog)))
		netlink_kernel_release(ebtulognl);

	if (ret == 0)
		nf_log_register(PF_BRIDGE, &ebt_ulog_logger);

	return ret;
}

static void __exit ebt_ulog_fini(void)
{
	ebt_ulog_buff_t *ub;
	int i;

	nf_log_unregister(&ebt_ulog_logger);
	ebt_unregister_watcher(&ulog);
	for (i = 0; i < EBT_ULOG_MAXNLGROUPS; i++) {
		ub = &ulog_buffers[i];
		if (timer_pending(&ub->timer))
			del_timer(&ub->timer);
		spin_lock_bh(&ub->lock);
		if (ub->skb) {
			kfree_skb(ub->skb);
			ub->skb = NULL;
		}
		spin_unlock_bh(&ub->lock);
	}
	netlink_kernel_release(ebtulognl);
}

module_init(ebt_ulog_init);
module_exit(ebt_ulog_fini);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Bart De Schuymer <bdschuym@pandora.be>");
MODULE_DESCRIPTION("Ebtables: Packet logging to netlink using ULOG");
Commit	Line	Data
1da177e4 LT	1	/*
	2	* netfilter module for userspace bridged Ethernet frames logging daemons
	3	*
	4	* Authors:
	5	* Bart De Schuymer <bdschuym@pandora.be>
d5228a4f	6	* Harald Welte <laforge@netfilter.org>
1da177e4 LT	7	*
	8	* November, 2004
	9	*
	10	* Based on ipt_ULOG.c, which is
	11	* (C) 2000-2002 by Harald Welte <laforge@netfilter.org>
	12	*
9d6f229f YH	13	* This module accepts two parameters:
9d6f229f YH	14	*
1da177e4 LT	15	* nlbufsiz:
	16	* The parameter specifies how big the buffer for each netlink multicast
	17	* group is. e.g. If you say nlbufsiz=8192, up to eight kb of packets will
	18	* get accumulated in the kernel until they are sent to userspace. It is
	19	* NOT possible to allocate more than 128kB, and it is strongly discouraged,
	20	* because atomically allocating 128kB inside the network rx softirq is not
	21	* reliable. Please also keep in mind that this buffer size is allocated for
	22	* each nlgroup you are using, so the total kernel memory usage increases
	23	* by that factor.
	24	*
	25	* flushtimeout:
	26	* Specify, after how many hundredths of a second the queue should be
	27	* flushed even if it is not full yet.
	28	*
	29	*/
	30
	31	#include <linux/module.h>
1da177e4 LT	32	#include <linux/spinlock.h>
	33	#include <linux/socket.h>
	34	#include <linux/skbuff.h>
	35	#include <linux/kernel.h>
	36	#include <linux/timer.h>
	37	#include <linux/netlink.h>
	38	#include <linux/netdevice.h>
1da177e4 LT	39	#include <linux/netfilter_bridge/ebtables.h>
1da177e4 LT	40	#include <linux/netfilter_bridge/ebt_ulog.h>
f01ffbd6	41	#include <net/netfilter/nf_log.h>
1da177e4 LT	42	#include <net/sock.h>
	43	#include "../br_private.h"
	44
	45	#define PRINTR(format, args...) do { if (net_ratelimit()) \
9d6f229f	46	printk(format , ## args); } while (0)
1da177e4	47
c2db2924	48	static unsigned int nlbufsiz = NLMSG_GOODSIZE;
1da177e4 LT	49	module_param(nlbufsiz, uint, 0600);
1da177e4 LT	50	MODULE_PARM_DESC(nlbufsiz, "netlink buffer size (number of bytes) "
9d6f229f	51	"(defaults to 4096)");
1da177e4 LT	52
	53	static unsigned int flushtimeout = 10;
	54	module_param(flushtimeout, uint, 0600);
	55	MODULE_PARM_DESC(flushtimeout, "buffer flush timeout (hundredths ofa second) "
9d6f229f	56	"(defaults to 10)");
1da177e4 LT	57
	58	typedef struct {
	59	unsigned int qlen; /* number of nlmsgs' in the skb */
	60	struct nlmsghdr lastnlh; / netlink header of last msg in skb */
	61	struct sk_buff skb; / the pre-allocated skb */
	62	struct timer_list timer; /* the timer function */
	63	spinlock_t lock; /* the per-queue lock */
	64	} ebt_ulog_buff_t;
	65
	66	static ebt_ulog_buff_t ulog_buffers[EBT_ULOG_MAXNLGROUPS];
	67	static struct sock *ebtulognl;
	68
	69	/* send one ulog_buff_t to userspace */
	70	static void ulog_send(unsigned int nlgroup)
	71	{
	72	ebt_ulog_buff_t *ub = &ulog_buffers[nlgroup];
	73
	74	if (timer_pending(&ub->timer))
	75	del_timer(&ub->timer);
	76
dcb7cd97 MH	77	if (!ub->skb)
	78	return;
	79
1da177e4 LT	80	/* last nlmsg needs NLMSG_DONE */
	81	if (ub->qlen > 1)
	82	ub->lastnlh->nlmsg_type = NLMSG_DONE;
	83
ac6d439d PM	84	NETLINK_CB(ub->skb).dst_group = nlgroup + 1;
ac6d439d PM	85	netlink_broadcast(ebtulognl, ub->skb, 0, nlgroup + 1, GFP_ATOMIC);
1da177e4 LT	86
	87	ub->qlen = 0;
	88	ub->skb = NULL;
	89	}
	90
	91	/* timer function to flush queue in flushtimeout time */
	92	static void ulog_timer(unsigned long data)
	93	{
	94	spin_lock_bh(&ulog_buffers[data].lock);
	95	if (ulog_buffers[data].skb)
	96	ulog_send(data);
	97	spin_unlock_bh(&ulog_buffers[data].lock);
	98	}
	99
	100	static struct sk_buff *ulog_alloc_skb(unsigned int size)
	101	{
	102	struct sk_buff *skb;
ad2ad0f9	103	unsigned int n;
1da177e4	104
ad2ad0f9 PM	105	n = max(size, nlbufsiz);
ad2ad0f9 PM	106	skb = alloc_skb(n, GFP_ATOMIC);
1da177e4 LT	107	if (!skb) {
1da177e4 LT	108	PRINTR(KERN_ERR "ebt_ulog: can't alloc whole buffer "
ad2ad0f9 PM	109	"of size %ub!\n", n);
ad2ad0f9 PM	110	if (n > size) {
1da177e4 LT	111	/* try to allocate only as much as we need for
	112	* current packet */
	113	skb = alloc_skb(size, GFP_ATOMIC);
	114	if (!skb)
	115	PRINTR(KERN_ERR "ebt_ulog: can't even allocate "
	116	"buffer of size %ub\n", size);
	117	}
	118	}
	119
	120	return skb;
	121	}
	122
d5228a4f	123	static void ebt_ulog_packet(unsigned int hooknr, const struct sk_buff *skb,
1da177e4	124	const struct net_device in, const struct net_device out,
d5228a4f	125	const struct ebt_ulog_info uloginfo, const char prefix)
1da177e4 LT	126	{
	127	ebt_ulog_packet_msg_t *pm;
	128	size_t size, copy_len;
	129	struct nlmsghdr *nlh;
1da177e4 LT	130	unsigned int group = uloginfo->nlgroup;
	131	ebt_ulog_buff_t *ub = &ulog_buffers[group];
	132	spinlock_t *lock = &ub->lock;
b7aa0bf7	133	ktime_t kt;
1da177e4 LT	134
	135	if ((uloginfo->cprange == 0) \|\|
	136	(uloginfo->cprange > skb->len + ETH_HLEN))
	137	copy_len = skb->len + ETH_HLEN;
	138	else
	139	copy_len = uloginfo->cprange;
	140
	141	size = NLMSG_SPACE(sizeof(*pm) + copy_len);
	142	if (size > nlbufsiz) {
	143	PRINTR("ebt_ulog: Size %Zd needed, but nlbufsiz=%d\n",
	144	size, nlbufsiz);
	145	return;
	146	}
	147
	148	spin_lock_bh(lock);
	149
	150	if (!ub->skb) {
	151	if (!(ub->skb = ulog_alloc_skb(size)))
	152	goto alloc_failure;
	153	} else if (size > skb_tailroom(ub->skb)) {
	154	ulog_send(group);
	155
	156	if (!(ub->skb = ulog_alloc_skb(size)))
	157	goto alloc_failure;
	158	}
	159
	160	nlh = NLMSG_PUT(ub->skb, 0, ub->qlen, 0,
9d6f229f	161	size - NLMSG_ALIGN(sizeof(*nlh)));
1da177e4 LT	162	ub->qlen++;
	163
	164	pm = NLMSG_DATA(nlh);
	165
	166	/* Fill in the ulog data */
	167	pm->version = EBT_ULOG_VERSION;
b7aa0bf7 ED	168	kt = ktime_get_real();
b7aa0bf7 ED	169	pm->stamp = ktime_to_timeval(kt);
1da177e4	170	if (ub->qlen == 1)
b7aa0bf7	171	ub->skb->tstamp = kt;
1da177e4	172	pm->data_len = copy_len;
82e91ffe	173	pm->mark = skb->mark;
1da177e4 LT	174	pm->hook = hooknr;
	175	if (uloginfo->prefix != NULL)
	176	strcpy(pm->prefix, uloginfo->prefix);
	177	else
	178	*(pm->prefix) = '\0';
	179
	180	if (in) {
	181	strcpy(pm->physindev, in->name);
	182	/* If in isn't a bridge, then physindev==indev */
	183	if (in->br_port)
	184	strcpy(pm->indev, in->br_port->br->dev->name);
	185	else
	186	strcpy(pm->indev, in->name);
	187	} else
	188	pm->indev[0] = pm->physindev[0] = '\0';
	189
	190	if (out) {
	191	/* If out exists, then out is a bridge port */
	192	strcpy(pm->physoutdev, out->name);
	193	strcpy(pm->outdev, out->br_port->br->dev->name);
	194	} else
	195	pm->outdev[0] = pm->physoutdev[0] = '\0';
	196
	197	if (skb_copy_bits(skb, -ETH_HLEN, pm->data, copy_len) < 0)
	198	BUG();
	199
	200	if (ub->qlen > 1)
	201	ub->lastnlh->nlmsg_flags \|= NLM_F_MULTI;
	202
	203	ub->lastnlh = nlh;
	204
	205	if (ub->qlen >= uloginfo->qthreshold)
	206	ulog_send(group);
	207	else if (!timer_pending(&ub->timer)) {
	208	ub->timer.expires = jiffies + flushtimeout * HZ / 100;
	209	add_timer(&ub->timer);
	210	}
	211
	212	unlock:
	213	spin_unlock_bh(lock);
	214
	215	return;
	216
	217	nlmsg_failure:
	218	printk(KERN_CRIT "ebt_ulog: error during NLMSG_PUT. This should "
	219	"not happen, please report to author.\n");
	220	goto unlock;
	221	alloc_failure:
	222	goto unlock;
	223	}
	224
d5228a4f	225	/* this function is registered with the netfilter core */
76108cea	226	static void ebt_log_packet(u_int8_t pf, unsigned int hooknum,
d5228a4f BDS	227	const struct sk_buff skb, const struct net_device in,
	228	const struct net_device out, const struct nf_loginfo li,
	229	const char *prefix)
	230	{
	231	struct ebt_ulog_info loginfo;
	232
	233	if (!li \|\| li->type != NF_LOG_TYPE_ULOG) {
	234	loginfo.nlgroup = EBT_ULOG_DEFAULT_NLGROUP;
	235	loginfo.cprange = 0;
	236	loginfo.qthreshold = EBT_ULOG_DEFAULT_QTHRESHOLD;
	237	loginfo.prefix[0] = '\0';
	238	} else {
	239	loginfo.nlgroup = li->u.ulog.group;
	240	loginfo.cprange = li->u.ulog.copy_len;
	241	loginfo.qthreshold = li->u.ulog.qthreshold;
	242	strlcpy(loginfo.prefix, prefix, sizeof(loginfo.prefix));
	243	}
	244
	245	ebt_ulog_packet(hooknum, skb, in, out, &loginfo, prefix);
	246	}
	247
	248	static void ebt_ulog(const struct sk_buff *skb, unsigned int hooknr,
	249	const struct net_device in, const struct net_device out,
	250	const void *data, unsigned int datalen)
	251	{
abfdf1c4	252	const struct ebt_ulog_info *uloginfo = data;
d5228a4f BDS	253
	254	ebt_ulog_packet(hooknr, skb, in, out, uloginfo, NULL);
	255	}
	256
	257
1da177e4 LT	258	static int ebt_ulog_check(const char *tablename, unsigned int hookmask,
	259	const struct ebt_entry e, void data, unsigned int datalen)
	260	{
abfdf1c4	261	struct ebt_ulog_info *uloginfo = data;
1da177e4 LT	262
	263	if (datalen != EBT_ALIGN(sizeof(struct ebt_ulog_info)) \|\|
	264	uloginfo->nlgroup > 31)
	265	return -EINVAL;
	266
	267	uloginfo->prefix[EBT_ULOG_PREFIX_LEN - 1] = '\0';
	268
	269	if (uloginfo->qthreshold > EBT_ULOG_MAX_QLEN)
	270	uloginfo->qthreshold = EBT_ULOG_MAX_QLEN;
	271
	272	return 0;
	273	}
	274
30083c95	275	static struct ebt_watcher ulog __read_mostly = {
1da177e4 LT	276	.name = EBT_ULOG_WATCHER,
	277	.watcher = ebt_ulog,
	278	.check = ebt_ulog_check,
	279	.me = THIS_MODULE,
	280	};
	281
7b2f9631	282	static const struct nf_logger ebt_ulog_logger = {
d5228a4f BDS	283	.name = EBT_ULOG_WATCHER,
	284	.logfn = &ebt_log_packet,
	285	.me = THIS_MODULE,
	286	};
	287
65b4b4e8	288	static int __init ebt_ulog_init(void)
1da177e4 LT	289	{
	290	int i, ret = 0;
	291
	292	if (nlbufsiz >= 128*1024) {
	293	printk(KERN_NOTICE "ebt_ulog: Netlink buffer has to be <= 128kB,"
	294	" please try a smaller nlbufsiz parameter.\n");
	295	return -EINVAL;
	296	}
	297
	298	/* initialize ulog_buffers */
	299	for (i = 0; i < EBT_ULOG_MAXNLGROUPS; i++) {
e6f689db	300	setup_timer(&ulog_buffers[i].timer, ulog_timer, i);
1da177e4 LT	301	spin_lock_init(&ulog_buffers[i].lock);
	302	}
	303
b4b51029 EB	304	ebtulognl = netlink_kernel_create(&init_net, NETLINK_NFLOG,
	305	EBT_ULOG_MAXNLGROUPS, NULL, NULL,
	306	THIS_MODULE);
1da177e4 LT	307	if (!ebtulognl)
	308	ret = -ENOMEM;
	309	else if ((ret = ebt_register_watcher(&ulog)))
b7c6ba6e	310	netlink_kernel_release(ebtulognl);
1da177e4	311
7e2acc7e PM	312	if (ret == 0)
7e2acc7e PM	313	nf_log_register(PF_BRIDGE, &ebt_ulog_logger);
d5228a4f	314
1da177e4 LT	315	return ret;
	316	}
	317
65b4b4e8	318	static void __exit ebt_ulog_fini(void)
1da177e4 LT	319	{
	320	ebt_ulog_buff_t *ub;
	321	int i;
	322
e92ad99c	323	nf_log_unregister(&ebt_ulog_logger);
1da177e4 LT	324	ebt_unregister_watcher(&ulog);
	325	for (i = 0; i < EBT_ULOG_MAXNLGROUPS; i++) {
	326	ub = &ulog_buffers[i];
	327	if (timer_pending(&ub->timer))
	328	del_timer(&ub->timer);
	329	spin_lock_bh(&ub->lock);
	330	if (ub->skb) {
	331	kfree_skb(ub->skb);
	332	ub->skb = NULL;
	333	}
	334	spin_unlock_bh(&ub->lock);
	335	}
b7c6ba6e	336	netlink_kernel_release(ebtulognl);
1da177e4 LT	337	}
1da177e4 LT	338
65b4b4e8 AM	339	module_init(ebt_ulog_init);
65b4b4e8 AM	340	module_exit(ebt_ulog_fini);
1da177e4 LT	341	MODULE_LICENSE("GPL");
1da177e4 LT	342	MODULE_AUTHOR("Bart De Schuymer <bdschuym@pandora.be>");
f776c4cd	343	MODULE_DESCRIPTION("Ebtables: Packet logging to netlink using ULOG");