ubase/sysctl.c

214 lines
3.2 KiB
C

/* See LICENSE file for copyright and license details. */
#include <fcntl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "text.h"
#include "util.h"
static void
replacestr(char *s, int a, int b)
{
for (; *s; s++)
if (*s == a)
*s = b;
}
static int
getsysctl(char *variable, char **value)
{
char path[PATH_MAX];
char *p;
char *buf, *tmp, c;
int fd;
ssize_t n;
size_t sz, i;
replacestr(variable, '.', '/');
strlcpy(path, "/proc/sys/", sizeof(path));
if (strlcat(path, variable, sizeof(path)) >= sizeof(path)) {
replacestr(variable, '/', '.');
return -1;
}
replacestr(variable, '/', '.');
fd = open(path, O_RDONLY);
if (fd < 0)
return -1;
i = 0;
sz = 1;
buf = NULL;
while (1) {
n = read(fd, &c, 1);
if (n < 0) {
close(fd);
free(buf);
return -1;
}
if (n == 0)
break;
if (i == sz - 1) {
sz *= 2;
tmp = realloc(buf, sz);
if (!tmp) {
close(fd);
free(buf);
return -1;
}
buf = tmp;
}
buf[i++] = c;
}
buf[i] = '\0';
p = strrchr(buf, '\n');
if (p)
*p = '\0';
*value = buf;
close(fd);
return 0;
}
static int
setsysctl(char *variable, char *value)
{
char path[PATH_MAX];
int fd;
ssize_t n;
replacestr(variable, '.', '/');
strlcpy(path, "/proc/sys/", sizeof(path));
if (strlcat(path, variable, sizeof(path)) >= sizeof(path)) {
replacestr(variable, '/', '.');
return -1;
}
replacestr(variable, '/', '.');
fd = open(path, O_WRONLY);
if (fd < 0)
return -1;
n = write(fd, value, strlen(value));
if ((size_t)n != strlen(value)) {
close(fd);
return -1;
}
close(fd);
return 0;
}
static int
parsepair(char *pair)
{
char *p;
char *variable;
char *value;
for (p = pair; *p; p++) {
if (p[0] == '.' && p[1] == '.') {
weprintf("malformed input: %s\n", pair);
return -1;
}
}
p = strchr(pair, '=');
if (p) {
if (p[1] == '\0') {
weprintf("malformed input: %s\n", pair);
return -1;
}
*p = '\0';
value = &p[1];
} else {
value = NULL;
}
variable = pair;
if (value) {
if (setsysctl(variable, value) < 0) {
weprintf("failed to set sysctl for %s\n", variable);
return -1;
}
} else {
if (getsysctl(variable, &value) < 0) {
weprintf("failed to get sysctl for %s\n", variable);
return -1;
}
printf("%s = %s\n", variable, value);
free(value);
}
return 0;
}
static void
usage(void)
{
eprintf("usage: %s [-p file] variable[=value]...\n", argv0);
}
int
main(int argc, char *argv[])
{
FILE *fp;
char *buf = NULL, *p;
char *file = NULL;
size_t size = 0;
int i;
int r = EXIT_SUCCESS;
ARGBEGIN {
case 'p':
file = EARGF(usage());
break;
default:
usage();
} ARGEND;
if (!file && argc < 1)
usage();
if (!file) {
for (i = 0; i < argc; i++)
if (parsepair(argv[i]) < 0)
r = EXIT_FAILURE;
} else {
fp = fopen(file, "r");
if (!fp)
eprintf("fopen %s:", file);
while (agetline(&buf, &size, fp) != -1) {
p = buf;
for (p = buf; *p == ' ' || *p == '\t'; p++)
;
if (*p == '#' || *p == '\n')
continue;
for (p = buf; *p; p++) {
if (*p == '\n') {
*p = '\0';
break;
}
}
p = buf;
if (parsepair(p) < 0)
r = EXIT_FAILURE;
}
if (ferror(fp))
eprintf("%s: read error:", file);
free(buf);
fclose(fp);
}
return r;
}