[mirror_ubuntu-artful-kernel.git] / net / netfilter / nft_log.c

/*
 * Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net>
 * Copyright (c) 2012-2014 Pablo Neira Ayuso <pablo@netfilter.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * Development of this code funded by Astaro AG (http://www.astaro.com/)
 */

#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/netlink.h>
#include <linux/netfilter.h>
#include <linux/netfilter/nf_tables.h>
#include <net/netfilter/nf_tables.h>
#include <net/netfilter/nf_log.h>
#include <linux/netdevice.h>

static const char *nft_log_null_prefix = "";

struct nft_log {
	struct nf_loginfo	loginfo;
	char			*prefix;
};

static void nft_log_eval(const struct nft_expr *expr,
			 struct nft_regs *regs,
			 const struct nft_pktinfo *pkt)
{
	const struct nft_log *priv = nft_expr_priv(expr);

	nf_log_packet(nft_net(pkt), nft_pf(pkt), nft_hook(pkt), pkt->skb,
		      nft_in(pkt), nft_out(pkt), &priv->loginfo, "%s",
		      priv->prefix);
}

static const struct nla_policy nft_log_policy[NFTA_LOG_MAX + 1] = {
	[NFTA_LOG_GROUP]	= { .type = NLA_U16 },
	[NFTA_LOG_PREFIX]	= { .type = NLA_STRING,
				    .len = NF_LOG_PREFIXLEN - 1 },
	[NFTA_LOG_SNAPLEN]	= { .type = NLA_U32 },
	[NFTA_LOG_QTHRESHOLD]	= { .type = NLA_U16 },
	[NFTA_LOG_LEVEL]	= { .type = NLA_U32 },
	[NFTA_LOG_FLAGS]	= { .type = NLA_U32 },
};

static int nft_log_init(const struct nft_ctx *ctx,
			const struct nft_expr *expr,
			const struct nlattr * const tb[])
{
	struct nft_log *priv = nft_expr_priv(expr);
	struct nf_loginfo *li = &priv->loginfo;
	const struct nlattr *nla;
	int err;

	li->type = NF_LOG_TYPE_LOG;
	if (tb[NFTA_LOG_LEVEL] != NULL &&
	    tb[NFTA_LOG_GROUP] != NULL)
		return -EINVAL;
	if (tb[NFTA_LOG_GROUP] != NULL) {
		li->type = NF_LOG_TYPE_ULOG;
		if (tb[NFTA_LOG_FLAGS] != NULL)
			return -EINVAL;
	}

	nla = tb[NFTA_LOG_PREFIX];
	if (nla != NULL) {
		priv->prefix = kmalloc(nla_len(nla) + 1, GFP_KERNEL);
		if (priv->prefix == NULL)
			return -ENOMEM;
		nla_strlcpy(priv->prefix, nla, nla_len(nla) + 1);
	} else {
		priv->prefix = (char *)nft_log_null_prefix;
	}

	switch (li->type) {
	case NF_LOG_TYPE_LOG:
		if (tb[NFTA_LOG_LEVEL] != NULL) {
			li->u.log.level =
				ntohl(nla_get_be32(tb[NFTA_LOG_LEVEL]));
		} else {
			li->u.log.level = LOGLEVEL_WARNING;
		}
		if (li->u.log.level > LOGLEVEL_DEBUG) {
			err = -EINVAL;
			goto err1;
		}

		if (tb[NFTA_LOG_FLAGS] != NULL) {
			li->u.log.logflags =
				ntohl(nla_get_be32(tb[NFTA_LOG_FLAGS]));
			if (li->u.log.logflags & ~NF_LOG_MASK) {
				err = -EINVAL;
				goto err1;
			}
		}
		break;
	case NF_LOG_TYPE_ULOG:
		li->u.ulog.group = ntohs(nla_get_be16(tb[NFTA_LOG_GROUP]));
		if (tb[NFTA_LOG_SNAPLEN] != NULL) {
			li->u.ulog.flags |= NF_LOG_F_COPY_LEN;
			li->u.ulog.copy_len =
				ntohl(nla_get_be32(tb[NFTA_LOG_SNAPLEN]));
		}
		if (tb[NFTA_LOG_QTHRESHOLD] != NULL) {
			li->u.ulog.qthreshold =
				ntohs(nla_get_be16(tb[NFTA_LOG_QTHRESHOLD]));
		}
		break;
	}

	err = nf_logger_find_get(ctx->afi->family, li->type);
	if (err < 0)
		goto err1;

	return 0;

err1:
	if (priv->prefix != nft_log_null_prefix)
		kfree(priv->prefix);
	return err;
}

