/* * Copyright (C) 2001-2004 Sistina Software, Inc. All rights reserved. * Copyright (C) 2004-2012 Red Hat, Inc. All rights reserved. * * This file is part of LVM2. * * This copyrighted material is made available to anyone wishing to use, * modify, copy, or redistribute it subject to the terms and conditions * of the GNU Lesser General Public License v.2.1. * * You should have received a copy of the GNU Lesser 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 "tools.h" #include "lvm2cmdline.h" #include "label.h" #include "lvm-version.h" #include "lvmlockd.h" #include "stub.h" #include "last-path-component.h" #include "format1.h" #include #include #include #include #include #include #include #ifdef HAVE_VALGRIND #include #endif #ifdef HAVE_GETOPTLONG # include # define GETOPTLONG_FN(a, b, c, d, e) getopt_long((a), (b), (c), (d), (e)) # define OPTIND_INIT 0 #else struct option { }; extern int optind; extern char *optarg; # define GETOPTLONG_FN(a, b, c, d, e) getopt((a), (b), (c)) # define OPTIND_INIT 1 #endif /* * Table of valid switches */ static struct arg_props _arg_props[ARG_COUNT + 1] = { #define arg(a, b, c, d, e, f) {b, "", "--" c, d, e, f}, #include "args.h" #undef arg }; static struct cmdline_context _cmdline; /* Command line args */ unsigned arg_count(const struct cmd_context *cmd, int a) { return cmd->arg_values ? cmd->arg_values[a].count : 0; } unsigned grouped_arg_count(const struct arg_values *av, int a) { return av ? av[a].count : 0; } unsigned arg_is_set(const struct cmd_context *cmd, int a) { return arg_count(cmd, a) ? 1 : 0; } int arg_from_list_is_set(const struct cmd_context *cmd, const char *err_found, ...) { int arg; va_list ap; va_start(ap, err_found); while ((arg = va_arg(ap, int)) != -1 && !arg_is_set(cmd, arg)) /* empty */; va_end(ap); if (arg == -1) return 0; if (err_found) log_error("%s %s.", arg_long_option_name(arg), err_found); return 1; } int arg_outside_list_is_set(const struct cmd_context *cmd, const char *err_found, ...) { int i, arg; va_list ap; for (i = 0; i < ARG_COUNT; ++i) { switch (i) { /* skip common options */ case commandprofile_ARG: case config_ARG: case debug_ARG: case driverloaded_ARG: case help2_ARG: case help_ARG: case profile_ARG: case quiet_ARG: case verbose_ARG: case version_ARG: case yes_ARG: continue; } if (!arg_is_set(cmd, i)) continue; /* unset */ va_start(ap, err_found); while (((arg = va_arg(ap, int)) != -1) && (arg != i)) /* empty */; va_end(ap); if (arg == i) continue; /* set and in list */ if (err_found) log_error("Option %s %s.", arg_long_option_name(i), err_found); return 1; } return 0; } int arg_from_list_is_negative(const struct cmd_context *cmd, const char *err_found, ...) { int arg, ret = 0; va_list ap; va_start(ap, err_found); while ((arg = va_arg(ap, int)) != -1) if (arg_sign_value(cmd, arg, SIGN_NONE) == SIGN_MINUS) { if (err_found) log_error("%s %s.", arg_long_option_name(arg), err_found); ret = 1; } va_end(ap); return ret; } int arg_from_list_is_zero(const struct cmd_context *cmd, const char *err_found, ...) { int arg, ret = 0; va_list ap; va_start(ap, err_found); while ((arg = va_arg(ap, int)) != -1) if (arg_is_set(cmd, arg) && !arg_int_value(cmd, arg, 0)) { if (err_found) log_error("%s %s.", arg_long_option_name(arg), err_found); ret = 1; } va_end(ap); return ret; } unsigned grouped_arg_is_set(const struct arg_values *av, int a) { return grouped_arg_count(av, a) ? 1 : 0; } const char *arg_long_option_name(int a) { return _cmdline.arg_props[a].long_arg; } const char *arg_value(const struct cmd_context *cmd, int a) { return cmd->arg_values ? cmd->arg_values[a].value : NULL; } const char *arg_str_value(const struct cmd_context *cmd, int a, const char *def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].value : def; } const char *grouped_arg_str_value(const struct arg_values *av, int a, const char *def) { return grouped_arg_count(av, a) ? av[a].value : def; } int32_t grouped_arg_int_value(const struct arg_values *av, int a, const int32_t def) { return grouped_arg_count(av, a) ? av[a].i_value : def; } int32_t first_grouped_arg_int_value(const struct cmd_context *cmd, int a, const int32_t def) { struct arg_value_group_list *current_group; struct arg_values *av; dm_list_iterate_items(current_group, &cmd->arg_value_groups) { av = current_group->arg_values; if (grouped_arg_count(av, a)) return grouped_arg_int_value(av, a, def); } return def; } int32_t arg_int_value(const struct cmd_context *cmd, int a, const int32_t def) { return (_cmdline.arg_props[a].flags & ARG_GROUPABLE) ? first_grouped_arg_int_value(cmd, a, def) : (arg_is_set(cmd, a) ? cmd->arg_values[a].i_value : def); } uint32_t arg_uint_value(const struct cmd_context *cmd, int a, const uint32_t def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].ui_value : def; } int64_t arg_int64_value(const struct cmd_context *cmd, int a, const int64_t def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].i64_value : def; } uint64_t arg_uint64_value(const struct cmd_context *cmd, int a, const uint64_t def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].ui64_value : def; } /* No longer used. const void *arg_ptr_value(struct cmd_context *cmd, int a, const void *def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].ptr : def; } */ sign_t arg_sign_value(const struct cmd_context *cmd, int a, const sign_t def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].sign : def; } percent_type_t arg_percent_value(const struct cmd_context *cmd, int a, const percent_type_t def) { return arg_is_set(cmd, a) ? cmd->arg_values[a].percent : def; } int arg_count_increment(struct cmd_context *cmd, int a) { return cmd->arg_values[a].count++; } int yes_no_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { av->sign = SIGN_NONE; av->percent = PERCENT_NONE; if (!strcmp(av->value, "y")) { av->i_value = 1; av->ui_value = 1; } else if (!strcmp(av->value, "n")) { av->i_value = 0; av->ui_value = 0; } else return 0; return 1; } int activation_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { av->sign = SIGN_NONE; av->percent = PERCENT_NONE; if (!strcmp(av->value, "e") || !strcmp(av->value, "ey") || !strcmp(av->value, "ye")) { av->i_value = CHANGE_AEY; av->ui_value = CHANGE_AEY; } else if (!strcmp(av->value, "s") || !strcmp(av->value, "sy") || !strcmp(av->value, "ys")) { av->i_value = CHANGE_ASY; av->ui_value = CHANGE_ASY; } else if (!strcmp(av->value, "y")) { av->i_value = CHANGE_AY; av->ui_value = CHANGE_AY; } else if (!strcmp(av->value, "a") || !strcmp(av->value, "ay") || !strcmp(av->value, "ya")) { av->i_value = CHANGE_AAY; av->ui_value = CHANGE_AAY; } else if (!strcmp(av->value, "n") || !strcmp(av->value, "en") || !strcmp(av->value, "ne")) { av->i_value = CHANGE_AN; av->ui_value = CHANGE_AN; } else if (!strcmp(av->value, "ln") || !strcmp(av->value, "nl")) { av->i_value = CHANGE_ALN; av->ui_value = CHANGE_ALN; } else if (!strcmp(av->value, "ly") || !strcmp(av->value, "yl")) { av->i_value = CHANGE_ALY; av->ui_value = CHANGE_ALY; } else return 0; return 1; } int cachemode_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { cache_mode_t mode; if (!set_cache_mode(&mode, av->value)) return_0; av->i_value = mode; av->ui_value = mode; return 1; } int discards_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { thin_discards_t discards; if (!set_pool_discards(&discards, av->value)) return_0; av->i_value = discards; av->ui_value = discards; return 1; } int mirrorlog_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { int log_count; if (!set_mirror_log_count(&log_count, av->value)) return_0; av->i_value = log_count; av->ui_value = log_count; return 1; } int metadatatype_arg(struct cmd_context *cmd, struct arg_values *av) { return get_format_by_name(cmd, av->value) ? 1 : 0; } static int _get_int_arg(struct arg_values *av, char **ptr) { char *val; unsigned long long v; av->percent = PERCENT_NONE; val = av->value; switch (*val) { case '+': av->sign = SIGN_PLUS; val++; break; case '-': av->sign = SIGN_MINUS; val++; break; default: av->sign = SIGN_NONE; } if (!isdigit(*val)) return 0; errno = 0; v = strtoull(val, ptr, 10); if (*ptr == val || errno) return 0; av->i_value = (int32_t) v; av->ui_value = (uint32_t) v; av->i64_value = (int64_t) v; av->ui64_value = (uint64_t) v; return 1; } static int _get_percent_arg(struct arg_values *av, const char *ptr) { if (!strcasecmp(ptr, "V") || !strcasecmp(ptr, "VG")) av->percent = PERCENT_VG; else if (!strcasecmp(ptr, "L") || !strcasecmp(ptr, "LV")) av->percent = PERCENT_LV; else if (!strcasecmp(ptr, "P") || !strcasecmp(ptr, "PV") || !strcasecmp(ptr, "PVS")) av->percent = PERCENT_PVS; else if (!strcasecmp(ptr, "F") || !strcasecmp(ptr, "FR") || !strcasecmp(ptr, "FREE")) av->percent = PERCENT_FREE; else if (!strcasecmp(ptr, "O") || !strcasecmp(ptr, "OR") || !strcasecmp(ptr, "ORIGIN")) av->percent = PERCENT_ORIGIN; else { log_error("Specified %%%s is unknown.", ptr); return 0; } return 1; } /* Size stored in sectors */ static int _size_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av, int factor, int percent) { char *ptr; int i; static const char *suffixes = "kmgtpebs"; char *val; double v; uint64_t v_tmp, adjustment; av->percent = PERCENT_NONE; val = av->value; switch (*val) { case '+': av->sign = SIGN_PLUS; val++; break; case '-': av->sign = SIGN_MINUS; val++; break; default: av->sign = SIGN_NONE; } if (!isdigit(*val)) return 0; v = strtod(val, &ptr); if (*ptr == '.') { /* * Maybe user has non-C locale with different decimal point ? * Lets be toleran and retry with standard C locales */ if (setlocale(LC_ALL, "C")) { v = strtod(val, &ptr); setlocale(LC_ALL, ""); } } if (ptr == val) return 0; if (percent && *ptr == '%') { if (!_get_percent_arg(av, ++ptr)) return_0; if ((uint64_t) v >= UINT32_MAX) { log_error("Percentage is too big (>=%d%%).", UINT32_MAX); return 0; } } else if (*ptr) { for (i = strlen(suffixes) - 1; i >= 0; i--) if (suffixes[i] == tolower((int) *ptr)) break; if (i < 0) { return 0; } else if (i == 7) { /* v is already in sectors */ ; } else if (i == 6) { /* bytes */ v_tmp = (uint64_t) v; adjustment = v_tmp % 512; if (adjustment) { v_tmp += (512 - adjustment); log_error("Size is not a multiple of 512. " "Try using %"PRIu64" or %"PRIu64".", v_tmp - 512, v_tmp); return 0; } v /= 512; } else { /* all other units: kmgtpe */ while (i-- > 0) v *= 1024; v *= 2; } } else v *= factor; if ((uint64_t) v >= (UINT64_MAX >> SECTOR_SHIFT)) { log_error("Size is too big (>=16EiB)."); return 0; } av->i_value = (int32_t) v; av->ui_value = (uint32_t) v; av->i64_value = (int64_t) v; av->ui64_value = (uint64_t) v; return 1; } int size_kb_arg(struct cmd_context *cmd, struct arg_values *av) { return _size_arg(cmd, av, 2, 0); } int size_mb_arg(struct cmd_context *cmd, struct arg_values *av) { return _size_arg(cmd, av, 2048, 0); } int size_mb_arg_with_percent(struct cmd_context *cmd, struct arg_values *av) { return _size_arg(cmd, av, 2048, 1); } int int_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { char *ptr; if (!_get_int_arg(av, &ptr) || (*ptr) || (av->sign == SIGN_MINUS)) return 0; return 1; } int int_arg_with_sign(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { char *ptr; if (!_get_int_arg(av, &ptr) || (*ptr)) return 0; return 1; } int int_arg_with_sign_and_percent(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { char *ptr; if (!_get_int_arg(av, &ptr)) return 0; if (!*ptr) return 1; if (*ptr++ != '%') return 0; if (!_get_percent_arg(av, ptr)) return_0; if (av->ui64_value >= UINT32_MAX) { log_error("Percentage is too big (>=%d%%).", UINT32_MAX); return 0; } return 1; } int string_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av __attribute__((unused))) { return 1; } int tag_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { char *pos = av->value; if (*pos == '@') pos++; if (!validate_tag(pos)) return 0; av->value = pos; return 1; } int permission_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { av->sign = SIGN_NONE; if ((!strcmp(av->value, "rw")) || (!strcmp(av->value, "wr"))) av->ui_value = LVM_READ | LVM_WRITE; else if (!strcmp(av->value, "r")) av->ui_value = LVM_READ; else return 0; return 1; } int alloc_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { alloc_policy_t alloc; av->sign = SIGN_NONE; alloc = get_alloc_from_string(av->value); if (alloc == ALLOC_INVALID) return 0; av->ui_value = (uint32_t) alloc; return 1; } int locktype_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { lock_type_t lock_type; av->sign = SIGN_NONE; lock_type = get_lock_type_from_string(av->value); if (lock_type == LOCK_TYPE_INVALID) return 0; return 1; } int segtype_arg(struct cmd_context *cmd, struct arg_values *av) { struct segment_type *segtype; const char *str = (!strcmp(av->value, SEG_TYPE_NAME_LINEAR)) ? SEG_TYPE_NAME_STRIPED : av->value; if (!(segtype = get_segtype_from_string(cmd, str))) return_0; return (!segtype_is_unknown(segtype)) ? 1 : 0; } /* * Positive integer, zero or "auto". */ int readahead_arg(struct cmd_context *cmd __attribute__((unused)), struct arg_values *av) { if (!strcasecmp(av->value, "auto")) { av->ui_value = DM_READ_AHEAD_AUTO; return 1; } if (!strcasecmp(av->value, "none")) { av->ui_value = DM_READ_AHEAD_NONE; return 1; } if (!_size_arg(cmd, av, 1, 0)) return 0; if (av->sign == SIGN_MINUS) return 0; return 1; } /* * Non-zero, positive integer, "all", or "unmanaged" */ int metadatacopies_arg(struct cmd_context *cmd, struct arg_values *av) { if (!strncmp(cmd->command->name, "vg", 2)) { if (!strcasecmp(av->value, "all")) { av->ui_value = VGMETADATACOPIES_ALL; return 1; } if (!strcasecmp(av->value, "unmanaged")) { av->ui_value = VGMETADATACOPIES_UNMANAGED; return 1; } } return int_arg(cmd, av); } static void __alloc(int size) { if (!(_cmdline.commands = dm_realloc(_cmdline.commands, sizeof(*_cmdline.commands) * size))) { log_fatal("Couldn't allocate memory."); exit(ECMD_FAILED); } _cmdline.commands_size = size; } static void _alloc_command(void) { if (!_cmdline.commands_size) __alloc(32); if (_cmdline.commands_size <= _cmdline.num_commands) __alloc(2 * _cmdline.commands_size); } static void _create_new_command(const char *name, command_fn command, unsigned flags, const char *desc, const char *usagestr, int nargs, int *args) { struct command *nc; _alloc_command(); nc = _cmdline.commands + _cmdline.num_commands++; nc->name = name; nc->desc = desc; nc->usage = usagestr; nc->fn = command; nc->flags = flags; nc->num_args = nargs; nc->valid_args = args; } static void _register_command(const char *name, command_fn fn, const char *desc, unsigned flags, const char *usagestr, ...) { int nargs = 0, i; int *args; va_list ap; /* count how many arguments we have */ va_start(ap, usagestr); while (va_arg(ap, int) >= 0) nargs++; va_end(ap); /* allocate space for them */ if (!(args = dm_malloc(sizeof(*args) * nargs))) { log_fatal("Out of memory."); exit(ECMD_FAILED); } /* fill them in */ va_start(ap, usagestr); for (i = 0; i < nargs; i++) args[i] = va_arg(ap, int); va_end(ap); /* enter the command in the register */ _create_new_command(name, fn, flags, desc, usagestr, nargs, args); } void lvm_register_commands(void) { #define xx(a, b, c, d...) _register_command(# a, a, b, c, ## d, \ driverloaded_ARG, \ debug_ARG, help_ARG, help2_ARG, \ version_ARG, verbose_ARG, \ yes_ARG, \ quiet_ARG, config_ARG, \ commandprofile_ARG, \ profile_ARG, -1); #include "commands.h" #undef xx } static struct command *_find_command(const char *name) { int i; const char *base; base = last_path_component(name); for (i = 0; i < _cmdline.num_commands; i++) { if (!strcmp(base, _cmdline.commands[i].name)) break; } if (i >= _cmdline.num_commands) return 0; return _cmdline.commands + i; } static void _short_usage(const char *name) { log_error("Run `%s --help' for more information.", name); } static int _usage(const char *name) { struct command *com = _find_command(name); if (!com) { log_print("%s: no such command.", name); return 0; } log_print("%s: %s\n\n%s", com->name, com->desc, com->usage); return 1; } /* * Sets up the short and long argument. If there * is no short argument then the index of the * argument in the the_args array is set as the * long opt value. Yuck. Of course this means we * can't have more than 'a' long arguments. */ static void _add_getopt_arg(int arg, char **ptr, struct option **o) { struct arg_props *a = _cmdline.arg_props + arg; if (a->short_arg) { *(*ptr)++ = a->short_arg; if (a->fn) *(*ptr)++ = ':'; } #ifdef HAVE_GETOPTLONG if (*(a->long_arg + 2)) { (*o)->name = a->long_arg + 2; (*o)->has_arg = a->fn ? 1 : 0; (*o)->flag = NULL; if (a->short_arg) (*o)->val = a->short_arg; else (*o)->val = arg + 128; (*o)++; } #endif } static int _find_arg(struct command *com, int opt) { struct arg_props *a; int i, arg; for (i = 0; i < com->num_args; i++) { arg = com->valid_args[i]; a = _cmdline.arg_props + arg; /* * opt should equal either the * short arg, or the index into * the_args. */ if ((a->short_arg && (opt == a->short_arg)) || (!a->short_arg && (opt == (arg + 128)))) return arg; } return -1; } static int _process_command_line(struct cmd_context *cmd, int *argc, char ***argv) { int i, opt, arg; char str[((ARG_COUNT + 1) * 2) + 1], *ptr = str; struct option opts[ARG_COUNT + 1], *o = opts; struct arg_props *a; struct arg_values *av; struct arg_value_group_list *current_group = NULL; if (!(cmd->arg_values = dm_pool_zalloc(cmd->mem, sizeof(*cmd->arg_values) * ARG_COUNT))) { log_fatal("Unable to allocate memory for command line arguments."); return 0; } /* fill in the short and long opts */ for (i = 0; i < cmd->command->num_args; i++) _add_getopt_arg(cmd->command->valid_args[i], &ptr, &o); *ptr = '\0'; memset(o, 0, sizeof(*o)); /* initialise getopt_long & scan for command line switches */ optarg = 0; optind = OPTIND_INIT; while ((opt = GETOPTLONG_FN(*argc, *argv, str, opts, NULL)) >= 0) { if (opt == '?') return 0; if ((arg = _find_arg(cmd->command, opt)) < 0) { log_fatal("Unrecognised option."); return 0; } a = _cmdline.arg_props + arg; av = &cmd->arg_values[arg]; if (a->flags & ARG_GROUPABLE) { /* * Start a new group of arguments: * - the first time, * - or if a non-countable argument is repeated, * - or if argument has higher priority than current group. */ if (!current_group || (current_group->arg_values[arg].count && !(a->flags & ARG_COUNTABLE)) || (current_group->prio < a->prio)) { /* FIXME Reduce size including only groupable args */ if (!(current_group = dm_pool_zalloc(cmd->mem, sizeof(struct arg_value_group_list) + sizeof(*cmd->arg_values) * ARG_COUNT))) { log_fatal("Unable to allocate memory for command line arguments."); return 0; } current_group->prio = a->prio; dm_list_add(&cmd->arg_value_groups, ¤t_group->list); } /* Maintain total argument count as well as count within each group */ av->count++; av = ¤t_group->arg_values[arg]; } if (av->count && !(a->flags & ARG_COUNTABLE)) { log_error("Option%s%c%s%s may not be repeated.", a->short_arg ? " -" : "", a->short_arg ? : ' ', (a->short_arg && a->long_arg) ? "/" : "", a->long_arg ? : ""); return 0; } if (a->fn) { if (!optarg) { log_error("Option requires argument."); return 0; } av->value = optarg; if (!a->fn(cmd, av)) { log_error("Invalid argument for %s: %s", a->long_arg, optarg); return 0; } } av->count++; } *argc -= optind; *argv += optind; return 1; } static void _copy_arg_values(struct arg_values *av, int oldarg, int newarg) { const struct arg_values *old = av + oldarg; struct arg_values *new = av + newarg; new->count = old->count; new->value = old->value; new->i_value = old->i_value; new->ui_value = old->ui_value; new->i64_value = old->i64_value; new->ui64_value = old->ui64_value; new->sign = old->sign; } static int _merge_synonym(struct cmd_context *cmd, int oldarg, int newarg) { struct arg_values *av; struct arg_value_group_list *current_group; if (arg_is_set(cmd, oldarg) && arg_is_set(cmd, newarg)) { log_error("%s and %s are synonyms. Please only supply one.", _cmdline.arg_props[oldarg].long_arg, _cmdline.arg_props[newarg].long_arg); return 0; } /* Not groupable? */ if (!(_cmdline.arg_props[oldarg].flags & ARG_GROUPABLE)) { if (arg_is_set(cmd, oldarg)) _copy_arg_values(cmd->arg_values, oldarg, newarg); return 1; } if (arg_is_set(cmd, oldarg)) cmd->arg_values[newarg].count = cmd->arg_values[oldarg].count; /* Groupable */ dm_list_iterate_items(current_group, &cmd->arg_value_groups) { av = current_group->arg_values; if (!grouped_arg_count(av, oldarg)) continue; _copy_arg_values(av, oldarg, newarg); } return 1; } int systemid(struct cmd_context *cmd __attribute__((unused)), int argc __attribute__((unused)), char **argv __attribute__((unused))) { log_print("system ID: %s", cmd->system_id ? : ""); return ECMD_PROCESSED; } int version(struct cmd_context *cmd __attribute__((unused)), int argc __attribute__((unused)), char **argv __attribute__((unused))) { char vsn[80]; log_print("LVM version: %s", LVM_VERSION); if (library_version(vsn, sizeof(vsn))) log_print("Library version: %s", vsn); if (driver_version(vsn, sizeof(vsn))) log_print("Driver version: %s", vsn); return ECMD_PROCESSED; } static int _get_settings(struct cmd_context *cmd) { const char *activation_mode; cmd->current_settings = cmd->default_settings; cmd->current_settings.verbose = arg_count(cmd, verbose_ARG); cmd->current_settings.test = arg_is_set(cmd, test_ARG); if (arg_is_set(cmd, debug_ARG)) cmd->current_settings.debug = _LOG_FATAL + (arg_count(cmd, debug_ARG) - 1); if (arg_is_set(cmd, quiet_ARG)) { cmd->current_settings.debug = 0; cmd->current_settings.verbose = 0; cmd->current_settings.silent = (arg_count(cmd, quiet_ARG) > 1) ? 1 : 0; } if (arg_is_set(cmd, driverloaded_ARG)) { cmd->current_settings.activation = arg_int_value(cmd, driverloaded_ARG, cmd->default_settings.activation); } cmd->current_settings.archive = arg_int_value(cmd, autobackup_ARG, cmd->current_settings.archive); cmd->current_settings.backup = arg_int_value(cmd, autobackup_ARG, cmd->current_settings.backup); cmd->current_settings.cache_vgmetadata = cmd->command->flags & CACHE_VGMETADATA ? 1 : 0; if (arg_is_set(cmd, readonly_ARG)) { cmd->current_settings.activation = 0; cmd->current_settings.archive = 0; cmd->current_settings.backup = 0; } if (cmd->command->flags & LOCKD_VG_SH) cmd->lockd_vg_default_sh = 1; cmd->partial_activation = 0; cmd->degraded_activation = 0; activation_mode = find_config_tree_str(cmd, activation_mode_CFG, NULL); if (!activation_mode) activation_mode = DEFAULT_ACTIVATION_MODE; if (arg_is_set(cmd, activationmode_ARG)) { activation_mode = arg_str_value(cmd, activationmode_ARG, activation_mode); /* complain only if the two arguments conflict */ if (arg_is_set(cmd, partial_ARG) && strcmp(activation_mode, "partial")) { log_error("--partial and --activationmode are mutually" " exclusive arguments"); return EINVALID_CMD_LINE; } } else if (arg_is_set(cmd, partial_ARG)) activation_mode = "partial"; if (!strcmp(activation_mode, "partial")) { cmd->partial_activation = 1; log_warn("PARTIAL MODE. Incomplete logical volumes will be processed."); } else if (!strcmp(activation_mode, "degraded")) cmd->degraded_activation = 1; else if (strcmp(activation_mode, "complete")) { log_error("Invalid activation mode given."); return EINVALID_CMD_LINE; } if (arg_is_set(cmd, ignorelockingfailure_ARG) || arg_is_set(cmd, sysinit_ARG)) init_ignorelockingfailure(1); else init_ignorelockingfailure(0); cmd->ignore_clustered_vgs = arg_is_set(cmd, ignoreskippedcluster_ARG); cmd->include_foreign_vgs = arg_is_set(cmd, foreign_ARG) ? 1 : 0; cmd->include_shared_vgs = arg_is_set(cmd, shared_ARG) ? 1 : 0; cmd->include_historical_lvs = arg_is_set(cmd, history_ARG) ? 1 : 0; cmd->record_historical_lvs = find_config_tree_bool(cmd, metadata_record_lvs_history_CFG, NULL) ? (arg_is_set(cmd, nohistory_ARG) ? 0 : 1) : 0; /* * This is set to zero by process_each which wants to print errors * itself rather than having them printed in vg_read. */ cmd->vg_read_print_access_error = 1; if (arg_is_set(cmd, nosuffix_ARG)) cmd->current_settings.suffix = 0; if (arg_is_set(cmd, units_ARG)) if (!(cmd->current_settings.unit_factor = dm_units_to_factor(arg_str_value(cmd, units_ARG, ""), &cmd->current_settings.unit_type, 1, NULL))) { log_error("Invalid units specification"); return EINVALID_CMD_LINE; } if (arg_is_set(cmd, binary_ARG)) cmd->report_binary_values_as_numeric = 1; if (arg_is_set(cmd, trustcache_ARG)) { if (arg_is_set(cmd, all_ARG)) { log_error("--trustcache is incompatible with --all"); return EINVALID_CMD_LINE; } init_trust_cache(1); log_warn("WARNING: Cache file of PVs will be trusted. " "New devices holding PVs may get ignored."); } else init_trust_cache(0); if (arg_is_set(cmd, noudevsync_ARG)) cmd->current_settings.udev_sync = 0; /* Handle synonyms */ if (!_merge_synonym(cmd, resizable_ARG, resizeable_ARG) || !_merge_synonym(cmd, allocation_ARG, allocatable_ARG) || !_merge_synonym(cmd, allocation_ARG, resizeable_ARG) || !_merge_synonym(cmd, virtualoriginsize_ARG, virtualsize_ARG) || !_merge_synonym(cmd, available_ARG, activate_ARG) || !_merge_synonym(cmd, raidrebuild_ARG, rebuild_ARG) || !_merge_synonym(cmd, raidsyncaction_ARG, syncaction_ARG) || !_merge_synonym(cmd, raidwritemostly_ARG, writemostly_ARG) || !_merge_synonym(cmd, raidminrecoveryrate_ARG, minrecoveryrate_ARG) || !_merge_synonym(cmd, raidmaxrecoveryrate_ARG, maxrecoveryrate_ARG) || !_merge_synonym(cmd, raidwritebehind_ARG, writebehind_ARG)) return EINVALID_CMD_LINE; if ((!strncmp(cmd->command->name, "pv", 2) && !_merge_synonym(cmd, metadatacopies_ARG, pvmetadatacopies_ARG)) || (!strncmp(cmd->command->name, "vg", 2) && !_merge_synonym(cmd, metadatacopies_ARG, vgmetadatacopies_ARG))) return EINVALID_CMD_LINE; /* Zero indicates success */ return 0; } static int _process_common_commands(struct cmd_context *cmd) { if (arg_is_set(cmd, help_ARG) || arg_is_set(cmd, help2_ARG)) { _usage(cmd->command->name); return ECMD_PROCESSED; } if (arg_is_set(cmd, version_ARG)) { return version(cmd, 0, (char **) NULL); } /* Zero indicates it's OK to continue processing this command */ return 0; } static void _display_help(void) { int i; log_error("Available lvm commands:"); log_error("Use 'lvm help ' for more information"); log_error(" "); for (i = 0; i < _cmdline.num_commands; i++) { struct command *com = _cmdline.commands + i; log_error("%-16.16s%s", com->name, com->desc); } } int help(struct cmd_context *cmd __attribute__((unused)), int argc, char **argv) { int ret = ECMD_PROCESSED; if (!argc) _display_help(); else { int i; for (i = 0; i < argc; i++) if (!_usage(argv[i])) ret = EINVALID_CMD_LINE; } return ret; } static void _apply_settings(struct cmd_context *cmd) { init_debug(cmd->current_settings.debug); init_debug_classes_logged(cmd->default_settings.debug_classes); init_verbose(cmd->current_settings.verbose + VERBOSE_BASE_LEVEL); init_silent(cmd->current_settings.silent); init_test(cmd->current_settings.test); init_full_scan_done(0); init_mirror_in_sync(0); init_dmeventd_monitor(DEFAULT_DMEVENTD_MONITOR); init_msg_prefix(cmd->default_settings.msg_prefix); init_cmd_name(cmd->default_settings.cmd_name); archive_enable(cmd, cmd->current_settings.archive); backup_enable(cmd, cmd->current_settings.backup); set_activation(cmd->current_settings.activation, cmd->metadata_read_only); cmd->fmt = get_format_by_name(cmd, arg_str_value(cmd, metadatatype_ARG, cmd->current_settings.fmt_name)); cmd->handles_missing_pvs = 0; } static const char *_copy_command_line(struct cmd_context *cmd, int argc, char **argv) { int i, space; /* * Build up the complete command line, used as a * description for backups. */ if (!dm_pool_begin_object(cmd->mem, 128)) goto_bad; for (i = 0; i < argc; i++) { space = strchr(argv[i], ' ') ? 1 : 0; if (space && !dm_pool_grow_object(cmd->mem, "'", 1)) goto_bad; if (!dm_pool_grow_object(cmd->mem, argv[i], strlen(argv[i]))) goto_bad; if (space && !dm_pool_grow_object(cmd->mem, "'", 1)) goto_bad; if (i < (argc - 1)) if (!dm_pool_grow_object(cmd->mem, " ", 1)) goto_bad; } /* * Terminate. */ if (!dm_pool_grow_object(cmd->mem, "\0", 1)) goto_bad; return dm_pool_end_object(cmd->mem); bad: log_error("Couldn't copy command line."); dm_pool_abandon_object(cmd->mem); return NULL; } static int _prepare_profiles(struct cmd_context *cmd) { static const char COMMAND_PROFILE_ENV_VAR_NAME[] = "LVM_COMMAND_PROFILE"; static const char _cmd_profile_arg_preferred_over_env_var_msg[] = "Giving " "preference to command profile specified on command " "line over the one specified via environment variable."; static const char _failed_to_add_profile_msg[] = "Failed to add %s %s."; static const char _failed_to_apply_profile_msg[] = "Failed to apply %s %s."; static const char _command_profile_source_name[] = "command profile"; static const char _metadata_profile_source_name[] = "metadata profile"; static const char _setting_global_profile_msg[] = "Setting global %s \"%s\"."; const char *env_cmd_profile_name = NULL; const char *name; struct profile *profile; config_source_t source; const char *source_name; /* Check whether default global command profile is set via env. var. */ if ((env_cmd_profile_name = getenv(COMMAND_PROFILE_ENV_VAR_NAME))) { if (!*env_cmd_profile_name) env_cmd_profile_name = NULL; else log_debug("Command profile '%s' requested via " "environment variable.", env_cmd_profile_name); } if (!arg_is_set(cmd, profile_ARG) && !arg_is_set(cmd, commandprofile_ARG) && !arg_is_set(cmd, metadataprofile_ARG) && !env_cmd_profile_name) /* nothing to do */ return 1; if (arg_is_set(cmd, profile_ARG)) { /* * If --profile is used with dumpconfig, it's used * to dump the profile without the profile being applied. */ if (!strcmp(cmd->command->name, "dumpconfig") || !strcmp(cmd->command->name, "lvmconfig") || !strcmp(cmd->command->name, "config")) return 1; /* * If --profile is used with lvcreate/lvchange/vgchange, * it's recognized as shortcut to --metadataprofile. * The --commandprofile is assumed otherwise. */ if (!strcmp(cmd->command->name, "lvcreate") || !strcmp(cmd->command->name, "vgcreate") || !strcmp(cmd->command->name, "lvchange") || !strcmp(cmd->command->name, "vgchange")) { if (arg_is_set(cmd, metadataprofile_ARG)) { log_error("Only one of --profile or " " --metadataprofile allowed."); return 0; } source = CONFIG_PROFILE_METADATA; source_name = _metadata_profile_source_name; } else { if (arg_is_set(cmd, commandprofile_ARG)) { log_error("Only one of --profile or " "--commandprofile allowed."); return 0; } /* * Prefer command profile specified on command * line over the profile specified via * COMMAND_PROFILE_ENV_VAR_NAME env. var. */ if (env_cmd_profile_name) { log_debug(_cmd_profile_arg_preferred_over_env_var_msg); env_cmd_profile_name = NULL; } source = CONFIG_PROFILE_COMMAND; source_name = _command_profile_source_name; } name = arg_str_value(cmd, profile_ARG, NULL); if (!(profile = add_profile(cmd, name, source))) { log_error(_failed_to_add_profile_msg, source_name, name); return 0; } if (source == CONFIG_PROFILE_COMMAND) { log_debug(_setting_global_profile_msg, _command_profile_source_name, profile->name); cmd->profile_params->global_command_profile = profile; } else if (source == CONFIG_PROFILE_METADATA) { log_debug(_setting_global_profile_msg, _metadata_profile_source_name, profile->name); /* This profile will override any VG/LV-based profile if present */ cmd->profile_params->global_metadata_profile = profile; } remove_config_tree_by_source(cmd, source); if (!override_config_tree_from_profile(cmd, profile)) { log_error(_failed_to_apply_profile_msg, source_name, name); return 0; } } if (arg_is_set(cmd, commandprofile_ARG) || env_cmd_profile_name) { if (arg_is_set(cmd, commandprofile_ARG)) { /* * Prefer command profile specified on command * line over the profile specified via * COMMAND_PROFILE_ENV_VAR_NAME env. var. */ if (env_cmd_profile_name) log_debug(_cmd_profile_arg_preferred_over_env_var_msg); name = arg_str_value(cmd, commandprofile_ARG, NULL); } else name = env_cmd_profile_name; source_name = _command_profile_source_name; if (!(profile = add_profile(cmd, name, CONFIG_PROFILE_COMMAND))) { log_error(_failed_to_add_profile_msg, source_name, name); return 0; } remove_config_tree_by_source(cmd, CONFIG_PROFILE_COMMAND); if (!override_config_tree_from_profile(cmd, profile)) { log_error(_failed_to_apply_profile_msg, source_name, name); return 0; } log_debug(_setting_global_profile_msg, _command_profile_source_name, profile->name); cmd->profile_params->global_command_profile = profile; if (!cmd->arg_values) cmd->profile_params->shell_profile = profile; } if (arg_is_set(cmd, metadataprofile_ARG)) { name = arg_str_value(cmd, metadataprofile_ARG, NULL); source_name = _metadata_profile_source_name; if (!(profile = add_profile(cmd, name, CONFIG_PROFILE_METADATA))) { log_error(_failed_to_add_profile_msg, source_name, name); return 0; } remove_config_tree_by_source(cmd, CONFIG_PROFILE_METADATA); if (!override_config_tree_from_profile(cmd, profile)) { log_error(_failed_to_apply_profile_msg, source_name, name); return 0; } log_debug(_setting_global_profile_msg, _metadata_profile_source_name, profile->name); cmd->profile_params->global_metadata_profile = profile; } if (!process_profilable_config(cmd)) return_0; return 1; } static int _init_lvmlockd(struct cmd_context *cmd) { const char *lvmlockd_socket; int use_lvmlockd = find_config_tree_bool(cmd, global_use_lvmlockd_CFG, NULL); if (use_lvmlockd && arg_is_set(cmd, nolocking_ARG)) { /* --nolocking is only allowed with vgs/lvs/pvs commands */ cmd->lockd_gl_disable = 1; cmd->lockd_vg_disable = 1; cmd->lockd_lv_disable = 1; return 1; } if (use_lvmlockd && locking_is_clustered()) { log_error("ERROR: configuration setting use_lvmlockd cannot be used with clustered locking_type 3."); return 0; } lvmlockd_disconnect(); /* start over when tool context is refreshed */ lvmlockd_socket = getenv("LVM_LVMLOCKD_SOCKET"); if (!lvmlockd_socket) lvmlockd_socket = DEFAULT_RUN_DIR "/lvmlockd.socket"; lvmlockd_set_socket(lvmlockd_socket); lvmlockd_set_use(use_lvmlockd); if (use_lvmlockd) { lvmlockd_init(cmd); lvmlockd_connect(); } return 1; } static int _cmd_no_meta_proc(struct cmd_context *cmd) { return cmd->command->flags & NO_METADATA_PROCESSING; } int lvm_run_command(struct cmd_context *cmd, int argc, char **argv) { struct dm_config_tree *config_string_cft, *config_profile_command_cft, *config_profile_metadata_cft; const char *reason = NULL; int ret = 0; int locking_type; int monitoring; char *arg_new, *arg; int i; int skip_hyphens; int refresh_done = 0; init_error_message_produced(0); /* each command should start out with sigint flag cleared */ sigint_clear(); /* eliminate '-' from all options starting with -- */ for (i = 1; i < argc; i++) { arg = argv[i]; if (*arg++ != '-' || *arg++ != '-') continue; /* If we reach "--" then stop. */ if (!*arg) break; arg_new = arg; skip_hyphens = 1; while (*arg) { /* If we encounter '=', stop any further hyphen removal. */ if (*arg == '=') skip_hyphens = 0; /* Do we need to keep the next character? */ if (*arg != '-' || !skip_hyphens) { if (arg_new != arg) *arg_new = *arg; ++arg_new; } arg++; } /* Terminate a shortened arg */ if (arg_new != arg) *arg_new = '\0'; } if (!(cmd->cmd_line = _copy_command_line(cmd, argc, argv))) return_ECMD_FAILED; log_debug("Parsing: %s", cmd->cmd_line); if (!(cmd->command = _find_command(argv[0]))) return ENO_SUCH_CMD; if (!_process_command_line(cmd, &argc, &argv)) { log_error("Error during parsing of command line."); return EINVALID_CMD_LINE; } set_cmd_name(cmd->command->name); if (arg_is_set(cmd, backgroundfork_ARG)) { if (!become_daemon(cmd, 1)) { /* parent - quit immediately */ ret = ECMD_PROCESSED; goto out; } } if (arg_is_set(cmd, config_ARG)) if (!override_config_tree_from_string(cmd, arg_str_value(cmd, config_ARG, ""))) { ret = EINVALID_CMD_LINE; goto_out; } if (arg_is_set(cmd, config_ARG) || !cmd->initialized.config || config_files_changed(cmd)) { /* Reinitialise various settings inc. logging, filters */ if (!refresh_toolcontext(cmd)) { if ((config_string_cft = remove_config_tree_by_source(cmd, CONFIG_STRING))) dm_config_destroy(config_string_cft); log_error("Updated config file invalid. Aborting."); return ECMD_FAILED; } refresh_done = 1; } if (!_prepare_profiles(cmd)) return_ECMD_FAILED; if (!cmd->initialized.connections && !_cmd_no_meta_proc(cmd) && !init_connections(cmd)) return_ECMD_FAILED; /* Note: Load persistent cache only if we haven't refreshed toolcontext! * If toolcontext has been refreshed, it means config has changed * and we can't rely on persistent cache anymore. */ if (!cmd->initialized.filters && !_cmd_no_meta_proc(cmd) && !init_filters(cmd, !refresh_done)) return_ECMD_FAILED; if (arg_is_set(cmd, readonly_ARG)) cmd->metadata_read_only = 1; if ((ret = _get_settings(cmd))) goto_out; _apply_settings(cmd); if (cmd->degraded_activation) log_debug("DEGRADED MODE. Incomplete RAID LVs will be processed."); if (!get_activation_monitoring_mode(cmd, &monitoring)) goto_out; init_dmeventd_monitor(monitoring); log_debug("Processing: %s", cmd->cmd_line); log_debug("Command pid: %d", getpid()); log_debug("system ID: %s", cmd->system_id ? : ""); #ifdef O_DIRECT_SUPPORT log_debug("O_DIRECT will be used"); #endif if ((ret = _process_common_commands(cmd))) { if (ret != ECMD_PROCESSED) stack; goto out; } if (!strcmp(cmd->fmt->name, FMT_LVM1_NAME) && lvmetad_used()) { log_warn("WARNING: Disabling lvmetad cache which does not support obsolete metadata."); lvmetad_set_disabled(cmd, "LVM1"); log_warn("WARNING: Not using lvmetad because lvm1 format is used."); lvmetad_make_unused(cmd); } if (cmd->metadata_read_only && !(cmd->command->flags & PERMITTED_READ_ONLY)) { log_error("%s: Command not permitted while global/metadata_read_only " "is set.", cmd->cmd_line); goto out; } if (_cmd_no_meta_proc(cmd)) locking_type = 0; else if (arg_is_set(cmd, readonly_ARG)) { if (find_config_tree_bool(cmd, global_use_lvmlockd_CFG, NULL)) { /* * FIXME: we could use locking_type 5 here if that didn't * cause CLUSTERED to be set, which conflicts with using lvmlockd. */ locking_type = 1; cmd->lockd_gl_disable = 1; cmd->lockd_vg_disable = 1; cmd->lockd_lv_disable = 1; } else { locking_type = 5; } if (lvmetad_used()) { lvmetad_make_unused(cmd); log_verbose("Not using lvmetad because read-only is set."); } } else if (arg_is_set(cmd, nolocking_ARG)) locking_type = 0; else locking_type = -1; if (!init_locking(locking_type, cmd, _cmd_no_meta_proc(cmd) || arg_is_set(cmd, sysinit_ARG))) { ret = ECMD_FAILED; goto_out; } if (!_cmd_no_meta_proc(cmd) && !_init_lvmlockd(cmd)) { ret = ECMD_FAILED; goto_out; } /* * pvscan/vgscan/lvscan/vgimport want their own control over rescanning * to populate lvmetad and have similar code of their own. * Other commands use this general policy for using lvmetad. * * The lvmetad cache may need to be repopulated before we use it because: * - We are reading foreign VGs which others hosts may have changed * which our lvmetad would not have seen. * - lvmetad may have just been started and no command has been run * to populate it yet (e.g. no pvscan --cache was run). * - Another local command may have run with a different global filter * which changed the content of lvmetad from what we want (recognized * by different token values.) * * lvmetad may have been previously disabled (or disabled during the * rescan done here) because duplicate devices or lvm1 metadata were seen. * In this case, disable the *use* of lvmetad by this command, reverting to * disk scanning. */ if (lvmetad_used() && !(cmd->command->flags & NO_LVMETAD_AUTOSCAN)) { if (cmd->include_foreign_vgs || !lvmetad_token_matches(cmd)) { if (lvmetad_used() && !lvmetad_pvscan_all_devs(cmd, cmd->include_foreign_vgs ? 1 : 0)) { log_warn("WARNING: Not using lvmetad because cache update failed."); lvmetad_make_unused(cmd); } } if (lvmetad_used() && lvmetad_is_disabled(cmd, &reason)) { log_warn("WARNING: Not using lvmetad because %s.", reason); lvmetad_make_unused(cmd); if (strstr(reason, "duplicate")) { log_warn("WARNING: Use multipath or vgimportclone to resolve duplicate PVs?"); if (!find_config_tree_bool(cmd, devices_multipath_component_detection_CFG, NULL)) log_warn("WARNING: Set multipath_component_detection=1 to hide multipath duplicates."); log_warn("WARNING: After duplicates are resolved, run \"pvscan --cache\" to enable lvmetad."); } } } /* * FIXME Break up into multiple functions. */ ret = cmd->command->fn(cmd, argc, argv); lvmlockd_disconnect(); fin_locking(); if (!_cmd_no_meta_proc(cmd) && find_config_tree_bool(cmd, global_notify_dbus_CFG, NULL)) lvmnotify_send(cmd); out: if (test_mode()) { log_verbose("Test mode: Wiping internal cache"); lvmcache_destroy(cmd, 1, 0); } if ((config_string_cft = remove_config_tree_by_source(cmd, CONFIG_STRING))) dm_config_destroy(config_string_cft); config_profile_command_cft = remove_config_tree_by_source(cmd, CONFIG_PROFILE_COMMAND); config_profile_metadata_cft = remove_config_tree_by_source(cmd, CONFIG_PROFILE_METADATA); cmd->profile_params->global_metadata_profile = NULL; if (config_string_cft) { /* Move this? */ if (!refresh_toolcontext(cmd)) stack; } else if (config_profile_command_cft || config_profile_metadata_cft) { if (!process_profilable_config(cmd)) stack; } if (ret == EINVALID_CMD_LINE && !cmd->is_interactive) _short_usage(cmd->command->name); log_debug("Completed: %s", cmd->cmd_line); cmd->current_settings = cmd->default_settings; _apply_settings(cmd); /* * free off any memory the command used. */ dm_list_init(&cmd->arg_value_groups); dm_pool_empty(cmd->mem); reset_lvm_errno(1); reset_log_duplicated(); return ret; } int lvm_return_code(int ret) { unlink_log_file(ret); return (ret == ECMD_PROCESSED ? 0 : ret); } int lvm_split(char *str, int *argc, char **argv, int max) { char *b = str, *e; char quote = 0; *argc = 0; while (*b) { while (*b && isspace(*b)) b++; if ((!*b) || (*b == '#')) break; if (*b == '\'' || *b == '"') { quote = *b; b++; } e = b; while (*e && (quote ? *e != quote : !isspace(*e))) e++; argv[(*argc)++] = b; if (!*e) break; *e++ = '\0'; quote = 0; b = e; if (*argc == max) break; } return *argc; } /* Make sure we have always valid filedescriptors 0,1,2 */ static int _check_standard_fds(void) { int err = is_valid_fd(STDERR_FILENO); if (!is_valid_fd(STDIN_FILENO) && !(stdin = fopen(_PATH_DEVNULL, "r"))) { if (err) perror("stdin stream open"); else printf("stdin stream open: %s\n", strerror(errno)); return 0; } if (!is_valid_fd(STDOUT_FILENO) && !(stdout = fopen(_PATH_DEVNULL, "w"))) { if (err) perror("stdout stream open"); /* else no stdout */ return 0; } if (!is_valid_fd(STDERR_FILENO) && !(stderr = fopen(_PATH_DEVNULL, "w"))) { printf("stderr stream open: %s\n", strerror(errno)); return 0; } return 1; } #define LVM_OUT_FD_ENV_VAR_NAME "LVM_OUT_FD" #define LVM_ERR_FD_ENV_VAR_NAME "LVM_ERR_FD" #define LVM_REPORT_FD_ENV_VAR_NAME "LVM_REPORT_FD" static int _do_get_custom_fd(const char *env_var_name, int *fd) { const char *str; char *endptr; long int tmp_fd; *fd = -1; if (!(str = getenv(env_var_name))) return 1; errno = 0; tmp_fd = strtol(str, &endptr, 10); if (errno || *endptr || (tmp_fd < 0) || (tmp_fd > INT_MAX)) { log_error("%s: invalid file descriptor.", env_var_name); return 0; } *fd = tmp_fd; return 1; } static int _get_custom_fds(struct custom_fds *custom_fds) { return _do_get_custom_fd(LVM_OUT_FD_ENV_VAR_NAME, &custom_fds->out) && _do_get_custom_fd(LVM_ERR_FD_ENV_VAR_NAME, &custom_fds->err) && _do_get_custom_fd(LVM_REPORT_FD_ENV_VAR_NAME, &custom_fds->report); } static const char *_get_cmdline(pid_t pid) { static char _proc_cmdline[32]; char buf[256]; int fd, n = 0; snprintf(buf, sizeof(buf), DEFAULT_PROC_DIR "/%u/cmdline", pid); /* FIXME Use generic read code. */ if ((fd = open(buf, O_RDONLY)) >= 0) { if ((n = read(fd, _proc_cmdline, sizeof(_proc_cmdline) - 1)) < 0) { log_sys_error("read", buf); n = 0; } if (close(fd)) log_sys_error("close", buf); } _proc_cmdline[n] = '\0'; return _proc_cmdline; } static const char *_get_filename(int fd) { static char filename[PATH_MAX]; char buf[32]; /* Assumes short DEFAULT_PROC_DIR */ int size; snprintf(buf, sizeof(buf), DEFAULT_PROC_DIR "/self/fd/%u", fd); if ((size = readlink(buf, filename, sizeof(filename) - 1)) == -1) filename[0] = '\0'; else filename[size] = '\0'; return filename; } static void _close_descriptor(int fd, unsigned suppress_warnings, const char *command, pid_t ppid, const char *parent_cmdline) { int r; const char *filename; /* Ignore bad file descriptors */ if (!is_valid_fd(fd)) return; if (!suppress_warnings) filename = _get_filename(fd); r = close(fd); if (suppress_warnings) return; if (!r) fprintf(stderr, "File descriptor %d (%s) leaked on " "%s invocation.", fd, filename, command); else if (errno == EBADF) return; else fprintf(stderr, "Close failed on stray file descriptor " "%d (%s): %s", fd, filename, strerror(errno)); fprintf(stderr, " Parent PID %" PRIpid_t ": %s\n", ppid, parent_cmdline); } static int _close_stray_fds(const char *command, struct custom_fds *custom_fds) { #ifndef VALGRIND_POOL struct rlimit rlim; int fd; unsigned suppress_warnings = 0; pid_t ppid = getppid(); const char *parent_cmdline = _get_cmdline(ppid); static const char _fd_dir[] = DEFAULT_PROC_DIR "/self/fd"; struct dirent *dirent; DIR *d; #ifdef HAVE_VALGRIND if (RUNNING_ON_VALGRIND) { log_debug("Skipping close of descriptors within valgrind execution."); return 1; } #endif if (getenv("LVM_SUPPRESS_FD_WARNINGS")) suppress_warnings = 1; if (!(d = opendir(_fd_dir))) { if (errno != ENOENT) { log_sys_error("opendir", _fd_dir); return 0; /* broken system */ } /* Path does not exist, use the old way */ if (getrlimit(RLIMIT_NOFILE, &rlim) < 0) { log_sys_error("getrlimit", "RLIMIT_NOFILE"); return 1; } for (fd = 3; fd < (int)rlim.rlim_cur; fd++) { if ((fd != custom_fds->out) && (fd != custom_fds->err) && (fd != custom_fds->report)) { _close_descriptor(fd, suppress_warnings, command, ppid, parent_cmdline); } } return 1; } while ((dirent = readdir(d))) { fd = atoi(dirent->d_name); if ((fd > 2) && (fd != dirfd(d)) && (fd != custom_fds->out) && (fd != custom_fds->err) && (fd != custom_fds->report)) { _close_descriptor(fd, suppress_warnings, command, ppid, parent_cmdline); } } if (closedir(d)) log_sys_error("closedir", _fd_dir); #endif return 1; } struct cmd_context *init_lvm(unsigned set_connections, unsigned set_filters) { struct cmd_context *cmd; if (!udev_init_library_context()) stack; /* * It's not necessary to use name mangling for LVM: * - the character set used for LV names is subset of udev character set * - when we check other devices (e.g. device_is_usable fn), we use major:minor, not dm names */ dm_set_name_mangling_mode(DM_STRING_MANGLING_NONE); if (!(cmd = create_toolcontext(0, NULL, 1, 0, set_connections, set_filters))) { udev_fin_library_context(); return_NULL; } _cmdline.arg_props = &_arg_props[0]; if (stored_errno()) { destroy_toolcontext(cmd); udev_fin_library_context(); return_NULL; } return cmd; } static void _fin_commands(void) { int i; for (i = 0; i < _cmdline.num_commands; i++) dm_free(_cmdline.commands[i].valid_args); dm_free(_cmdline.commands); _cmdline.commands = NULL; _cmdline.num_commands = 0; _cmdline.commands_size = 0; } void lvm_fin(struct cmd_context *cmd) { _fin_commands(); destroy_toolcontext(cmd); udev_fin_library_context(); } static int _run_script(struct cmd_context *cmd, int argc, char **argv) { FILE *script; char buffer[CMD_LEN]; int ret = 0; int magic_number = 0; char *script_file = argv[0]; if ((script = fopen(script_file, "r")) == NULL) return ENO_SUCH_CMD; while (fgets(buffer, sizeof(buffer), script) != NULL) { if (!magic_number) { if (buffer[0] == '#' && buffer[1] == '!') magic_number = 1; else { ret = ENO_SUCH_CMD; break; } } if ((strlen(buffer) == sizeof(buffer) - 1) && (buffer[sizeof(buffer) - 1] - 2 != '\n')) { buffer[50] = '\0'; log_error("Line too long (max 255) beginning: %s", buffer); ret = EINVALID_CMD_LINE; break; } if (lvm_split(buffer, &argc, argv, MAX_ARGS) == MAX_ARGS) { buffer[50] = '\0'; log_error("Too many arguments: %s", buffer); ret = EINVALID_CMD_LINE; break; } if (!argc) continue; if (!strcmp(argv[0], "quit") || !strcmp(argv[0], "exit")) break; ret = lvm_run_command(cmd, argc, argv); if (ret != ECMD_PROCESSED) { if (!error_message_produced()) { log_debug(INTERNAL_ERROR "Failed command did not use log_error"); log_error("Command failed with status code %d.", ret); } break; } } if (fclose(script)) log_sys_error("fclose", script_file); return ret; } /* * Determine whether we should fall back and exec the equivalent LVM1 tool */ static int _lvm1_fallback(struct cmd_context *cmd) { char vsn[80]; int dm_present; if (!find_config_tree_bool(cmd, global_fallback_to_lvm1_CFG, NULL) || strncmp(cmd->kernel_vsn, "2.4.", 4)) return 0; log_suppress(1); dm_present = driver_version(vsn, sizeof(vsn)); log_suppress(0); if (dm_present || !lvm1_present(cmd)) return 0; return 1; } static void _exec_lvm1_command(char **argv) { char path[PATH_MAX]; if (dm_snprintf(path, sizeof(path), "%s.lvm1", argv[0]) < 0) { log_error("Failed to create LVM1 tool pathname"); return; } execvp(path, argv); log_sys_error("execvp", path); } static void _nonroot_warning(void) { if (getuid() || geteuid()) log_warn("WARNING: Running as a non-root user. Functionality may be unavailable."); } int lvm2_main(int argc, char **argv) { const char *base; int ret, alias = 0; struct custom_fds custom_fds; struct cmd_context *cmd; if (!argv) return -1; base = last_path_component(argv[0]); if (strcmp(base, "lvm") && strcmp(base, "lvm.static") && strcmp(base, "initrd-lvm")) alias = 1; if (!_check_standard_fds()) return -1; if (!_get_custom_fds(&custom_fds)) return -1; if (!_close_stray_fds(base, &custom_fds)) return -1; if (!init_custom_log_streams(&custom_fds)) return -1; if (is_static() && strcmp(base, "lvm.static") && path_exists(LVM_PATH) && !getenv("LVM_DID_EXEC")) { if (setenv("LVM_DID_EXEC", base, 1)) log_sys_error("setenv", "LVM_DID_EXEC"); if (execvp(LVM_PATH, argv) == -1) log_sys_error("execvp", LVM_PATH); if (unsetenv("LVM_DID_EXEC")) log_sys_error("unsetenv", "LVM_DID_EXEC"); } /* "version" command is simple enough so it doesn't need any complex init */ if (!alias && argc > 1 && !strcmp(argv[1], "version")) return lvm_return_code(version(NULL, argc, argv)); if (!(cmd = init_lvm(0, 0))) return -1; cmd->argv = argv; lvm_register_commands(); if (_lvm1_fallback(cmd)) { /* Attempt to run equivalent LVM1 tool instead */ if (!alias) { argv++; argc--; } if (!argc) { log_error("Falling back to LVM1 tools, but no " "command specified."); ret = ECMD_FAILED; goto out; } _exec_lvm1_command(argv); ret = ECMD_FAILED; goto_out; } #ifdef READLINE_SUPPORT if (!alias && argc == 1) { _nonroot_warning(); if (!_prepare_profiles(cmd)) { ret = ECMD_FAILED; goto out; } ret = lvm_shell(cmd, &_cmdline); goto out; } #endif if (!alias) { if (argc < 2) { log_fatal("Please supply an LVM command."); _display_help(); ret = EINVALID_CMD_LINE; goto out; } argc--; argv++; } _nonroot_warning(); ret = lvm_run_command(cmd, argc, argv); if ((ret == ENO_SUCH_CMD) && (!alias)) ret = _run_script(cmd, argc, argv); if (ret == ENO_SUCH_CMD) log_error("No such command. Try 'help'."); if ((ret != ECMD_PROCESSED) && !error_message_produced()) { log_debug(INTERNAL_ERROR "Failed command did not use log_error"); log_error("Command failed with status code %d.", ret); } out: lvm_fin(cmd); return lvm_return_code(ret); }