X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fsysctl.c;h=9f7acfce8b09f76d67247c0d82caa2c35baed5aa;hp=5ffb9695c6439349d6c26fa6c4a00828d41f26e8;hb=f68c5a70762da0cbb6228d05a87254a7c6c46ba6;hpb=55af3897854263eddc0818d5cc4614ccbdae7f32 diff --git a/src/sysctl.c b/src/sysctl.c index 5ffb9695c..9f7acfce8 100644 --- a/src/sysctl.c +++ b/src/sysctl.c @@ -30,10 +30,11 @@ #include "log.h" #include "strv.h" #include "util.h" +#include "strv.h" #define PROC_SYS_PREFIX "/proc/sys/" -static const char *arg_prefix = NULL; +static char **arg_prefixes = NULL; static int apply_sysctl(const char *property, const char *value) { char *p, *n; @@ -54,10 +55,21 @@ static int apply_sysctl(const char *property, const char *value) { if (*n == '.') *n = '/'; - if (arg_prefix && !path_startswith(p, arg_prefix)) { - log_debug("Skipping %s", p); - free(p); - return 0; + if (!strv_isempty(arg_prefixes)) { + char **i; + bool good = false; + + STRV_FOREACH(i, arg_prefixes) + if (path_startswith(p, *i)) { + good = true; + break; + } + + if (!good) { + log_debug("Skipping %s", p); + free(p); + return 0; + } } k = write_one_line_file(p, value); @@ -170,12 +182,20 @@ static int parse_argv(int argc, char *argv[]) { case ARG_PREFIX: { char *p; + char **l; for (p = optarg; *p; p++) if (*p == '.') *p = '/'; - arg_prefix = optarg; + l = strv_append(arg_prefixes, optarg); + if (!l) { + log_error("Out of memory"); + return -ENOMEM; + } + + strv_free(arg_prefixes); + arg_prefixes = l; break; } @@ -238,5 +258,7 @@ int main(int argc, char *argv[]) { strv_free(files); } finish: + strv_free(arg_prefixes); + return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; }