static void nft_log_destroy(const struct nft_ctx *ctx,
			    const struct nft_expr *expr)
{
	struct nft_log *priv = nft_expr_priv(expr);
	struct nf_loginfo *li = &priv->loginfo;

	if (priv->prefix != nft_log_null_prefix)
		kfree(priv->prefix);

	nf_logger_put(ctx->afi->family, li->type);
}

static int nft_log_dump(struct sk_buff *skb, const struct nft_expr *expr)
{
	const struct nft_log *priv = nft_expr_priv(expr);
	const struct nf_loginfo *li = &priv->loginfo;

	if (priv->prefix != nft_log_null_prefix)
		if (nla_put_string(skb, NFTA_LOG_PREFIX, priv->prefix))
			goto nla_put_failure;
	switch (li->type) {
	case NF_LOG_TYPE_LOG:
		if (nla_put_be32(skb, NFTA_LOG_LEVEL, htonl(li->u.log.level)))
			goto nla_put_failure;

		if (li->u.log.logflags) {
			if (nla_put_be32(skb, NFTA_LOG_FLAGS,
					 htonl(li->u.log.logflags)))
				goto nla_put_failure;
		}
		break;
	case NF_LOG_TYPE_ULOG:
		if (nla_put_be16(skb, NFTA_LOG_GROUP, htons(li->u.ulog.group)))
			goto nla_put_failure;

		if (li->u.ulog.flags & NF_LOG_F_COPY_LEN) {
			if (nla_put_be32(skb, NFTA_LOG_SNAPLEN,
					 htonl(li->u.ulog.copy_len)))
				goto nla_put_failure;
		}
		if (li->u.ulog.qthreshold) {
			if (nla_put_be16(skb, NFTA_LOG_QTHRESHOLD,
					 htons(li->u.ulog.qthreshold)))
				goto nla_put_failure;
		}
		break;
	}
	return 0;

nla_put_failure:
	return -1;
}

static struct nft_expr_type nft_log_type;
static const struct nft_expr_ops nft_log_ops = {
	.type		= &nft_log_type,
	.size		= NFT_EXPR_SIZE(sizeof(struct nft_log)),
	.eval		= nft_log_eval,
	.init		= nft_log_init,
	.destroy	= nft_log_destroy,
	.dump		= nft_log_dump,
};

static struct nft_expr_type nft_log_type __read_mostly = {
	.name		= "log",
	.ops		= &nft_log_ops,
	.policy		= nft_log_policy,
	.maxattr	= NFTA_LOG_MAX,
	.owner		= THIS_MODULE,
};

static int __init nft_log_module_init(void)
{
	return nft_register_expr(&nft_log_type);
}

static void __exit nft_log_module_exit(void)
{
	nft_unregister_expr(&nft_log_type);
}

