sysctl: add -e: supress warnings about invalid key (Jeremy Kerr <jk@ozlabs.org>)

code shrink:

   text    data     bss     dec     hex filename
 775565     929    9100  785594   bfcba busybox_old
 775296     929    9100  785325   bfbad busybox_unstripped
This commit is contained in:
Denis Vlasenko 2007-11-29 06:40:28 +00:00
parent 745cd17926
commit 64309f8669
2 changed files with 74 additions and 99 deletions

View File

@ -3505,16 +3505,17 @@ USE_FEATURE_RUN_PARTS_FANCY("\n -l Prints names of all matching files even when
"Configure kernel parameters at runtime" \ "Configure kernel parameters at runtime" \
"\n\nOptions:\n" \ "\n\nOptions:\n" \
" -n Disable printing of key names\n" \ " -n Disable printing of key names\n" \
" -e Don't warn about unknown keys\n" \
" -w Change sysctl setting\n" \ " -w Change sysctl setting\n" \
" -p FILE Load sysctl settings from FILE (default /etc/sysctl.conf)\n" \ " -p FILE Load sysctl settings from FILE (default /etc/sysctl.conf)\n" \
" -a Display all values\n" \ " -a Display all values\n" \
" -A Display all values in table form" " -A Display all values in table form"
#define sysctl_example_usage \ #define sysctl_example_usage \
"sysctl [-n] variable...\n" \ "sysctl [-n] [-e] variable...\n" \
"sysctl [-n] -w variable=value...\n" \ "sysctl [-n] [-e] -w variable=value...\n" \
"sysctl [-n] -a\n" \ "sysctl [-n] [-e] -a\n" \
"sysctl [-n] -p file (default /etc/sysctl.conf)\n" \ "sysctl [-n] [-e] -p file (default /etc/sysctl.conf)\n" \
"sysctl [-n] -A\n" "sysctl [-n] [-e] -A\n"
#define syslogd_trivial_usage \ #define syslogd_trivial_usage \
"[OPTION]..." "[OPTION]..."

View File

@ -16,24 +16,16 @@
#include "libbb.h" #include "libbb.h"
/* static int sysctl_read_setting(const char *setting);
* Function Prototypes static int sysctl_write_setting(const char *setting);
*/ static int sysctl_display_all(const char *path);
static int sysctl_read_setting(const char *setting, int output); static int sysctl_preload_file_and_exit(const char *filename);
static int sysctl_write_setting(const char *setting, int output);
static int sysctl_preload_file(const char *filename, int output);
static int sysctl_display_all(const char *path, int output, int show_table);
/*
* Globals...
*/
static const char ETC_SYSCTL_CONF[] ALIGN1 = "/etc/sysctl.conf"; static const char ETC_SYSCTL_CONF[] ALIGN1 = "/etc/sysctl.conf";
static const char PROC_SYS[] ALIGN1 = "/proc/sys/"; static const char PROC_SYS[] ALIGN1 = "/proc/sys/";
enum { strlen_PROC_SYS = sizeof(PROC_SYS) - 1 }; enum { strlen_PROC_SYS = sizeof(PROC_SYS) - 1 };
/* error messages */ /* error messages */
static const char ERR_UNKNOWN_PARAMETER[] ALIGN1 =
"error: unknown parameter '%s'";
static const char ERR_MALFORMED_SETTING[] ALIGN1 = static const char ERR_MALFORMED_SETTING[] ALIGN1 =
"error: malformed setting '%s'"; "error: malformed setting '%s'";
static const char ERR_NO_EQUALS[] ALIGN1 = static const char ERR_NO_EQUALS[] ALIGN1 =
@ -46,8 +38,6 @@ static const char ERR_UNKNOWN_READING[] ALIGN1 =
"error reading key '%s'"; "error reading key '%s'";
static const char ERR_PERMISSION_DENIED[] ALIGN1 = static const char ERR_PERMISSION_DENIED[] ALIGN1 =
"error: permission denied on key '%s'"; "error: permission denied on key '%s'";
static const char ERR_PRELOAD_FILE[] ALIGN1 =
"error: cannot open preload file '%s'";
static const char WARN_BAD_LINE[] ALIGN1 = static const char WARN_BAD_LINE[] ALIGN1 =
"warning: %s(%d): invalid syntax, continuing"; "warning: %s(%d): invalid syntax, continuing";
@ -57,66 +47,50 @@ static void dwrite_str(int fd, const char *buf)
write(fd, buf, strlen(buf)); write(fd, buf, strlen(buf));
} }
/* enum {
* sysctl_main()... FLAG_SHOW_KEYS = 1 << 0,
*/ FLAG_SHOW_KEY_ERRORS = 1 << 1,
FLAG_TABLE_FORMAT = 1 << 2, /* not implemented */
FLAG_SHOW_ALL = 1 << 3,
FLAG_PRELOAD_FILE = 1 << 4,
FLAG_WRITE = 1 << 5,
};
int sysctl_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int sysctl_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int sysctl_main(int argc, char **argv) int sysctl_main(int argc, char **argv)
{ {
int retval = 0; int retval;
int output = 1; int opt;
int write_mode = 0;
int switches_allowed = 1;
if (argc < 2) opt = getopt32(argv, "+neAapw"); /* '+' - stop on first non-option */
bb_show_usage(); argv += optind;
opt ^= (FLAG_SHOW_KEYS | FLAG_SHOW_KEY_ERRORS);
option_mask32 ^= (FLAG_SHOW_KEYS | FLAG_SHOW_KEY_ERRORS);
argv++; if (opt & (FLAG_TABLE_FORMAT | FLAG_SHOW_ALL))
return sysctl_display_all(PROC_SYS);
if (opt & FLAG_PRELOAD_FILE)
return sysctl_preload_file_and_exit(*argv ? *argv : ETC_SYSCTL_CONF);
for (; *argv /*&& **argv*/; argv++) { retval = 0;
if (switches_allowed && **argv == '-') { /* we have a switch */ while (*argv) {
switch ((*argv)[1]) { if (opt & FLAG_WRITE)
case 'n': retval |= sysctl_write_setting(*argv);
output = 0; else
break; retval |= sysctl_read_setting(*argv);
case 'w': argv++;
write_mode = 1;
switches_allowed = 0;
break;
case 'p':
argv++;
return sysctl_preload_file(((*argv /*&& **argv*/) ? *argv : ETC_SYSCTL_CONF),
output);
case 'a':
case 'A':
return sysctl_display_all(PROC_SYS, output,
((*argv)[1] == 'A'));
default:
bb_error_msg(ERR_UNKNOWN_PARAMETER, *argv);
/* fall through */
//case 'h':
//case '?':
bb_show_usage();
}
} else {
switches_allowed = 0;
if (write_mode)
retval |= sysctl_write_setting(*argv, output);
else
sysctl_read_setting(*argv, output);
}
} }
return retval; return retval;
} /* end sysctl_main() */ } /* end sysctl_main() */
/* /*
* sysctl_preload_file
* preload the sysctl's from a conf file * preload the sysctl's from a conf file
* - we parse the file and then reform it (strip out whitespace) * - we parse the file and then reform it (strip out whitespace)
*/ */
#define PRELOAD_BUF 256 #define PRELOAD_BUF 256
static int sysctl_preload_file(const char *filename, int output) static int sysctl_preload_file_and_exit(const char *filename)
{ {
int lineno; int lineno;
char oneline[PRELOAD_BUF]; char oneline[PRELOAD_BUF];
@ -124,10 +98,7 @@ static int sysctl_preload_file(const char *filename, int output)
char *name, *value; char *name, *value;
FILE *fp; FILE *fp;
fp = fopen(filename, "r"); fp = xfopen(filename, "r");
if (fp == NULL) {
bb_error_msg_and_die(ERR_PRELOAD_FILE, filename);
}
lineno = 0; lineno = 0;
while (fgets(oneline, sizeof(oneline) - 1, fp)) { while (fgets(oneline, sizeof(oneline) - 1, fp)) {
@ -163,35 +134,36 @@ static int sysctl_preload_file(const char *filename, int output)
/* safe because sizeof(oneline) == sizeof(buffer) */ /* safe because sizeof(oneline) == sizeof(buffer) */
sprintf(buffer, "%s=%s", name, value); sprintf(buffer, "%s=%s", name, value);
sysctl_write_setting(buffer, output); sysctl_write_setting(buffer);
} }
fclose(fp); if (ENABLE_FEATURE_CLEAN_UP)
fclose(fp);
return 0; return 0;
} /* end sysctl_preload_file() */ } /* end sysctl_preload_file_and_exit() */
/* /*
* Write a single sysctl setting * Write a single sysctl setting
*/ */
static int sysctl_write_setting(const char *setting, int output) static int sysctl_write_setting(const char *setting)
{ {
int retval = 0; int retval;
const char *name; const char *name;
const char *value; const char *value;
const char *equals; const char *equals;
char *tmpname, *outname, *cptr; char *tmpname, *outname, *cptr;
int fd = -1; int fd;
name = setting; name = setting;
equals = strchr(setting, '='); equals = strchr(setting, '=');
if (!equals) { if (!equals) {
bb_error_msg(ERR_NO_EQUALS, setting); bb_error_msg(ERR_NO_EQUALS, setting);
return -1; return EXIT_FAILURE;
} }
value = equals + 1; /* point to the value in name=value */ value = equals + 1; /* point to the value in name=value */
if (name == equals || !*value) { if (name == equals || !*value) {
bb_error_msg(ERR_MALFORMED_SETTING, setting); bb_error_msg(ERR_MALFORMED_SETTING, setting);
return -2; return EXIT_FAILURE;
} }
tmpname = xasprintf("%s%.*s", PROC_SYS, (int)(equals - name), name); tmpname = xasprintf("%s%.*s", PROC_SYS, (int)(equals - name), name);
@ -207,7 +179,8 @@ static int sysctl_write_setting(const char *setting, int output)
if (fd < 0) { if (fd < 0) {
switch (errno) { switch (errno) {
case ENOENT: case ENOENT:
bb_error_msg(ERR_INVALID_KEY, outname); if (option_mask32 & FLAG_SHOW_KEY_ERRORS)
bb_error_msg(ERR_INVALID_KEY, outname);
break; break;
case EACCES: case EACCES:
bb_perror_msg(ERR_PERMISSION_DENIED, outname); bb_perror_msg(ERR_PERMISSION_DENIED, outname);
@ -216,16 +189,15 @@ static int sysctl_write_setting(const char *setting, int output)
bb_perror_msg(ERR_UNKNOWN_WRITING, outname); bb_perror_msg(ERR_UNKNOWN_WRITING, outname);
break; break;
} }
retval = -1; retval = EXIT_FAILURE;
} else { } else {
dwrite_str(fd, value); dwrite_str(fd, value);
close(fd); close(fd);
if (output) { if (option_mask32 & FLAG_SHOW_KEYS) {
dwrite_str(STDOUT_FILENO, outname); printf("%s = ", outname);
dwrite_str(STDOUT_FILENO, " = ");
} }
dwrite_str(STDOUT_FILENO, value); puts(value);
dwrite_str(STDOUT_FILENO, "\n"); retval = EXIT_SUCCESS;
} }
free(tmpname); free(tmpname);
@ -236,18 +208,19 @@ static int sysctl_write_setting(const char *setting, int output)
/* /*
* Read a sysctl setting * Read a sysctl setting
*/ */
static int sysctl_read_setting(const char *setting, int output) static int sysctl_read_setting(const char *name)
{ {
int retval = 0; int retval;
char *tmpname, *outname, *cptr; char *tmpname, *outname, *cptr;
char inbuf[1025]; char inbuf[1025];
const char *name;
FILE *fp; FILE *fp;
if (!*setting) if (!*name) {
bb_error_msg(ERR_INVALID_KEY, setting); if (option_mask32 & FLAG_SHOW_KEY_ERRORS)
bb_error_msg(ERR_INVALID_KEY, name);
return -1;
}
name = setting;
tmpname = concat_path_file(PROC_SYS, name); tmpname = concat_path_file(PROC_SYS, name);
outname = xstrdup(tmpname + strlen_PROC_SYS); outname = xstrdup(tmpname + strlen_PROC_SYS);
@ -260,7 +233,8 @@ static int sysctl_read_setting(const char *setting, int output)
if (fp == NULL) { if (fp == NULL) {
switch (errno) { switch (errno) {
case ENOENT: case ENOENT:
bb_error_msg(ERR_INVALID_KEY, outname); if (option_mask32 & FLAG_SHOW_KEY_ERRORS)
bb_error_msg(ERR_INVALID_KEY, outname);
break; break;
case EACCES: case EACCES:
bb_error_msg(ERR_PERMISSION_DENIED, outname); bb_error_msg(ERR_PERMISSION_DENIED, outname);
@ -269,16 +243,16 @@ static int sysctl_read_setting(const char *setting, int output)
bb_perror_msg(ERR_UNKNOWN_READING, outname); bb_perror_msg(ERR_UNKNOWN_READING, outname);
break; break;
} }
retval = -1; retval = EXIT_FAILURE;
} else { } else {
while (fgets(inbuf, sizeof(inbuf) - 1, fp)) { while (fgets(inbuf, sizeof(inbuf) - 1, fp)) {
if (output) { if (option_mask32 & FLAG_SHOW_KEYS) {
dwrite_str(STDOUT_FILENO, outname); printf("%s = ", outname);
dwrite_str(STDOUT_FILENO, " = ");
} }
dwrite_str(STDOUT_FILENO, inbuf); fputs(inbuf, stdout);
} }
fclose(fp); fclose(fp);
retval = EXIT_SUCCESS;
} }
free(tmpname); free(tmpname);
@ -289,7 +263,7 @@ static int sysctl_read_setting(const char *setting, int output)
/* /*
* Display all the sysctl settings * Display all the sysctl settings
*/ */
static int sysctl_display_all(const char *path, int output, int show_table) static int sysctl_display_all(const char *path)
{ {
int retval = 0; int retval = 0;
DIR *dp; DIR *dp;
@ -299,18 +273,18 @@ static int sysctl_display_all(const char *path, int output, int show_table)
dp = opendir(path); dp = opendir(path);
if (!dp) { if (!dp) {
return -1; return EXIT_FAILURE;
} }
while ((de = readdir(dp)) != NULL) { while ((de = readdir(dp)) != NULL) {
tmpdir = concat_subpath_file(path, de->d_name); tmpdir = concat_subpath_file(path, de->d_name);
if (tmpdir == NULL) if (tmpdir == NULL)
continue; continue; /* . or .. */
if (stat(tmpdir, &ts) != 0) { if (stat(tmpdir, &ts) != 0) {
bb_perror_msg(tmpdir); bb_perror_msg(tmpdir);
} else if (S_ISDIR(ts.st_mode)) { } else if (S_ISDIR(ts.st_mode)) {
sysctl_display_all(tmpdir, output, show_table); retval |= sysctl_display_all(tmpdir);
} else { } else {
retval |= sysctl_read_setting(tmpdir + strlen_PROC_SYS, output); retval |= sysctl_read_setting(tmpdir + strlen_PROC_SYS);
} }
free(tmpdir); free(tmpdir);
} /* end while */ } /* end while */