2019-05-20 09:19:13 +02:00
|
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
2011-12-23 14:28:59 +01:00
|
|
|
/*
|
|
|
|
* (C) 2011 Pablo Neira Ayuso <pablo@netfilter.org>
|
|
|
|
* (C) 2011 Intra2net AG <http://www.intra2net.com>
|
|
|
|
*/
|
2018-02-09 15:52:07 +01:00
|
|
|
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
|
|
|
|
|
2011-12-23 14:28:59 +01:00
|
|
|
#include <linux/module.h>
|
|
|
|
#include <linux/skbuff.h>
|
|
|
|
|
|
|
|
#include <linux/netfilter/x_tables.h>
|
|
|
|
#include <linux/netfilter/nfnetlink_acct.h>
|
|
|
|
#include <linux/netfilter/xt_nfacct.h>
|
|
|
|
|
|
|
|
MODULE_AUTHOR("Pablo Neira Ayuso <pablo@netfilter.org>");
|
|
|
|
MODULE_DESCRIPTION("Xtables: match for the extended accounting infrastructure");
|
|
|
|
MODULE_LICENSE("GPL");
|
|
|
|
MODULE_ALIAS("ipt_nfacct");
|
|
|
|
MODULE_ALIAS("ip6t_nfacct");
|
|
|
|
|
|
|
|
static bool nfacct_mt(const struct sk_buff *skb, struct xt_action_param *par)
|
|
|
|
{
|
2014-04-21 02:57:36 +02:00
|
|
|
int overquota;
|
2011-12-23 14:28:59 +01:00
|
|
|
const struct xt_nfacct_match_info *info = par->targinfo;
|
|
|
|
|
|
|
|
nfnl_acct_update(skb, info->nfacct);
|
|
|
|
|
2018-02-11 11:17:20 +01:00
|
|
|
overquota = nfnl_acct_overquota(xt_net(par), info->nfacct);
|
2014-04-21 02:57:36 +02:00
|
|
|
|
|
|
|
return overquota == NFACCT_UNDERQUOTA ? false : true;
|
2011-12-23 14:28:59 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
nfacct_mt_checkentry(const struct xt_mtchk_param *par)
|
|
|
|
{
|
|
|
|
struct xt_nfacct_match_info *info = par->matchinfo;
|
|
|
|
struct nf_acct *nfacct;
|
|
|
|
|
2015-08-05 17:51:45 +02:00
|
|
|
nfacct = nfnl_acct_find_get(par->net, info->name);
|
2011-12-23 14:28:59 +01:00
|
|
|
if (nfacct == NULL) {
|
2018-02-09 15:52:07 +01:00
|
|
|
pr_info_ratelimited("accounting object `%s' does not exists\n",
|
|
|
|
info->name);
|
2011-12-23 14:28:59 +01:00
|
|
|
return -ENOENT;
|
|
|
|
}
|
|
|
|
info->nfacct = nfacct;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
nfacct_mt_destroy(const struct xt_mtdtor_param *par)
|
|
|
|
{
|
|
|
|
const struct xt_nfacct_match_info *info = par->matchinfo;
|
|
|
|
|
|
|
|
nfnl_acct_put(info->nfacct);
|
|
|
|
}
|
|
|
|
|
|
|
|
static struct xt_match nfacct_mt_reg __read_mostly = {
|
|
|
|
.name = "nfacct",
|
|
|
|
.family = NFPROTO_UNSPEC,
|
|
|
|
.checkentry = nfacct_mt_checkentry,
|
|
|
|
.match = nfacct_mt,
|
|
|
|
.destroy = nfacct_mt_destroy,
|
|
|
|
.matchsize = sizeof(struct xt_nfacct_match_info),
|
2018-01-29 13:21:20 +01:00
|
|
|
.usersize = offsetof(struct xt_nfacct_match_info, nfacct),
|
2011-12-23 14:28:59 +01:00
|
|
|
.me = THIS_MODULE,
|
|
|
|
};
|
|
|
|
|
|
|
|
static int __init nfacct_mt_init(void)
|
|
|
|
{
|
|
|
|
return xt_register_match(&nfacct_mt_reg);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void __exit nfacct_mt_exit(void)
|
|
|
|
{
|
|
|
|
xt_unregister_match(&nfacct_mt_reg);
|
|
|
|
}
|
|
|
|
|
|
|
|
module_init(nfacct_mt_init);
|
|
|
|
module_exit(nfacct_mt_exit);
|