module_init(nft_log_module_init);
module_exit(nft_log_module_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
MODULE_ALIAS_NFT_EXPR("log");
Commit	Line	Data
96518518	1	/*
ef1f7df9	2	* Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net>
09d27b88	3	* Copyright (c) 2012-2014 Pablo Neira Ayuso <pablo@netfilter.org>
96518518 PM	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License version 2 as
	7	* published by the Free Software Foundation.
	8	*
	9	* Development of this code funded by Astaro AG (http://www.astaro.com/)
	10	*/
	11
	12	#include <linux/kernel.h>
	13	#include <linux/init.h>
	14	#include <linux/module.h>
	15	#include <linux/netlink.h>
	16	#include <linux/netfilter.h>
	17	#include <linux/netfilter/nf_tables.h>
	18	#include <net/netfilter/nf_tables.h>
	19	#include <net/netfilter/nf_log.h>
	20	#include <linux/netdevice.h>
	21
	22	static const char *nft_log_null_prefix = "";
	23
	24	struct nft_log {
	25	struct nf_loginfo loginfo;
	26	char *prefix;
96518518 PM	27	};
	28
	29	static void nft_log_eval(const struct nft_expr *expr,
a55e22e9	30	struct nft_regs *regs,
96518518 PM	31	const struct nft_pktinfo *pkt)
	32	{
	33	const struct nft_log *priv = nft_expr_priv(expr);
96518518	34
0e5a1c7e PNA	35	nf_log_packet(nft_net(pkt), nft_pf(pkt), nft_hook(pkt), pkt->skb,
	36	nft_in(pkt), nft_out(pkt), &priv->loginfo, "%s",
	37	priv->prefix);
96518518 PM	38	}
	39
	40	static const struct nla_policy nft_log_policy[NFTA_LOG_MAX + 1] = {
	41	[NFTA_LOG_GROUP] = { .type = NLA_U16 },
5ce6b04c LZ	42	[NFTA_LOG_PREFIX] = { .type = NLA_STRING,
5ce6b04c LZ	43	.len = NF_LOG_PREFIXLEN - 1 },
96518518 PM	44	[NFTA_LOG_SNAPLEN] = { .type = NLA_U32 },
96518518 PM	45	[NFTA_LOG_QTHRESHOLD] = { .type = NLA_U16 },
09d27b88 PNA	46	[NFTA_LOG_LEVEL] = { .type = NLA_U32 },
09d27b88 PNA	47	[NFTA_LOG_FLAGS] = { .type = NLA_U32 },
96518518 PM	48	};
	49
	50	static int nft_log_init(const struct nft_ctx *ctx,
	51	const struct nft_expr *expr,
	52	const struct nlattr * const tb[])
	53	{
	54	struct nft_log *priv = nft_expr_priv(expr);
	55	struct nf_loginfo *li = &priv->loginfo;
	56	const struct nlattr *nla;
c2d9a429 LZ	57	int err;
	58
	59	li->type = NF_LOG_TYPE_LOG;
	60	if (tb[NFTA_LOG_LEVEL] != NULL &&
	61	tb[NFTA_LOG_GROUP] != NULL)
	62	return -EINVAL;
ff107d27	63	if (tb[NFTA_LOG_GROUP] != NULL) {
c2d9a429	64	li->type = NF_LOG_TYPE_ULOG;
ff107d27 LZ	65	if (tb[NFTA_LOG_FLAGS] != NULL)
	66	return -EINVAL;
	67	}
96518518	68
96518518 PM	69	nla = tb[NFTA_LOG_PREFIX];
	70	if (nla != NULL) {
	71	priv->prefix = kmalloc(nla_len(nla) + 1, GFP_KERNEL);
	72	if (priv->prefix == NULL)
	73	return -ENOMEM;
	74	nla_strlcpy(priv->prefix, nla, nla_len(nla) + 1);
09d27b88	75	} else {
96518518	76	priv->prefix = (char *)nft_log_null_prefix;
09d27b88	77	}
96518518	78
09d27b88 PNA	79	switch (li->type) {
	80	case NF_LOG_TYPE_LOG:
	81	if (tb[NFTA_LOG_LEVEL] != NULL) {
	82	li->u.log.level =
5cbfda20	83	ntohl(nla_get_be32(tb[NFTA_LOG_LEVEL]));
09d27b88	84	} else {
a81b2ce8	85	li->u.log.level = LOGLEVEL_WARNING;
09d27b88	86	}
1bc4e013 LZ	87	if (li->u.log.level > LOGLEVEL_DEBUG) {
	88	err = -EINVAL;
	89	goto err1;
	90	}
	91
09d27b88 PNA	92	if (tb[NFTA_LOG_FLAGS] != NULL) {
	93	li->u.log.logflags =
	94	ntohl(nla_get_be32(tb[NFTA_LOG_FLAGS]));
ff107d27 LZ	95	if (li->u.log.logflags & ~NF_LOG_MASK) {
	96	err = -EINVAL;
	97	goto err1;
	98	}
09d27b88 PNA	99	}
	100	break;
	101	case NF_LOG_TYPE_ULOG:
96518518	102	li->u.ulog.group = ntohs(nla_get_be16(tb[NFTA_LOG_GROUP]));
09d27b88	103	if (tb[NFTA_LOG_SNAPLEN] != NULL) {
cc37c1ad	104	li->u.ulog.flags \|= NF_LOG_F_COPY_LEN;
09d27b88 PNA	105	li->u.ulog.copy_len =
	106	ntohl(nla_get_be32(tb[NFTA_LOG_SNAPLEN]));
	107	}
	108	if (tb[NFTA_LOG_QTHRESHOLD] != NULL) {
	109	li->u.ulog.qthreshold =
	110	ntohs(nla_get_be16(tb[NFTA_LOG_QTHRESHOLD]));
	111	}
	112	break;
96518518 PM	113	}
96518518 PM	114
c2d9a429 LZ	115	err = nf_logger_find_get(ctx->afi->family, li->type);
	116	if (err < 0)
	117	goto err1;
	118
	119	return 0;
	120
	121	err1:
	122	if (priv->prefix != nft_log_null_prefix)
	123	kfree(priv->prefix);
	124	return err;
96518518 PM	125	}
96518518 PM	126
62472bce PM	127	static void nft_log_destroy(const struct nft_ctx *ctx,
62472bce PM	128	const struct nft_expr *expr)
96518518 PM	129	{
96518518 PM	130	struct nft_log *priv = nft_expr_priv(expr);
85d30e24	131	struct nf_loginfo *li = &priv->loginfo;
96518518 PM	132
	133	if (priv->prefix != nft_log_null_prefix)
	134	kfree(priv->prefix);
85d30e24	135
f3bb5333	136	nf_logger_put(ctx->afi->family, li->type);
96518518 PM	137	}
	138
	139	static int nft_log_dump(struct sk_buff skb, const struct nft_expr expr)
	140	{
	141	const struct nft_log *priv = nft_expr_priv(expr);
	142	const struct nf_loginfo *li = &priv->loginfo;
	143
	144	if (priv->prefix != nft_log_null_prefix)
	145	if (nla_put_string(skb, NFTA_LOG_PREFIX, priv->prefix))
	146	goto nla_put_failure;
09d27b88 PNA	147	switch (li->type) {
	148	case NF_LOG_TYPE_LOG:
	149	if (nla_put_be32(skb, NFTA_LOG_LEVEL, htonl(li->u.log.level)))
96518518	150	goto nla_put_failure;
09d27b88 PNA	151
	152	if (li->u.log.logflags) {
	153	if (nla_put_be32(skb, NFTA_LOG_FLAGS,
	154	htonl(li->u.log.logflags)))
	155	goto nla_put_failure;
	156	}
	157	break;
	158	case NF_LOG_TYPE_ULOG:
	159	if (nla_put_be16(skb, NFTA_LOG_GROUP, htons(li->u.ulog.group)))
96518518	160	goto nla_put_failure;
09d27b88	161
cc37c1ad	162	if (li->u.ulog.flags & NF_LOG_F_COPY_LEN) {
09d27b88 PNA	163	if (nla_put_be32(skb, NFTA_LOG_SNAPLEN,
	164	htonl(li->u.ulog.copy_len)))
	165	goto nla_put_failure;
	166	}
	167	if (li->u.ulog.qthreshold) {
	168	if (nla_put_be16(skb, NFTA_LOG_QTHRESHOLD,
	169	htons(li->u.ulog.qthreshold)))
	170	goto nla_put_failure;
	171	}
	172	break;
	173	}
96518518 PM	174	return 0;
	175
	176	nla_put_failure:
	177	return -1;
	178	}
	179
ef1f7df9 PM	180	static struct nft_expr_type nft_log_type;
	181	static const struct nft_expr_ops nft_log_ops = {
	182	.type = &nft_log_type,
96518518	183	.size = NFT_EXPR_SIZE(sizeof(struct nft_log)),
96518518 PM	184	.eval = nft_log_eval,
	185	.init = nft_log_init,
	186	.destroy = nft_log_destroy,
	187	.dump = nft_log_dump,
ef1f7df9 PM	188	};
	189
	190	static struct nft_expr_type nft_log_type __read_mostly = {
	191	.name = "log",
	192	.ops = &nft_log_ops,
96518518 PM	193	.policy = nft_log_policy,
96518518 PM	194	.maxattr = NFTA_LOG_MAX,
ef1f7df9	195	.owner = THIS_MODULE,
96518518 PM	196	};
	197
	198	static int __init nft_log_module_init(void)
	199	{
ef1f7df9	200	return nft_register_expr(&nft_log_type);
96518518 PM	201	}
	202
	203	static void __exit nft_log_module_exit(void)
	204	{
ef1f7df9	205	nft_unregister_expr(&nft_log_type);
96518518 PM	206	}
	207
	208	module_init(nft_log_module_init);
	209	module_exit(nft_log_module_exit);
	210
	211	MODULE_LICENSE("GPL");
	212	MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
	213	MODULE_ALIAS_NFT_EXPR("log");