From fb7228d4781b181b078404b82988c456b58430de Mon Sep 17 00:00:00 2001 From: Phil Sutter Date: Wed, 29 Nov 2023 17:35:48 +0100 Subject: extensions: libebt_mark_m: Use guided option parser Can't use XTTYPE_MARKMASK32 here because in 'val/mask', 'val' is optional. Would have to extend xtopt_parse_markmask() to accept this, maybe guarded by a new XTOPT_ flag to avoid unexpected changes in behaviour? --- extensions/libebt_mark_m.c | 58 ++++++++++++++++++---------------------------- 1 file changed, 22 insertions(+), 36 deletions(-) (limited to 'extensions/libebt_mark_m.c') diff --git a/extensions/libebt_mark_m.c b/extensions/libebt_mark_m.c index 178c9ece..f17fe99a 100644 --- a/extensions/libebt_mark_m.c +++ b/extensions/libebt_mark_m.c @@ -12,15 +12,17 @@ #include #include #include -#include #include #include -#define MARK '1' +enum { + O_MARK = 0, +}; -static const struct option brmark_m_opts[] = { - { .name = "mark", .has_arg = true, .val = MARK }, - XT_GETOPT_TABLEEND, +static const struct xt_option_entry brmark_m_opts[] = { + { .name = "mark", .id = O_MARK, .type = XTTYPE_STRING, + .flags = XTOPT_INVERT | XTOPT_MAND }, + XTOPT_TABLEEND, }; static void brmark_m_print_help(void) @@ -30,45 +32,30 @@ static void brmark_m_print_help(void) "--mark [!] [value][/mask]: Match nfmask value (see man page)\n"); } -#define OPT_MARK 0x01 -static int -brmark_m_parse(int c, char **argv, int invert, unsigned int *flags, - const void *entry, struct xt_entry_match **match) +static void brmark_m_parse(struct xt_option_call *cb) { - struct ebt_mark_m_info *info = (struct ebt_mark_m_info *) - (*match)->data; + struct ebt_mark_m_info *info = cb->data; char *end; - switch (c) { - case MARK: - if (invert) - info->invert = 1; - info->mark = strtoul(optarg, &end, 0); + xtables_option_parse(cb); + + switch (cb->entry->id) { + case O_MARK: + info->invert = cb->invert; + info->mark = strtoul(cb->arg, &end, 0); info->bitmask = EBT_MARK_AND; if (*end == '/') { - if (end == optarg) + if (end == cb->arg) info->bitmask = EBT_MARK_OR; info->mask = strtoul(end+1, &end, 0); } else { - info->mask = 0xffffffff; + info->mask = UINT32_MAX; } - if (*end != '\0' || end == optarg) + if (*end != '\0' || end == cb->arg) xtables_error(PARAMETER_PROBLEM, "Bad mark value '%s'", - optarg); + cb->arg); break; - default: - return 0; } - - *flags |= info->bitmask; - return 1; -} - -static void brmark_m_final_check(unsigned int flags) -{ - if (!flags) - xtables_error(PARAMETER_PROBLEM, - "You must specify proper arguments"); } static void brmark_m_print(const void *ip, const struct xt_entry_match *match, @@ -101,7 +88,7 @@ static int brmark_m_xlate(struct xt_xlate *xl, if (info->bitmask == EBT_MARK_OR) { xt_xlate_add(xl, "and 0x%x %s0 ", (uint32_t)info->mask, info->invert ? "" : "!= "); - } else if (info->mask != 0xffffffffU) { + } else if (info->mask != UINT32_MAX) { xt_xlate_add(xl, "and 0x%x %s0x%x ", (uint32_t)info->mask, op == XT_OP_EQ ? "" : "!= ", (uint32_t)info->mark); } else { @@ -119,11 +106,10 @@ static struct xtables_match brmark_m_match = { .size = XT_ALIGN(sizeof(struct ebt_mark_m_info)), .userspacesize = XT_ALIGN(sizeof(struct ebt_mark_m_info)), .help = brmark_m_print_help, - .parse = brmark_m_parse, - .final_check = brmark_m_final_check, + .x6_parse = brmark_m_parse, .print = brmark_m_print, .xlate = brmark_m_xlate, - .extra_opts = brmark_m_opts, + .x6_options = brmark_m_opts, }; void _init(void) -- cgit v1.2.3