![Jan Engelhardt](/assets/img/avatar_default.png)
The function signatures for Xtables extensions have grown over time. It involves a lot of typing/replication, and also a bit of stack space even if they are not used. Realize an NFWS2008 idea and pack them into structs. The skb remains outside of the struct so gcc can continue to apply its optimizations. This patch does this for match extensions' match functions. A few ambiguities have also been addressed. The "offset" parameter for example has been renamed to "fragoff" (there are so many different offsets already) and "protoff" to "thoff" (there is more than just one protocol here, so clarify). Signed-off-by: Jan Engelhardt <jengelh@medozas.de> Signed-off-by: Patrick McHardy <kaber@trash.net>
76 lines
1.7 KiB
C
76 lines
1.7 KiB
C
/*
|
|
* netfilter module to enforce network quotas
|
|
*
|
|
* Sam Johnston <samj@samj.net>
|
|
*/
|
|
#include <linux/skbuff.h>
|
|
#include <linux/spinlock.h>
|
|
|
|
#include <linux/netfilter/x_tables.h>
|
|
#include <linux/netfilter/xt_quota.h>
|
|
|
|
MODULE_LICENSE("GPL");
|
|
MODULE_AUTHOR("Sam Johnston <samj@samj.net>");
|
|
MODULE_DESCRIPTION("Xtables: countdown quota match");
|
|
MODULE_ALIAS("ipt_quota");
|
|
MODULE_ALIAS("ip6t_quota");
|
|
|
|
static DEFINE_SPINLOCK(quota_lock);
|
|
|
|
static bool
|
|
quota_mt(const struct sk_buff *skb, const struct xt_match_param *par)
|
|
{
|
|
struct xt_quota_info *q =
|
|
((const struct xt_quota_info *)par->matchinfo)->master;
|
|
bool ret = q->flags & XT_QUOTA_INVERT;
|
|
|
|
spin_lock_bh("a_lock);
|
|
if (q->quota >= skb->len) {
|
|
q->quota -= skb->len;
|
|
ret = !ret;
|
|
} else {
|
|
/* we do not allow even small packets from now on */
|
|
q->quota = 0;
|
|
}
|
|
spin_unlock_bh("a_lock);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static bool
|
|
quota_mt_check(const char *tablename, const void *entry,
|
|
const struct xt_match *match, void *matchinfo,
|
|
unsigned int hook_mask)
|
|
{
|
|
struct xt_quota_info *q = matchinfo;
|
|
|
|
if (q->flags & ~XT_QUOTA_MASK)
|
|
return false;
|
|
/* For SMP, we only want to use one set of counters. */
|
|
q->master = q;
|
|
return true;
|
|
}
|
|
|
|
static struct xt_match quota_mt_reg __read_mostly = {
|
|
.name = "quota",
|
|
.revision = 0,
|
|
.family = NFPROTO_UNSPEC,
|
|
.match = quota_mt,
|
|
.checkentry = quota_mt_check,
|
|
.matchsize = sizeof(struct xt_quota_info),
|
|
.me = THIS_MODULE,
|
|
};
|
|
|
|
static int __init quota_mt_init(void)
|
|
{
|
|
return xt_register_match("a_mt_reg);
|
|
}
|
|
|
|
static void __exit quota_mt_exit(void)
|
|
{
|
|
xt_unregister_match("a_mt_reg);
|
|
}
|
|
|
|
module_init(quota_mt_init);
|
|
module_exit(quota_mt_exit);
|