libxt_quota.c revision 42979363f3958b4436c6d2503753c182c58e55ea
1/* 2 * Shared library add-on to iptables to add quota support 3 * 4 * Sam Johnston <samj@samj.net> 5 */ 6#include <stddef.h> 7#include <stdio.h> 8#include <stdlib.h> 9#include <getopt.h> 10#include <xtables.h> 11 12#include <linux/netfilter/xt_quota.h> 13 14static const struct option quota_opts[] = { 15 {"quota", 1, NULL, '1'}, 16 { .name = NULL } 17}; 18 19static void quota_help(void) 20{ 21 printf("quota match options:\n" 22 " --quota quota quota (bytes)\n"); 23} 24 25static void 26quota_print(const void *ip, const struct xt_entry_match *match, int numeric) 27{ 28 const struct xt_quota_info *q = (const void *)match->data; 29 printf("quota: %llu bytes", (unsigned long long) q->quota); 30} 31 32static void 33quota_save(const void *ip, const struct xt_entry_match *match) 34{ 35 const struct xt_quota_info *q = (const void *)match->data; 36 printf("--quota %llu ", (unsigned long long) q->quota); 37} 38 39/* parse quota option */ 40static int 41parse_quota(const char *s, u_int64_t * quota) 42{ 43 *quota = strtoull(s, NULL, 10); 44 45#ifdef DEBUG_XT_QUOTA 46 printf("Quota: %llu\n", *quota); 47#endif 48 49 if (*quota == UINT64_MAX) 50 xtables_error(PARAMETER_PROBLEM, "quota invalid: '%s'\n", s); 51 else 52 return 1; 53} 54 55static int 56quota_parse(int c, char **argv, int invert, unsigned int *flags, 57 const void *entry, struct xt_entry_match **match) 58{ 59 struct xt_quota_info *info = (struct xt_quota_info *) (*match)->data; 60 61 switch (c) { 62 case '1': 63 if (xtables_check_inverse(optarg, &invert, NULL, 0)) 64 xtables_error(PARAMETER_PROBLEM, "quota: unexpected '!'"); 65 if (!parse_quota(optarg, &info->quota)) 66 xtables_error(PARAMETER_PROBLEM, 67 "bad quota: '%s'", optarg); 68 break; 69 70 default: 71 return 0; 72 } 73 return 1; 74} 75 76static struct xtables_match quota_match = { 77 .family = NFPROTO_UNSPEC, 78 .name = "quota", 79 .version = XTABLES_VERSION, 80 .size = XT_ALIGN(sizeof (struct xt_quota_info)), 81 .userspacesize = offsetof(struct xt_quota_info, quota), 82 .help = quota_help, 83 .parse = quota_parse, 84 .print = quota_print, 85 .save = quota_save, 86 .extra_opts = quota_opts, 87}; 88 89void 90_init(void) 91{ 92 xtables_register_match("a_match); 93} 94