|
Packit Service |
d1fe03 |
/*
|
|
Packit Service |
d1fe03 |
* Shared library add-on to iptables to add quota support
|
|
Packit Service |
d1fe03 |
*
|
|
Packit Service |
d1fe03 |
* Sam Johnston <samj@samj.net>
|
|
Packit Service |
d1fe03 |
*/
|
|
Packit Service |
d1fe03 |
#include <stdio.h>
|
|
Packit Service |
d1fe03 |
#include <xtables.h>
|
|
Packit Service |
d1fe03 |
#include <linux/netfilter/xt_quota.h>
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
enum {
|
|
Packit Service |
d1fe03 |
O_QUOTA = 0,
|
|
Packit Service |
d1fe03 |
};
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static const struct xt_option_entry quota_opts[] = {
|
|
Packit Service |
d1fe03 |
{.name = "quota", .id = O_QUOTA, .type = XTTYPE_UINT64,
|
|
Packit Service |
d1fe03 |
.flags = XTOPT_MAND | XTOPT_INVERT | XTOPT_PUT,
|
|
Packit Service |
d1fe03 |
XTOPT_POINTER(struct xt_quota_info, quota)},
|
|
Packit Service |
d1fe03 |
XTOPT_TABLEEND,
|
|
Packit Service |
d1fe03 |
};
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static void quota_help(void)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
printf("quota match options:\n"
|
|
Packit Service |
d1fe03 |
"[!] --quota quota quota (bytes)\n");
|
|
Packit Service |
d1fe03 |
}
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static void
|
|
Packit Service |
d1fe03 |
quota_print(const void *ip, const struct xt_entry_match *match, int numeric)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
const struct xt_quota_info *q = (const void *)match->data;
|
|
Packit Service |
d1fe03 |
printf(" quota: %llu bytes", (unsigned long long)q->quota);
|
|
Packit Service |
d1fe03 |
}
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static void
|
|
Packit Service |
d1fe03 |
quota_save(const void *ip, const struct xt_entry_match *match)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
const struct xt_quota_info *q = (const void *)match->data;
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
if (q->flags & XT_QUOTA_INVERT)
|
|
Packit Service |
d1fe03 |
printf(" !");
|
|
Packit Service |
d1fe03 |
printf(" --quota %llu", (unsigned long long) q->quota);
|
|
Packit Service |
d1fe03 |
}
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static void quota_parse(struct xt_option_call *cb)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
struct xt_quota_info *info = cb->data;
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
xtables_option_parse(cb);
|
|
Packit Service |
d1fe03 |
if (cb->invert)
|
|
Packit Service |
d1fe03 |
info->flags |= XT_QUOTA_INVERT;
|
|
Packit Service |
d1fe03 |
}
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static int quota_xlate(struct xt_xlate *xl,
|
|
Packit Service |
d1fe03 |
const struct xt_xlate_mt_params *params)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
const struct xt_quota_info *q = (void *)params->match->data;
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
xt_xlate_add(xl, "quota %s%llu bytes",
|
|
Packit Service |
d1fe03 |
q->flags & XT_QUOTA_INVERT ? "over " : "",
|
|
Packit Service |
d1fe03 |
(unsigned long long) q->quota);
|
|
Packit Service |
d1fe03 |
return 1;
|
|
Packit Service |
d1fe03 |
}
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
static struct xtables_match quota_match = {
|
|
Packit Service |
d1fe03 |
.family = NFPROTO_UNSPEC,
|
|
Packit Service |
d1fe03 |
.name = "quota",
|
|
Packit Service |
d1fe03 |
.version = XTABLES_VERSION,
|
|
Packit Service |
d1fe03 |
.size = XT_ALIGN(sizeof (struct xt_quota_info)),
|
|
Packit Service |
d1fe03 |
.userspacesize = offsetof(struct xt_quota_info, master),
|
|
Packit Service |
d1fe03 |
.help = quota_help,
|
|
Packit Service |
d1fe03 |
.print = quota_print,
|
|
Packit Service |
d1fe03 |
.save = quota_save,
|
|
Packit Service |
d1fe03 |
.x6_parse = quota_parse,
|
|
Packit Service |
d1fe03 |
.x6_options = quota_opts,
|
|
Packit Service |
d1fe03 |
.xlate = quota_xlate,
|
|
Packit Service |
d1fe03 |
};
|
|
Packit Service |
d1fe03 |
|
|
Packit Service |
d1fe03 |
void
|
|
Packit Service |
d1fe03 |
_init(void)
|
|
Packit Service |
d1fe03 |
{
|
|
Packit Service |
d1fe03 |
xtables_register_match("a_match);
|
|
Packit Service |
d1fe03 |
}
|