diff options
Diffstat (limited to 'src/utils_args.c')
| -rw-r--r-- | src/utils_args.c | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/src/utils_args.c b/src/utils_args.c new file mode 100644 index 0000000..fda2350 --- /dev/null +++ b/src/utils_args.c @@ -0,0 +1,131 @@ +/* + * Command line arguments parsing helpers + * + * Copyright (C) 2020-2023 Red Hat, Inc. All rights reserved. + * Copyright (C) 2020-2023 Ondrej Kozina + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include "cryptsetup.h" + +void tools_parse_arg_value(poptContext popt_context, crypt_arg_type_info type, struct tools_arg *arg, const char *popt_arg, int popt_val, bool(*needs_size_conv_fn)(unsigned arg_id)) +{ + char *end, msg[128]; + long long int ll; + long long unsigned int ull; + + errno = 0; + + switch (type) { + case CRYPT_ARG_BOOL: + break; + case CRYPT_ARG_STRING: + if (arg->set) + free(arg->u.str_value); + arg->u.str_value = poptGetOptArg(popt_context); + break; + case CRYPT_ARG_INT32: + ll = strtoll(popt_arg, &end, 10); + if (*end || !*popt_arg || ll > INT32_MAX || ll < INT32_MIN || errno == ERANGE) + usage(popt_context, EXIT_FAILURE, poptStrerror(POPT_ERROR_BADNUMBER), + poptGetInvocationName(popt_context)); + arg->u.i32_value = (int32_t)ll; + break; + case CRYPT_ARG_UINT32: + ull = strtoull(popt_arg, &end, 10); + if (*end || !*popt_arg || ull > UINT32_MAX || errno == ERANGE) + usage(popt_context, EXIT_FAILURE, poptStrerror(POPT_ERROR_BADNUMBER), + poptGetInvocationName(popt_context)); + arg->u.u32_value = (uint32_t)ull; + break; + case CRYPT_ARG_INT64: + ll = strtoll(popt_arg, &end, 10); + if (*end || !*popt_arg || errno == ERANGE) + usage(popt_context, EXIT_FAILURE, poptStrerror(POPT_ERROR_BADNUMBER), + poptGetInvocationName(popt_context)); + arg->u.i64_value = ll; + break; + case CRYPT_ARG_UINT64: + /* special size strings with units converted to integers */ + if (needs_size_conv_fn && needs_size_conv_fn(popt_val)) { + if (tools_string_to_size(popt_arg, &arg->u.u64_value)) { + if (snprintf(msg, sizeof(msg), _("Invalid size specification in parameter --%s."), arg->name) < 0) + msg[0] = '\0'; + usage(popt_context, EXIT_FAILURE, msg, + poptGetInvocationName(popt_context)); + } + } else { + ull = strtoull(popt_arg, &end, 10); + if (*end || !*popt_arg || errno == ERANGE) + usage(popt_context, EXIT_FAILURE, poptStrerror(POPT_ERROR_BADNUMBER), + poptGetInvocationName(popt_context)); + arg->u.u64_value = ull; + } + break; + case CRYPT_ARG_ALIAS: + tools_parse_arg_value(popt_context, arg->u.o.ptr->type, arg->u.o.ptr, popt_arg, arg->u.o.id, needs_size_conv_fn); + break; + default: + /* this signals internal tools coding mistake */ + abort(); + } + + arg->set = true; +} + +void tools_args_free(struct tools_arg *args, size_t args_size) +{ + size_t i; + + for (i = 0; i < args_size; i++) { + if (args[i].set && args[i].type == CRYPT_ARG_STRING) + free(args[i].u.str_value); + args[i].set = false; + } +} + +static bool action_allowed(const char *action, const char * const* list, size_t list_size) +{ + size_t i; + + if (!list[0]) + return true; + + for (i = 0; i < list_size && list[i]; i++) { + if (!strcmp(action, list[i])) + return true; + } + + return false; +} + +void tools_check_args(const char *action, const struct tools_arg *args, size_t args_size, poptContext popt_context) +{ + size_t i; + char msg[256]; + + for (i = 1; i < args_size; i++) { + if (args[i].set) { + if (action_allowed(action, args[i].actions_array, MAX_ACTIONS)) { + continue; + } else { + if (snprintf(msg, sizeof(msg), _("Option --%s is not allowed with %s action."), args[i].name, action) < 0) + msg[0] = '\0'; + usage(popt_context, EXIT_FAILURE, msg, poptGetInvocationName(popt_context)); + } + } + } +} |