diff options
author | Jan Engelhardt <jengelh@medozas.de> | 2011-05-07 03:18:11 +0200 |
---|---|---|
committer | Jan Engelhardt <jengelh@medozas.de> | 2011-05-09 17:11:42 +0200 |
commit | 7e79d139c1ea6e1b72bbedc53c0426c9d5ffa0e0 (patch) | |
tree | 81fdd0a3d551f2f15a3efd3d67f44b2a0ec68945 /extensions/libxt_mac.c | |
parent | cb225e26856accf5661dcbc3cf34d7f77b2f0c36 (diff) |
libxt_mac: use guided option parser
Signed-off-by: Jan Engelhardt <jengelh@medozas.de>
Diffstat (limited to 'extensions/libxt_mac.c')
-rw-r--r-- | extensions/libxt_mac.c | 80 |
1 files changed, 20 insertions, 60 deletions
diff --git a/extensions/libxt_mac.c b/extensions/libxt_mac.c index d7e65dac..f171d153 100644 --- a/extensions/libxt_mac.c +++ b/extensions/libxt_mac.c @@ -1,10 +1,4 @@ -/* Shared library add-on to iptables to add MAC address support. */ -#include <stdbool.h> #include <stdio.h> -#include <netdb.h> -#include <string.h> -#include <stdlib.h> -#include <getopt.h> #if defined(__GLIBC__) && __GLIBC__ == 2 #include <net/ethernet.h> #else @@ -13,6 +7,10 @@ #include <xtables.h> #include <linux/netfilter/xt_mac.h> +enum { + O_MAC = 0, +}; + static void mac_help(void) { printf( @@ -21,70 +19,33 @@ static void mac_help(void) " Match source MAC address\n"); } -static const struct option mac_opts[] = { - {.name = "mac-source", .has_arg = true, .val = '1'}, - XT_GETOPT_TABLEEND, +#define s struct xt_mac_info +static const struct xt_option_entry mac_opts[] = { + {.name = "mac-source", .id = O_MAC, .type = XTTYPE_ETHERMAC, + .flags = XTOPT_MAND | XTOPT_INVERT | XTOPT_PUT, + XTOPT_POINTER(s, srcaddr)}, + XTOPT_TABLEEND, }; +#undef s -static void -parse_mac(const char *mac, struct xt_mac_info *info) -{ - unsigned int i = 0; - - if (strlen(mac) != ETH_ALEN*3-1) - xtables_error(PARAMETER_PROBLEM, "Bad mac address \"%s\"", mac); - - for (i = 0; i < ETH_ALEN; i++) { - long number; - char *end; - - number = strtol(mac + i*3, &end, 16); - - if (end == mac + i*3 + 2 - && number >= 0 - && number <= 255) - info->srcaddr[i] = number; - else - xtables_error(PARAMETER_PROBLEM, - "Bad mac address `%s'", mac); - } -} - -static int -mac_parse(int c, char **argv, int invert, unsigned int *flags, - const void *entry, struct xt_entry_match **match) +static void mac_parse(struct xt_option_call *cb) { - struct xt_mac_info *macinfo = (struct xt_mac_info *)(*match)->data; - - switch (c) { - case '1': - xtables_check_inverse(optarg, &invert, &optind, 0, argv); - parse_mac(optarg, macinfo); - if (invert) - macinfo->invert = 1; - *flags = 1; - break; - } + struct xt_mac_info *macinfo = cb->data; - return 1; + xtables_option_parse(cb); + if (cb->invert) + macinfo->invert = 1; } -static void print_mac(const unsigned char macaddress[ETH_ALEN]) +static void print_mac(const unsigned char *macaddress) { unsigned int i; printf(" %02X", macaddress[0]); - for (i = 1; i < ETH_ALEN; i++) + for (i = 1; i < ETH_ALEN; ++i) printf(":%02X", macaddress[i]); } -static void mac_check(unsigned int flags) -{ - if (!flags) - xtables_error(PARAMETER_PROBLEM, - "You must specify `--mac-source'"); -} - static void mac_print(const void *ip, const struct xt_entry_match *match, int numeric) { @@ -115,11 +76,10 @@ static struct xtables_match mac_match = { .size = XT_ALIGN(sizeof(struct xt_mac_info)), .userspacesize = XT_ALIGN(sizeof(struct xt_mac_info)), .help = mac_help, - .parse = mac_parse, - .final_check = mac_check, + .x6_parse = mac_parse, .print = mac_print, .save = mac_save, - .extra_opts = mac_opts, + .x6_options = mac_opts, }; void _init(void) |