Merge commit 'origin/trunk@12184' into fedora

[iptables.git] / trunk / extensions / libxt_quota.c

diff --git a/trunk/extensions/libxt_quota.c b/trunk/extensions/libxt_quota.c
new file mode 100644 (file)
index 0000000..8e178fe
--- /dev/null
+++ b/trunk/extensions/libxt_quota.c
@@ -0,0 +1,97 @@
+/*
+ * Shared library add-on to iptables to add quota support
+ *
+ * Sam Johnston <samj@samj.net>
+ */
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <getopt.h>
+#include <xtables.h>
+
+#include <linux/netfilter/xt_quota.h>
+
+static const struct option quota_opts[] = {
+       {"quota", 1, NULL, '1'},
+       { .name = NULL }
+};
+
+/* print usage */
+static void quota_help(void)
+{
+       printf("quota match options:\n"
+              " --quota quota                  quota (bytes)\n");
+}
+
+/* print matchinfo */
+static void
+quota_print(const void *ip, const struct xt_entry_match *match, int numeric)
+{
+       struct xt_quota_info *q = (struct xt_quota_info *) match->data;
+       printf("quota: %llu bytes", (unsigned long long) q->quota);
+}
+
+/* save matchinfo */
+static void
+quota_save(const void *ip, const struct xt_entry_match *match)
+{
+       struct xt_quota_info *q = (struct xt_quota_info *) match->data;
+       printf("--quota %llu ", (unsigned long long) q->quota);
+}
+
+/* parse quota option */
+static int
+parse_quota(const char *s, u_int64_t * quota)
+{
+       *quota = strtoull(s, NULL, 10);
+
+#ifdef DEBUG_XT_QUOTA
+       printf("Quota: %llu\n", *quota);
+#endif
+
+       if (*quota == (u_int64_t)-1)
+               exit_error(PARAMETER_PROBLEM, "quota invalid: '%s'\n", s);
+       else
+               return 1;
+}
+
+/* parse all options, returning true if we found any for us */
+static int
+quota_parse(int c, char **argv, int invert, unsigned int *flags,
+           const void *entry, struct xt_entry_match **match)
+{
+       struct xt_quota_info *info = (struct xt_quota_info *) (*match)->data;
+
+       switch (c) {
+       case '1':
+               if (check_inverse(optarg, &invert, NULL, 0))
+                       exit_error(PARAMETER_PROBLEM, "quota: unexpected '!'");
+               if (!parse_quota(optarg, &info->quota))
+                       exit_error(PARAMETER_PROBLEM,
+                                  "bad quota: '%s'", optarg);
+               break;
+
+       default:
+               return 0;
+       }
+       return 1;
+}
+
+static struct xtables_match quota_match = {
+       .family         = AF_UNSPEC,
+       .name           = "quota",
+       .version        = XTABLES_VERSION,
+       .size           = XT_ALIGN(sizeof (struct xt_quota_info)),
+       .userspacesize  = offsetof(struct xt_quota_info, quota),
+       .help           = quota_help,
+       .parse          = quota_parse,
+       .print          = quota_print,
+       .save           = quota_save,
+       .extra_opts     = quota_opts,
+};
+
+void
+_init(void)
+{
+       xtables_register_match(&quota_match);
+}