Bump ubox-2016-07-19-aead2c0cbffdda9b46d74a998a4c6aeef423b21a

This commit is contained in:
jbnadal
2016-12-06 17:42:06 +01:00
parent 578707a4e6
commit 9f6ea62b0a
12 changed files with 3624 additions and 0 deletions

View File

@@ -0,0 +1,62 @@
cmake_minimum_required(VERSION 2.6)
PROJECT(ubox C)
ADD_DEFINITIONS(-Os -ggdb -Wall -Werror --std=gnu99 -Wmissing-declarations)
SET(CMAKE_SHARED_LIBRARY_LINK_C_FLAGS "")
IF(APPLE)
INCLUDE_DIRECTORIES(/opt/local/include)
LINK_DIRECTORIES(/opt/local/lib)
ENDIF()
find_library(json NAMES json-c json)
IF(DEBUG)
ADD_DEFINITIONS(-DDEBUG -g3)
ENDIF()
ADD_EXECUTABLE(getrandom getrandom.c)
INSTALL(TARGETS getrandom
RUNTIME DESTINATION bin
)
ADD_EXECUTABLE(kmodloader kmodloader.c)
TARGET_LINK_LIBRARIES(kmodloader ubox)
INSTALL(TARGETS kmodloader
RUNTIME DESTINATION sbin
)
ADD_EXECUTABLE(lsbloader lsbloader.c)
TARGET_LINK_LIBRARIES(lsbloader ubox ubus)
INSTALL(TARGETS lsbloader
RUNTIME DESTINATION sbin
)
ADD_LIBRARY(validate SHARED validate/validate.c)
INSTALL(TARGETS validate
LIBRARY DESTINATION lib
)
FIND_PATH(uci_include_dir uci.h)
INCLUDE_DIRECTORIES(${uci_include_dir})
ADD_EXECUTABLE(validate_data validate/cli.c)
TARGET_LINK_LIBRARIES(validate_data ubox uci validate)
INSTALL(TARGETS validate_data
RUNTIME DESTINATION sbin
)
ADD_EXECUTABLE(logd log/logd.c log/syslog.c)
TARGET_LINK_LIBRARIES(logd ubox ubus)
INSTALL(TARGETS logd
RUNTIME DESTINATION sbin
)
ADD_EXECUTABLE(logread log/logread.c)
TARGET_LINK_LIBRARIES(logread ubox ubus ${json} blobmsg_json)
INSTALL(TARGETS logread
RUNTIME DESTINATION sbin
)

View File

@@ -0,0 +1,33 @@
cmake_minimum_required (VERSION 3.0)
project (ubox)
ADD_DEFINITIONS(-Os -ggdb -Wall -Werror --std=gnu99 -Wmissing-declarations)
set (CMAKE_MODULE_PATH "${MODULE_PATH}")
set(DISABLE_TARGET_OPTIMIZATION ON)
include (aw)
# Logd
add_executable (logd
$ENV{AWOXCVS}/AwoxAudio/Products/External/ubox/log/logd.c
$ENV{AWOXCVS}/AwoxAudio/Products/External/ubox/log/syslog.c)
target_link_libraries (logd ubox ubus)
install (TARGETS logd RUNTIME DESTINATION sbin)
# Logread
add_executable (logread
$ENV{AWOXCVS}/AwoxAudio/Products/External/ubox/log/logread.c)
target_link_libraries (logread ubox ubus json-c blobmsg_json)
install (TARGETS logread RUNTIME DESTINATION sbin)
# kmodloader
add_executable (kmodloader
$ENV{AWOXCVS}/AwoxAudio/Products/External/ubox/kmodloader.c)
target_link_libraries (kmodloader ubox)
install (TARGETS kmodloader RUNTIME DESTINATION ../sbin)

58
src/3P/ubox/getrandom.c Normal file
View File

@@ -0,0 +1,58 @@
/*
* Copyright (C) 2016 Etienne Champetier <champetier.etienne@gmail.com>
*
* 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.
*/
#define _GNU_SOURCE
#include <errno.h>
#include <linux/random.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/syscall.h>
#include <unistd.h>
#define ERROR_EXIT(fmt, ...) do { \
fprintf(stderr, fmt, ## __VA_ARGS__); \
return EXIT_FAILURE; \
} while (0)
static int usage(char *name)
{
fprintf(stderr, "Usage: %s <nb>\n", name);
fprintf(stderr, " => return <nb> bytes from getrandom()\n");
return EXIT_FAILURE;
}
int main(int argc, char *argv[])
{
if (argc != 2)
return usage(argv[0]);
if (isatty(STDOUT_FILENO))
ERROR_EXIT("Not outputting random to a tty\n");
int nbtot = atoi(argv[1]);
if (nbtot < 1)
ERROR_EXIT("Invalid <nb> param (must be > 0)\n");
char buf[256];
int len = sizeof(buf);
while (nbtot > 0) {
if (nbtot <= sizeof(buf))
len = nbtot;
if (syscall(SYS_getrandom, buf, len, 0) == -1)
ERROR_EXIT("getrandom() failed: %s\n", strerror(errno));
if (write(STDOUT_FILENO, buf, len) != len)
ERROR_EXIT("write() failed: %s\n", strerror(errno));
nbtot -= sizeof(buf);
}
}

878
src/3P/ubox/kmodloader.c Normal file
View File

@@ -0,0 +1,878 @@
/*
* Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
* Copyright (C) 2013 John Crispin <blogic@openwrt.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation
*
* 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.
*/
//#define _GNU_SOURCE
#include <sys/syscall.h>
#include <sys/mman.h>
#include <sys/utsname.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/syscall.h>
#include <sys/types.h>
#include <values.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <syslog.h>
#include <libgen.h>
#include <glob.h>
#include <elf.h>
#include <libubox/avl.h>
#include <libubox/avl-cmp.h>
#include <libubox/utils.h>
#include <libubox/ulog.h>
#define DEF_MOD_PATH "/modules/%s/"
enum {
SCANNED,
PROBE,
LOADED,
};
struct module {
struct avl_node avl;
char *name;
char *depends;
char *opts;
int size;
int usage;
int state;
int error;
};
static struct avl_tree modules;
static char **module_folders = NULL;
static int init_module_folders(void)
{
int n = 0;
struct stat st;
struct utsname ver;
char *s, *e, *p, path[256], ldpath[256];
e = ldpath;
s = getenv("LD_LIBRARY_PATH");
if (s)
e += snprintf(ldpath, sizeof(ldpath), "%s:", s);
e += snprintf(e, sizeof(ldpath) - (e - ldpath), "/lib");
uname(&ver);
for (s = p = ldpath; p <= e; p++) {
if (*p != ':' && *p != '\0')
continue;
*p = 0;
snprintf(path, sizeof(path), "%s" DEF_MOD_PATH, s, ver.release);
if (!stat(path, &st) && S_ISDIR(st.st_mode)) {
module_folders = realloc(module_folders, sizeof(p) * (n + 2));
if (!module_folders)
return -1;
module_folders[n++] = strdup(path);
}
s = p + 1;
}
if (!module_folders)
return -1;
module_folders[n] = NULL;
return 0;
}
static struct module *find_module(const char *name)
{
struct module *m;
return avl_find_element(&modules, name, m, avl);
}
static void free_modules(void)
{
struct module *m, *tmp;
avl_remove_all_elements(&modules, m, avl, tmp)
free(m);
}
static char* get_module_path(char *name)
{
char **p;
static char path[256];
struct stat s;
if (!stat(name, &s) && S_ISREG(s.st_mode))
return name;
for (p = module_folders; *p; p++) {
snprintf(path, sizeof(path), "%s%s.ko", *p, name);
if (!stat(path, &s) && S_ISREG(s.st_mode))
return path;
}
return NULL;
}
static char* get_module_name(char *path)
{
static char name[32];
char *t;
strncpy(name, basename(path), sizeof(name));
t = strstr(name, ".ko");
if (t)
*t = '\0';
return name;
}
static int elf64_find_section(char *map, const char *section, unsigned int *offset, unsigned int *size)
{
const char *secnames;
Elf64_Ehdr *e;
Elf64_Shdr *sh;
int i;
e = (Elf64_Ehdr *) map;
sh = (Elf64_Shdr *) (map + e->e_shoff);
secnames = map + sh[e->e_shstrndx].sh_offset;
for (i = 0; i < e->e_shnum; i++) {
if (!strcmp(section, secnames + sh[i].sh_name)) {
*size = sh[i].sh_size;
*offset = sh[i].sh_offset;
return 0;
}
}
return -1;
}
static int elf32_find_section(char *map, const char *section, unsigned int *offset, unsigned int *size)
{
const char *secnames;
Elf32_Ehdr *e;
Elf32_Shdr *sh;
int i;
e = (Elf32_Ehdr *) map;
sh = (Elf32_Shdr *) (map + e->e_shoff);
secnames = map + sh[e->e_shstrndx].sh_offset;
for (i = 0; i < e->e_shnum; i++) {
if (!strcmp(section, secnames + sh[i].sh_name)) {
*size = sh[i].sh_size;
*offset = sh[i].sh_offset;
return 0;
}
}
return -1;
}
static int elf_find_section(char *map, const char *section, unsigned int *offset, unsigned int *size)
{
int clazz = map[EI_CLASS];
if (clazz == ELFCLASS32)
return elf32_find_section(map, section, offset, size);
else if (clazz == ELFCLASS64)
return elf64_find_section(map, section, offset, size);
ULOG_ERR("unknown elf format %d\n", clazz);
return -1;
}
static struct module *
alloc_module(const char *name, const char *depends, int size)
{
struct module *m;
char *_name, *_dep;
m = calloc_a(sizeof(*m),
&_name, strlen(name) + 1,
&_dep, depends ? strlen(depends) + 2 : 0);
if (!m)
return NULL;
m->avl.key = m->name = strcpy(_name, name);
m->opts = 0;
if (depends) {
m->depends = strcpy(_dep, depends);
while (*_dep) {
if (*_dep == ',')
*_dep = '\0';
_dep++;
}
}
m->size = size;
avl_insert(&modules, &m->avl);
return m;
}
static int scan_loaded_modules(void)
{
size_t buf_len = 0;
char *buf = NULL;
FILE *fp;
fp = fopen("/proc/modules", "r");
if (!fp) {
ULOG_ERR("failed to open /proc/modules\n");
return -1;
}
while (getline(&buf, &buf_len, fp) > 0) {
struct module m;
struct module *n;
m.name = strtok(buf, " ");
m.size = atoi(strtok(NULL, " "));
m.usage = atoi(strtok(NULL, " "));
m.depends = strtok(NULL, " ");
if (!m.name || !m.depends)
continue;
n = alloc_module(m.name, m.depends, m.size);
n->usage = m.usage;
n->state = LOADED;
}
free(buf);
fclose(fp);
return 0;
}
static struct module* get_module_info(const char *module, const char *name)
{
int fd = open(module, O_RDONLY);
unsigned int offset, size;
char *map = MAP_FAILED, *strings, *dep = NULL;
struct module *m = NULL;
struct stat s;
if (fd < 0) {
ULOG_ERR("failed to open %s\n", module);
goto out;
}
if (fstat(fd, &s) == -1) {
ULOG_ERR("failed to stat %s\n", module);
goto out;
}
map = mmap(NULL, s.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
if (map == MAP_FAILED) {
ULOG_ERR("failed to mmap %s\n", module);
goto out;
}
if (elf_find_section(map, ".modinfo", &offset, &size)) {
ULOG_ERR("failed to load the .modinfo section from %s\n", module);
goto out;
}
strings = map + offset;
while (strings && (strings < map + offset + size)) {
char *sep;
int len;
while (!strings[0])
strings++;
sep = strstr(strings, "=");
if (!sep)
break;
len = sep - strings;
sep++;
if (!strncmp(strings, "depends=", len + 1))
dep = sep;
strings = &sep[strlen(sep)];
}
m = alloc_module(name, dep, s.st_size);
if (m)
m->state = SCANNED;
out:
if (map != MAP_FAILED)
munmap(map, s.st_size);
if (fd >= 0)
close(fd);
return m;
}
static int scan_module_folder(const char *dir)
{
int gl_flags = GLOB_NOESCAPE | GLOB_MARK;
struct utsname ver;
char *path;
glob_t gl;
int j;
uname(&ver);
path = alloca(strlen(dir) + sizeof("*.ko") + 1);
sprintf(path, "%s*.ko", dir);
if (glob(path, gl_flags, NULL, &gl) < 0)
return -1;
for (j = 0; j < gl.gl_pathc; j++) {
char *name = get_module_name(gl.gl_pathv[j]);
struct module *m;
if (!name)
continue;
m = find_module(name);
if (!m)
get_module_info(gl.gl_pathv[j], name);
}
globfree(&gl);
return 0;
}
static int scan_module_folders(void)
{
int rv = 0;
char **p;
if (init_module_folders())
return -1;
for (p = module_folders; *p; p++)
rv |= scan_module_folder(*p);
return rv;
}
static int print_modinfo(char *module)
{
int fd = open(module, O_RDONLY);
unsigned int offset, size;
struct stat s;
char *map = MAP_FAILED, *strings;
int rv = -1;
if (fd < 0) {
ULOG_ERR("failed to open %s\n", module);
goto out;
}
if (fstat(fd, &s) == -1) {
ULOG_ERR("failed to stat %s\n", module);
goto out;
}
map = mmap(NULL, s.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
if (map == MAP_FAILED) {
ULOG_ERR("failed to mmap %s\n", module);
goto out;
}
if (elf_find_section(map, ".modinfo", &offset, &size)) {
ULOG_ERR("failed to load the .modinfo section from %s\n", module);
goto out;
}
strings = map + offset;
printf("module:\t\t%s\n", module);
while (strings && (strings < map + offset + size)) {
char *dup = NULL;
char *sep;
while (!strings[0])
strings++;
sep = strstr(strings, "=");
if (!sep)
break;
dup = strndup(strings, sep - strings);
sep++;
if (strncmp(strings, "parm", 4)) {
if (strlen(dup) < 7)
printf("%s:\t\t%s\n", dup, sep);
else
printf("%s:\t%s\n", dup, sep);
}
strings = &sep[strlen(sep)];
if (dup)
free(dup);
}
rv = 0;
out:
if (map != MAP_FAILED)
munmap(map, s.st_size);
if (fd >= 0)
close(fd);
return rv;
}
static int deps_available(struct module *m, int verbose)
{
char *dep;
int err = 0;
if (!m->depends || !strcmp(m->depends, "-") || !strcmp(m->depends, ""))
return 0;
dep = m->depends;
while (*dep) {
m = find_module(dep);
if (verbose && !m)
ULOG_ERR("missing dependency %s\n", dep);
if (verbose && m && (m->state != LOADED))
ULOG_ERR("dependency not loaded %s\n", dep);
if (!m || (m->state != LOADED))
err++;
dep += strlen(dep) + 1;
}
return err;
}
static int insert_module(char *path, const char *options)
{
void *data = 0;
struct stat s;
int fd, ret = -1;
if (stat(path, &s)) {
ULOG_ERR("missing module %s\n", path);
return ret;
}
fd = open(path, O_RDONLY);
if (fd < 0) {
ULOG_ERR("cannot open %s\n", path);
return ret;
}
data = malloc(s.st_size);
if (read(fd, data, s.st_size) == s.st_size)
ret = syscall(__NR_init_module, data, (unsigned long) s.st_size, options);
else
ULOG_ERR("failed to read full module %s\n", path);
close(fd);
free(data);
return ret;
}
static void load_moddeps(struct module *_m)
{
char *dep;
struct module *m;
if (!strcmp(_m->depends, "-") || !strcmp(_m->depends, ""))
return;
dep = _m->depends;
while (*dep) {
m = find_module(dep);
if (!m)
ULOG_ERR("failed to find dependency %s\n", dep);
if (m && (m->state != LOADED)) {
m->state = PROBE;
load_moddeps(m);
}
dep = dep + strlen(dep) + 1;
}
}
static int iterations = 0;
static int load_modprobe(void)
{
int loaded, todo;
struct module *m;
avl_for_each_element(&modules, m, avl)
if (m->state == PROBE)
load_moddeps(m);
do {
loaded = 0;
todo = 0;
avl_for_each_element(&modules, m, avl) {
if ((m->state == PROBE) && (!deps_available(m, 0))) {
if (!insert_module(get_module_path(m->name), (m->opts) ? (m->opts) : (""))) {
m->state = LOADED;
m->error = 0;
loaded++;
continue;
}
m->error = 1;
}
if ((m->state == PROBE) || m->error)
todo++;
}
iterations++;
} while (loaded);
return todo;
}
static int print_insmod_usage(void)
{
ULOG_INFO("Usage:\n\tinsmod filename [args]\n");
return -1;
}
static int print_usage(char *arg)
{
ULOG_INFO("Usage:\n\t%s module\n", arg);
return -1;
}
static int main_insmod(int argc, char **argv)
{
char *name, *cur, *options;
int i, ret, len;
if (argc < 2)
return print_insmod_usage();
name = get_module_name(argv[1]);
if (!name) {
ULOG_ERR("cannot find module - %s\n", argv[1]);
return -1;
}
if (scan_loaded_modules())
return -1;
if (find_module(name)) {
ULOG_ERR("module is already loaded - %s\n", name);
return -1;
}
free_modules();
for (len = 0, i = 2; i < argc; i++)
len += strlen(argv[i]) + 1;
options = malloc(len);
options[0] = 0;
cur = options;
for (i = 2; i < argc; i++) {
if (options[0]) {
*cur = ' ';
cur++;
}
cur += sprintf(cur, "%s", argv[i]);
}
init_module_folders();
if (get_module_path(argv[1])) {
name = argv[1];
} else if (!get_module_path(name)) {
fprintf(stderr, "Failed to find %s. Maybe it is a built in module ?\n", name);
return -1;
}
ret = insert_module(get_module_path(name), options);
free(options);
if (ret)
ULOG_ERR("failed to insert %s\n", get_module_path(name));
return ret;
}
static int main_rmmod(int argc, char **argv)
{
struct module *m;
char *name;
int ret;
if (argc != 2)
return print_usage("rmmod");
if (scan_loaded_modules())
return -1;
name = get_module_name(argv[1]);
m = find_module(name);
if (!m) {
ULOG_ERR("module is not loaded\n");
return -1;
}
ret = syscall(__NR_delete_module, m->name, 0);
if (ret)
ULOG_ERR("unloading the module failed\n");
free_modules();
return ret;
}
static int main_lsmod(int argc, char **argv)
{
struct module *m;
char *dep;
if (scan_loaded_modules())
return -1;
avl_for_each_element(&modules, m, avl)
if (m->state == LOADED) {
printf("%-20s%8d%3d ",
m->name, m->size, m->usage);
if (m->depends && strcmp(m->depends, "-") && strcmp(m->depends, "")) {
dep = m->depends;
while (*dep) {
printf("%s", dep);
dep = dep + strlen(dep) + 1;
if (*dep)
printf(",");
}
}
printf("\n");
}
free_modules();
return 0;
}
static int main_modinfo(int argc, char **argv)
{
struct module *m;
char *name;
if (argc != 2)
return print_usage("modinfo");
if (scan_module_folders())
return -1;
name = get_module_name(argv[1]);
m = find_module(name);
if (!m) {
ULOG_ERR("cannot find module - %s\n", argv[1]);
return -1;
}
name = get_module_path(m->name);
if (!name) {
ULOG_ERR("cannot find path of module - %s\n", m->name);
return -1;
}
print_modinfo(name);
return 0;
}
static int main_modprobe(int argc, char **argv)
{
struct module *m;
char *name;
if (argc != 2)
return print_usage("modprobe");
if (scan_loaded_modules())
return -1;
if (scan_module_folders())
return -1;
name = get_module_name(argv[1]);
m = find_module(name);
if (m && m->state == LOADED) {
ULOG_ERR("%s is already loaded\n", name);
return -1;
} else if (!m) {
ULOG_ERR("failed to find a module named %s\n", name);
} else {
int fail;
m->state = PROBE;
fail = load_modprobe();
if (fail) {
ULOG_ERR("%d module%s could not be probed\n",
fail, (fail == 1) ? ("") : ("s"));
avl_for_each_element(&modules, m, avl)
if ((m->state == PROBE) || m->error)
ULOG_ERR("- %s\n", m->name);
}
}
free_modules();
return 0;
}
static int main_loader(int argc, char **argv)
{
int gl_flags = GLOB_NOESCAPE | GLOB_MARK;
char *dir = "/etc/modules.d/*";
struct module *m;
glob_t gl;
char *path;
int fail, j;
if (argc > 1)
dir = argv[1];
path = malloc(strlen(dir) + 2);
strcpy(path, dir);
strcat(path, "*");
if (scan_loaded_modules())
return -1;
if (scan_module_folders())
return -1;
syslog(LOG_INFO, "kmodloader: loading kernel modules from %s\n", path);
if (glob(path, gl_flags, NULL, &gl) < 0)
goto out;
for (j = 0; j < gl.gl_pathc; j++) {
FILE *fp = fopen(gl.gl_pathv[j], "r");
size_t mod_len = 0;
char *mod = NULL;
if (!fp) {
ULOG_ERR("failed to open %s\n", gl.gl_pathv[j]);
continue;
}
while (getline(&mod, &mod_len, fp) > 0) {
char *nl = strchr(mod, '\n');
struct module *m;
char *opts;
if (nl)
*nl = '\0';
opts = strchr(mod, ' ');
if (opts)
*opts++ = '\0';
m = find_module(get_module_name(mod));
if (!m || (m->state == LOADED))
continue;
if (opts)
m->opts = strdup(opts);
m->state = PROBE;
if (basename(gl.gl_pathv[j])[0] - '0' <= 9)
load_modprobe();
}
free(mod);
fclose(fp);
}
fail = load_modprobe();
if (fail) {
ULOG_ERR("%d module%s could not be probed\n",
fail, (fail == 1) ? ("") : ("s"));
avl_for_each_element(&modules, m, avl)
if ((m->state == PROBE) || (m->error))
ULOG_ERR("- %s - %d\n", m->name, deps_available(m, 1));
}
out:
globfree(&gl);
free(path);
return 0;
}
static int avl_modcmp(const void *k1, const void *k2, void *ptr)
{
const char *s1 = k1;
const char *s2 = k2;
while (*s1 && ((*s1 == *s2) ||
((*s1 == '_') && (*s2 == '-')) ||
((*s1 == '-') && (*s2 == '_'))))
{
s1++;
s2++;
}
return *(const unsigned char *)s1 - *(const unsigned char *)s2;
}
int main(int argc, char **argv)
{
char *exec = basename(*argv);
avl_init(&modules, avl_modcmp, false, NULL);
if (!strcmp(exec, "insmod"))
return main_insmod(argc, argv);
if (!strcmp(exec, "rmmod"))
return main_rmmod(argc, argv);
if (!strcmp(exec, "lsmod"))
return main_lsmod(argc, argv);
if (!strcmp(exec, "modinfo"))
return main_modinfo(argc, argv);
if (!strcmp(exec, "modprobe"))
return main_modprobe(argc, argv);
return main_loader(argc, argv);
}

202
src/3P/ubox/log/logd.c Normal file
View File

@@ -0,0 +1,202 @@
/*
* Copyright (C) 2013 John Crispin <blogic@openwrt.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation
*
* 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.
*/
#include <stdio.h>
#include <unistd.h>
#include <syslog.h>
#include <unistd.h>
#include <linux/types.h>
#include <libubox/uloop.h>
#include <libubox/blobmsg.h>
#include <libubox/list.h>
#include <libubox/ustream.h>
#include <ubus/libubus.h>
#include "syslog.h"
int debug = 0;
static struct blob_buf b;
static struct ubus_auto_conn conn;
static LIST_HEAD(clients);
static const struct blobmsg_policy read_policy =
{ .name = "lines", .type = BLOBMSG_TYPE_INT32 };
static const struct blobmsg_policy write_policy =
{ .name = "event", .type = BLOBMSG_TYPE_STRING };
struct client {
struct list_head list;
struct ustream_fd s;
int fd;
};
static void
client_close(struct ustream *s)
{
struct client *cl = container_of(s, struct client, s.stream);
list_del(&cl->list);
ustream_free(s);
close(cl->fd);
free(cl);
}
static void client_notify_state(struct ustream *s)
{
client_close(s);
}
static int
read_log(struct ubus_context *ctx, struct ubus_object *obj,
struct ubus_request_data *req, const char *method,
struct blob_attr *msg)
{
struct client *cl;
struct blob_attr *tb;
struct log_head *l;
int count = 0;
int fds[2];
int ret;
if (msg) {
blobmsg_parse(&read_policy, 1, &tb, blob_data(msg), blob_len(msg));
if (tb)
count = blobmsg_get_u32(tb);
}
if (pipe(fds) == -1) {
fprintf(stderr, "logd: failed to create pipe: %s\n", strerror(errno));
return -1;
}
ubus_request_set_fd(ctx, req, fds[0]);
cl = calloc(1, sizeof(*cl));
cl->s.stream.notify_state = client_notify_state;
cl->fd = fds[1];
ustream_fd_init(&cl->s, cl->fd);
list_add(&cl->list, &clients);
l = log_list(count, NULL);
while ((!tb || count) && l) {
blob_buf_init(&b, 0);
blobmsg_add_string(&b, "msg", l->data);
blobmsg_add_u32(&b, "id", l->id);
blobmsg_add_u32(&b, "priority", l->priority);
blobmsg_add_u32(&b, "source", l->source);
blobmsg_add_u64(&b, "time", l->ts.tv_sec * 1000LL);
l = log_list(count, l);
ret = ustream_write(&cl->s.stream, (void *) b.head, blob_len(b.head) + sizeof(struct blob_attr), false);
blob_buf_free(&b);
if (ret < 0)
break;
}
return 0;
}
static int
write_log(struct ubus_context *ctx, struct ubus_object *obj,
struct ubus_request_data *req, const char *method,
struct blob_attr *msg)
{
struct blob_attr *tb;
char *event;
if (msg) {
blobmsg_parse(&write_policy, 1, &tb, blob_data(msg), blob_len(msg));
if (tb) {
event = blobmsg_get_string(tb);
log_add(event, strlen(event) + 1, SOURCE_SYSLOG);
}
}
return 0;
}
static const struct ubus_method log_methods[] = {
{ .name = "read", .handler = read_log, .policy = &read_policy, .n_policy = 1 },
{ .name = "write", .handler = write_log, .policy = &write_policy, .n_policy = 1 },
};
static struct ubus_object_type log_object_type =
UBUS_OBJECT_TYPE("log", log_methods);
static struct ubus_object log_object = {
.name = "log",
.type = &log_object_type,
.methods = log_methods,
.n_methods = ARRAY_SIZE(log_methods),
};
void
ubus_notify_log(struct log_head *l)
{
struct client *c;
if (list_empty(&clients))
return;
blob_buf_init(&b, 0);
blobmsg_add_string(&b, "msg", l->data);
blobmsg_add_u32(&b, "id", l->id);
blobmsg_add_u32(&b, "priority", l->priority);
blobmsg_add_u32(&b, "source", l->source);
blobmsg_add_u64(&b, "time", (((__u64) l->ts.tv_sec) * 1000) + (l->ts.tv_nsec / 1000000));
list_for_each_entry(c, &clients, list)
ustream_write(&c->s.stream, (void *) b.head, blob_len(b.head) + sizeof(struct blob_attr), false);
blob_buf_free(&b);
}
static void
ubus_connect_handler(struct ubus_context *ctx)
{
int ret;
ret = ubus_add_object(ctx, &log_object);
if (ret) {
fprintf(stderr, "Failed to add object: %s\n", ubus_strerror(ret));
exit(1);
}
fprintf(stderr, "log: connected to ubus\n");
}
int
main(int argc, char **argv)
{
int ch, log_size = 16;
signal(SIGPIPE, SIG_IGN);
while ((ch = getopt(argc, argv, "S:")) != -1) {
switch (ch) {
case 'S':
log_size = atoi(optarg);
if (log_size < 1)
log_size = 16;
break;
}
}
log_size *= 1024;
uloop_init();
log_init(log_size);
conn.cb = ubus_connect_handler;
ubus_auto_connect(&conn);
uloop_run();
log_shutdown();
uloop_done();
return 0;
}

358
src/3P/ubox/log/logread.c Normal file
View File

@@ -0,0 +1,358 @@
/*
* Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
* Copyright (C) 2013 John Crispin <blogic@openwrt.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation
*
* 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.
*/
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <time.h>
#include <regex.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#define SYSLOG_NAMES
#include <syslog.h>
#include <libubox/ustream.h>
#include <libubox/blobmsg_json.h>
#include <libubox/usock.h>
#include <libubox/uloop.h>
#include <ubus/libubus.h>
#include "syslog.h"
enum {
LOG_STDOUT,
LOG_FILE,
LOG_NET,
};
enum {
LOG_MSG,
LOG_ID,
LOG_PRIO,
LOG_SOURCE,
LOG_TIME,
__LOG_MAX
};
static const struct blobmsg_policy log_policy[] = {
[LOG_MSG] = { .name = "msg", .type = BLOBMSG_TYPE_STRING },
[LOG_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
[LOG_PRIO] = { .name = "priority", .type = BLOBMSG_TYPE_INT32 },
[LOG_SOURCE] = { .name = "source", .type = BLOBMSG_TYPE_INT32 },
[LOG_TIME] = { .name = "time", .type = BLOBMSG_TYPE_INT64 },
};
static struct uloop_timeout retry;
static struct uloop_fd sender;
static regex_t regexp_preg;
static const char *log_file, *log_ip, *log_port, *log_prefix, *pid_file, *hostname, *regexp_pattern;
static int log_type = LOG_STDOUT;
static int log_size, log_udp, log_follow, log_trailer_null = 0;
static const char* getcodetext(int value, CODE *codetable) {
CODE *i;
if (value >= 0)
for (i = codetable; i->c_val != -1; i++)
if (i->c_val == value)
return (i->c_name);
return "<unknown>";
};
static void log_handle_reconnect(struct uloop_timeout *timeout)
{
sender.fd = usock((log_udp) ? (USOCK_UDP) : (USOCK_TCP), log_ip, log_port);
if (sender.fd < 0) {
fprintf(stderr, "failed to connect: %s\n", strerror(errno));
uloop_timeout_set(&retry, 1000);
} else {
uloop_fd_add(&sender, ULOOP_READ);
syslog(LOG_INFO, "Logread connected to %s:%s\n", log_ip, log_port);
}
}
static void log_handle_fd(struct uloop_fd *u, unsigned int events)
{
if (u->eof) {
uloop_fd_delete(u);
close(sender.fd);
sender.fd = -1;
uloop_timeout_set(&retry, 1000);
}
}
static int log_notify(struct blob_attr *msg)
{
struct blob_attr *tb[__LOG_MAX];
struct stat s;
char buf[512];
uint32_t p;
char *str;
time_t t;
char *c, *m;
int ret = 0;
if (sender.fd < 0)
return 0;
blobmsg_parse(log_policy, ARRAY_SIZE(log_policy), tb, blob_data(msg), blob_len(msg));
if (!tb[LOG_ID] || !tb[LOG_PRIO] || !tb[LOG_SOURCE] || !tb[LOG_TIME] || !tb[LOG_MSG])
return 1;
if ((log_type == LOG_FILE) && log_size && (!stat(log_file, &s)) && (s.st_size > log_size)) {
char *old = malloc(strlen(log_file) + 5);
close(sender.fd);
if (old) {
sprintf(old, "%s.old", log_file);
rename(log_file, old);
free(old);
}
sender.fd = open(log_file, O_CREAT | O_WRONLY | O_APPEND, 0600);
if (sender.fd < 0) {
fprintf(stderr, "failed to open %s: %s\n", log_file, strerror(errno));
exit(-1);
}
}
m = blobmsg_get_string(tb[LOG_MSG]);
if (regexp_pattern &&
regexec(&regexp_preg, m, 0, NULL, 0) == REG_NOMATCH)
return 0;
t = blobmsg_get_u64(tb[LOG_TIME]) / 1000;
c = ctime(&t);
p = blobmsg_get_u32(tb[LOG_PRIO]);
c[strlen(c) - 1] = '\0';
str = blobmsg_format_json(msg, true);
if (log_type == LOG_NET) {
int err;
snprintf(buf, sizeof(buf), "<%u>", p);
strncat(buf, c + 4, 16);
if (hostname) {
strncat(buf, hostname, sizeof(buf) - strlen(buf) - 1);
strncat(buf, " ", sizeof(buf) - strlen(buf) - 1);
}
if (log_prefix) {
strncat(buf, log_prefix, sizeof(buf) - strlen(buf) - 1);
strncat(buf, ": ", sizeof(buf) - strlen(buf) - 1);
}
if (blobmsg_get_u32(tb[LOG_SOURCE]) == SOURCE_KLOG)
strncat(buf, "kernel: ", sizeof(buf) - strlen(buf) - 1);
strncat(buf, m, sizeof(buf) - strlen(buf) - 1);
if (log_udp)
err = write(sender.fd, buf, strlen(buf));
else {
size_t buflen = strlen(buf);
if (!log_trailer_null)
buf[buflen] = '\n';
err = send(sender.fd, buf, buflen + 1, 0);
}
if (err < 0) {
syslog(LOG_INFO, "failed to send log data to %s:%s via %s\n",
log_ip, log_port, (log_udp) ? ("udp") : ("tcp"));
uloop_fd_delete(&sender);
close(sender.fd);
sender.fd = -1;
uloop_timeout_set(&retry, 1000);
}
} else {
snprintf(buf, sizeof(buf), "%s %s.%s%s %s\n",
c, getcodetext(LOG_FAC(p) << 3, facilitynames), getcodetext(LOG_PRI(p), prioritynames),
(blobmsg_get_u32(tb[LOG_SOURCE])) ? ("") : (" kernel:"), m);
ret = write(sender.fd, buf, strlen(buf));
}
free(str);
if (log_type == LOG_FILE)
fsync(sender.fd);
return ret;
}
static int usage(const char *prog)
{
fprintf(stderr, "Usage: %s [options]\n"
"Options:\n"
" -s <path> Path to ubus socket\n"
" -l <count> Got only the last 'count' messages\n"
" -e <pattern> Filter messages with a regexp\n"
" -r <server> <port> Stream message to a server\n"
" -F <file> Log file\n"
" -S <bytes> Log size\n"
" -p <file> PID file\n"
" -h <hostname> Add hostname to the message\n"
" -P <prefix> Prefix custom text to streamed messages\n"
" -f Follow log messages\n"
" -u Use UDP as the protocol\n"
" -0 Use \\0 instead of \\n as trailer when using TCP\n"
"\n", prog);
return 1;
}
static void logread_fd_data_cb(struct ustream *s, int bytes)
{
while (true) {
struct blob_attr *a;
int len, cur_len;
a = (void*) ustream_get_read_buf(s, &len);
if (len < sizeof(*a))
break;
cur_len = blob_len(a) + sizeof(*a);
if (len < cur_len)
break;
log_notify(a);
ustream_consume(s, cur_len);
}
if (!log_follow)
uloop_end();
}
static void logread_fd_cb(struct ubus_request *req, int fd)
{
static struct ustream_fd test_fd;
test_fd.stream.notify_read = logread_fd_data_cb;
ustream_fd_init(&test_fd, fd);
}
int main(int argc, char **argv)
{
static struct ubus_request req;
struct ubus_context *ctx;
uint32_t id;
const char *ubus_socket = NULL;
int ch, ret, lines = 0;
static struct blob_buf b;
int tries = 5;
signal(SIGPIPE, SIG_IGN);
while ((ch = getopt(argc, argv, "u0fcs:l:r:F:p:S:P:h:e:")) != -1) {
switch (ch) {
case 'u':
log_udp = 1;
break;
case '0':
log_trailer_null = 1;
break;
case 's':
ubus_socket = optarg;
break;
case 'r':
log_ip = optarg++;
log_port = argv[optind++];
break;
case 'F':
log_file = optarg;
break;
case 'p':
pid_file = optarg;
break;
case 'P':
log_prefix = optarg;
break;
case 'f':
log_follow = 1;
break;
case 'l':
lines = atoi(optarg);
break;
case 'S':
log_size = atoi(optarg);
if (log_size < 1)
log_size = 1;
log_size *= 1024;
break;
case 'h':
hostname = optarg;
break;
case 'e':
if (!regcomp(&regexp_preg, optarg, REG_NOSUB)) {
regexp_pattern = optarg;
}
break;
default:
return usage(*argv);
}
}
uloop_init();
ctx = ubus_connect(ubus_socket);
if (!ctx) {
fprintf(stderr, "Failed to connect to ubus\n");
return -1;
}
ubus_add_uloop(ctx);
/* ugly ugly ugly ... we need a real reconnect logic */
do {
ret = ubus_lookup_id(ctx, "log", &id);
if (ret) {
fprintf(stderr, "Failed to find log object: %s\n", ubus_strerror(ret));
sleep(1);
continue;
}
blob_buf_init(&b, 0);
if (lines)
blobmsg_add_u32(&b, "lines", lines);
else if (log_follow)
blobmsg_add_u32(&b, "lines", 0);
if (log_follow) {
if (pid_file) {
FILE *fp = fopen(pid_file, "w+");
if (fp) {
fprintf(fp, "%d", getpid());
fclose(fp);
}
}
}
if (log_ip && log_port) {
openlog("logread", LOG_PID, LOG_DAEMON);
log_type = LOG_NET;
sender.cb = log_handle_fd;
retry.cb = log_handle_reconnect;
uloop_timeout_set(&retry, 1000);
} else if (log_file) {
log_type = LOG_FILE;
sender.fd = open(log_file, O_CREAT | O_WRONLY| O_APPEND, 0600);
if (sender.fd < 0) {
fprintf(stderr, "failed to open %s: %s\n", log_file, strerror(errno));
exit(-1);
}
} else {
sender.fd = STDOUT_FILENO;
}
ubus_invoke_async(ctx, id, "read", b.head, &req);
req.fd_cb = logread_fd_cb;
ubus_complete_request_async(ctx, &req);
uloop_run();
ubus_free(ctx);
uloop_done();
} while (ret && tries--);
return ret;
}

309
src/3P/ubox/log/syslog.c Normal file
View File

@@ -0,0 +1,309 @@
/*
* Copyright (C) 2013 John Crispin <blogic@openwrt.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation
*
* 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.
*/
#include <linux/version.h>
#if LINUX_VERSION_CODE == KERNEL_VERSION(2,6,21)
typedef unsigned short sa_family_t;
#endif
#include <linux/un.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <regex.h>
#include <time.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <syslog.h>
#include <libubox/uloop.h>
#include <libubox/usock.h>
#include <libubox/ustream.h>
#include "syslog.h"
#define LOG_DEFAULT_SIZE (16 * 1024)
#define LOG_DEFAULT_SOCKET "/dev/log"
#define LOG_LINE_LEN 256
#define SYSLOG_PADDING 16
#define KLOG_DEFAULT_PROC "/proc/kmsg"
#define PAD(x) (x % 4) ? (((x) - (x % 4)) + 4) : (x)
static char *log_dev = LOG_DEFAULT_SOCKET;
static int log_size = LOG_DEFAULT_SIZE;
static struct log_head *log, *log_end, *oldest, *newest;
static int current_id = 0;
static regex_t pat_prio;
static regex_t pat_tstamp;
static struct log_head*
log_next(struct log_head *h, int size)
{
struct log_head *n = (struct log_head *) &h->data[PAD(sizeof(struct log_head) + size)];
return (n >= log_end) ? (log) : (n);
}
void
log_add(char *buf, int size, int source)
{
regmatch_t matches[4];
struct log_head *next;
int priority = 0;
int ret;
/* bounce out if we don't have init'ed yet (regmatch etc will blow) */
if (!log) {
fprintf(stderr, "%s", buf);
return;
}
/* strip trailing newline */
if (buf[size - 2] == '\n') {
buf[size - 2] = '\0';
size -= 1;
}
/* strip the priority */
ret = regexec(&pat_prio, buf, 3, matches, 0);
if (!ret) {
priority = atoi(&buf[matches[1].rm_so]);
size -= matches[2].rm_so;
buf += matches[2].rm_so;
}
#if 0
/* strip kernel timestamp */
ret = regexec(&pat_tstamp,buf, 4, matches, 0);
if ((source == SOURCE_KLOG) && !ret) {
size -= matches[3].rm_so;
buf += matches[3].rm_so;
}
#endif
/* strip syslog timestamp */
if ((source == SOURCE_SYSLOG) && (size > SYSLOG_PADDING) && (buf[SYSLOG_PADDING - 1] == ' ')) {
size -= SYSLOG_PADDING;
buf += SYSLOG_PADDING;
}
//fprintf(stderr, "-> %d - %s\n", priority, buf);
/* find new oldest entry */
next = log_next(newest, size);
if (next > newest) {
while ((oldest > newest) && (oldest <= next) && (oldest != log))
oldest = log_next(oldest, oldest->size);
} else {
//fprintf(stderr, "Log wrap\n");
newest->size = 0;
next = log_next(log, size);
for (oldest = log; oldest <= next; oldest = log_next(oldest, oldest->size))
;
newest = log;
}
/* add the log message */
newest->size = size;
newest->id = current_id++;
newest->priority = priority;
newest->source = source;
clock_gettime(CLOCK_REALTIME, &newest->ts);
strcpy(newest->data, buf);
ubus_notify_log(newest);
newest = next;
}
static void
slog_cb(struct ustream *s, int bytes)
{
struct ustream_buf *buf = s->r.head;
char *str;
int len;
do {
str = ustream_get_read_buf(s, NULL);
if (!str)
break;
len = strlen(buf->data);
if (!len) {
bytes -= 1;
ustream_consume(s, 1);
continue;
}
log_add(buf->data, len + 1, SOURCE_SYSLOG);
ustream_consume(s, len);
bytes -= len;
} while (bytes > 0);
}
static void
klog_cb(struct ustream *s, int bytes)
{
struct ustream_buf *buf = s->r.head;
char *newline, *str;
int len;
do {
str = ustream_get_read_buf(s, NULL);
if (!str)
break;
newline = strchr(buf->data, '\n');
if (!newline)
break;
*newline = 0;
len = newline + 1 - str;
log_add(buf->data, len, SOURCE_KLOG);
ustream_consume(s, len);
} while (1);
}
struct ustream_fd slog = {
.stream.string_data = true,
.stream.notify_read = slog_cb,
};
struct ustream_fd klog = {
.stream.string_data = true,
.stream.notify_read = klog_cb,
};
static int
klog_open(void)
{
int fd;
fd = open(KLOG_DEFAULT_PROC, O_RDONLY | O_NONBLOCK);
if (fd < 0) {
fprintf(stderr, "Failed to open %s\n", KLOG_DEFAULT_PROC);
return -1;
}
fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC);
ustream_fd_init(&klog, fd);
return 0;
}
static int
syslog_open(void)
{
int fd;
unlink(log_dev);
fd = usock(USOCK_UNIX | USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK, log_dev, NULL);
if (fd < 0) {
fprintf(stderr,"Failed to open %s\n", log_dev);
return -1;
}
chmod(log_dev, 0666);
ustream_fd_init(&slog, fd);
return 0;
}
struct log_head*
log_list(int count, struct log_head *h)
{
unsigned int min = count;
if (count)
min = (count < current_id) ? (current_id - count) : (0);
if (!h && oldest->id >= min)
return oldest;
if (!h)
h = oldest;
while (h != newest) {
h = log_next(h, h->size);
if (!h->size && (h > newest))
h = log;
if (h->id >= min && (h != newest))
return h;
}
return NULL;
}
int
log_buffer_init(int size)
{
struct log_head *_log = malloc(size);
if (!_log) {
fprintf(stderr, "Failed to initialize log buffer with size %d\n", log_size);
return -1;
}
memset(_log, 0, size);
if (log && ((log_size + sizeof(struct log_head)) < size)) {
struct log_head *start = _log;
struct log_head *end = ((void*) _log) + size;
struct log_head *l;
l = log_list(0, NULL);
while ((start < end) && l && l->size) {
memcpy(start, l, PAD(sizeof(struct log_head) + l->size));
start = (struct log_head *) &l->data[PAD(l->size)];
l = log_list(0, l);
}
free(log);
newest = start;
newest->size = 0;
oldest = log = _log;
log_end = ((void*) log) + size;
} else {
oldest = newest = log = _log;
log_end = ((void*) log) + size;
}
log_size = size;
return 0;
}
void
log_init(int _log_size)
{
if (_log_size > 0)
log_size = _log_size;
regcomp(&pat_prio, "^<([0-9]*)>(.*)", REG_EXTENDED);
regcomp(&pat_tstamp, "^\[[ 0]*([0-9]*).([0-9]*)] (.*)", REG_EXTENDED);
if (log_buffer_init(log_size)) {
fprintf(stderr, "Failed to allocate log memory\n");
exit(-1);
}
syslog_open();
klog_open();
openlog("sysinit", LOG_CONS, LOG_DAEMON);
}
void
log_shutdown(void)
{
ustream_free(&slog.stream);
ustream_free(&klog.stream);
close(slog.fd.fd);
close(klog.fd.fd);
free(log);
}

42
src/3P/ubox/log/syslog.h Normal file
View File

@@ -0,0 +1,42 @@
/*
* Copyright (C) 2013 John Crispin <blogic@openwrt.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation
*
* 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.
*/
#ifndef __SYSLOG_H
#define __SYSLOG_H
enum {
SOURCE_KLOG = 0,
SOURCE_SYSLOG = 1,
SOURCE_INTERNAL = 2,
SOURCE_ANY = 0xff,
};
struct log_head {
unsigned int size;
unsigned int id;
int priority;
int source;
struct timespec ts;
char data[];
};
void log_init(int log_size);
void log_shutdown(void);
typedef void (*log_list_cb)(struct log_head *h);
struct log_head* log_list(int count, struct log_head *h);
int log_buffer_init(int size);
void log_add(char *buf, int size, int source);
void ubus_notify_log(struct log_head *l);
#endif

187
src/3P/ubox/lsbloader.c Normal file
View File

@@ -0,0 +1,187 @@
/*
* 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; version 2 of the License
*
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
*
* Copyright (C) 2012 John Crispin <blogic@openwrt.org>
*/
#include <stdio.h>
#include <stdlib.h>
#define __USE_GNU
#include <string.h>
#include <regex.h>
#include <glob.h>
#include <libubox/list.h>
#include <libubox/blobmsg_json.h>
#include "libubus.h"
struct initd {
struct list_head list;
char *name;
char *exec;
char *desc;
char *tpl;
char **deps;
int start;
int stop;
};
static LIST_HEAD(initds);
static regex_t pat_provides, pat_require, pat_start, pat_stop, pat_desc, pat_exec, pat_tpl;
static struct ubus_context *ctx;
static struct blob_buf b;
static uint32_t service;
static void initd_free(struct initd *i)
{
if (i->name)
free(i->name);
if (i->exec)
free(i->exec);
if (i->desc)
free(i->desc);
if (i->tpl)
free(i->tpl);
}
static int initd_parse(const char *file)
{
FILE *fp;
struct initd *i;
regmatch_t matches[2];
char buffer[1024];
ssize_t len;
fp = fopen(file, "r");
if (!fp) {
fprintf(stderr, "failed to open %s\n", file);
return -1;
}
len = fread(buffer, 1, sizeof(buffer) - 1, fp);
fclose(fp);
if (len < 1) {
fprintf(stderr, "failed to read from %s\n", file);
return -1;
}
buffer[len] = '\0';
i = malloc(sizeof(struct initd));
if (!i) {
fprintf(stderr, "failed to alloc initd struct\n");
return -1;
}
memset(i, 0, sizeof(*i));
if (!regexec(&pat_provides, buffer, 2, matches, 0))
i->name = strndup(buffer + matches[1].rm_so, (size_t)matches[1].rm_eo - matches[1].rm_so);
if (!regexec(&pat_exec, buffer, 2, matches, 0))
i->exec = strndup(buffer + matches[1].rm_so, matches[1].rm_eo - matches[1].rm_so);
if (!regexec(&pat_desc, buffer, 2, matches, 0))
i->desc = strndup(buffer + matches[1].rm_so, matches[1].rm_eo - matches[1].rm_so);
if (!regexec(&pat_tpl, buffer, 2, matches, 0))
i->tpl = strndup(buffer + matches[1].rm_so, matches[1].rm_eo - matches[1].rm_so);
if (!regexec(&pat_start, buffer, 2, matches, 0))
i->start += atoi(buffer + matches[1].rm_so);
if (!regexec(&pat_stop, buffer, 2, matches, 0))
i->stop += atoi(buffer + matches[1].rm_so);
if (i->name && i->exec)
list_add(&i->list, &initds);
else
initd_free(i);
return 0;
}
static void initd_init(void)
{
int gl_flags = GLOB_NOESCAPE | GLOB_MARK;
glob_t gl;
regcomp(&pat_provides, "# Provides:[ \t]*([a-zA-Z0-9]+)", REG_EXTENDED);
regcomp(&pat_require, "# Required-Start:[ \t]*([a-zA-Z0-9 ]+)", REG_EXTENDED);
regcomp(&pat_start, "# Default-Start:[ \t]*([0-9])", REG_EXTENDED);
regcomp(&pat_stop, "# Default-Stop:[ \t]*([0-9])", REG_EXTENDED);
regcomp(&pat_desc, "# Description:[ \t]*([a-zA-Z0-9 ]+)", REG_EXTENDED);
regcomp(&pat_exec, "# X-Exec:[ \t]*([a-zA-Z0-9/ ]+)", REG_EXTENDED);
regcomp(&pat_tpl, "# X-Template:[ \t]*([a-zA-Z0-9/.]+)", REG_EXTENDED);
if (glob("/etc/rc.d/P*", gl_flags, NULL, &gl) >= 0) {
int j;
for (j = 0; j < gl.gl_pathc; j++)
initd_parse(gl.gl_pathv[j]);
}
globfree(&gl);
regfree(&pat_provides);
regfree(&pat_require);
regfree(&pat_start);
regfree(&pat_stop);
regfree(&pat_desc);
regfree(&pat_exec);
regfree(&pat_tpl);
}
static int init_services(void)
{
struct initd *i;
void *instances, *instance, *command;
list_for_each_entry(i, &initds, list) {
char *t;
blob_buf_init(&b, 0);
blobmsg_add_string(&b, "name", i->name);
instances = blobmsg_open_table(&b, "instances");
instance = blobmsg_open_table(&b, "instance");
command = blobmsg_open_array(&b, "command");
t = strtok(i->exec, " ");
while (t) {
blobmsg_add_string(&b, NULL, t);
t = strtok(NULL, " ");
}
blobmsg_close_array(&b, command);
blobmsg_close_table(&b, instance);
blobmsg_close_table(&b, instances);
ubus_invoke(ctx, service, "add", b.head, NULL, 0, 1000);
}
return 0;
}
int main(int argc, char **argv)
{
int ret;
initd_init();
if (list_empty(&initds))
return 0;
ctx = ubus_connect(NULL);
if (!ctx) {
fprintf(stderr, "Failed to connect to ubus\n");
return -1;
}
ret = ubus_lookup_id(ctx, "service", &service);
if (ret) {
fprintf(stderr, "Failed to find service object: %s\n", ubus_strerror(ret));
return -1;
}
return init_services();
}

255
src/3P/ubox/validate/cli.c Normal file
View File

@@ -0,0 +1,255 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include <ctype.h>
#include <arpa/inet.h>
#include <netinet/ether.h>
#include <sys/stat.h>
#include <uci.h>
#include "libvalidate.h"
static void
print_usage(char *argv)
{
fprintf(stderr, "%s <datatype> <value>\t- validate a value against a type\n", argv);
fprintf(stderr, "%s <package> <section_type> <section_name> 'option:datatype:default' 'option:datatype:default' ...\n", argv);
}
static const char *
bool_to_num(const char *val)
{
if (!strcmp(val, "0") || !strcmp(val, "off") || !strcmp(val, "false") || !strcmp(val, "no") || !strcmp(val, "disabled"))
return "0";
if (!strcmp(val, "1") || !strcmp(val, "on") || !strcmp(val, "true") || !strcmp(val, "yes") || !strcmp(val, "enabled"))
return "1";
return "";
}
static bool
parse_tuple(char *tuple, char **option, char **expr, char **def)
{
char *p;
bool esc;
for (esc = false, p = *option = tuple, *expr = NULL, *def = NULL; *p; p++)
{
if (!esc && *p == '\\')
{
esc = true;
continue;
}
if (!esc && *p == ':')
{
*p++ = 0;
if (!*expr)
*expr = p;
else if (!*def)
*def = p;
else
break;
}
esc = false;
}
return (*expr != NULL);
}
static void
escape_value(enum dt_type type, const char *val)
{
const char *p;
switch(type)
{
case DT_BOOL:
printf("%s", bool_to_num(val));
break;
case DT_STRING:
printf("'");
for (p = val; *p; p++)
if (*p == '\'')
printf("'\"'\"'");
else
printf("%c", *p);
printf("'");
break;
default:
printf("%s", val);
break;
}
}
static void
export_value(enum dt_type type, const char *name, const char *val)
{
if ((type == DT_INVALID) || !val || !*val)
{
printf("unset -v %s; ", name);
return;
}
printf("%s=", name);
escape_value(type, val);
printf("; ");
}
static int
validate_value(struct uci_ptr *ptr, const char *expr, const char *def)
{
int i = 0;
bool empty = true, first = true;
enum dt_type type = DT_INVALID;
struct uci_element *e;
struct uci_option *opt = ptr->o;
if (opt->type == UCI_TYPE_LIST)
{
uci_foreach_element(&opt->v.list, e)
{
if (!e->name || !*e->name)
continue;
empty = false;
break;
}
if (empty)
{
export_value(DT_STRING, ptr->option, def);
return 0;
}
uci_foreach_element(&opt->v.list, e)
{
if (!e->name || !*e->name)
continue;
if (first)
printf("%s=", ptr->option);
else
printf("\\ ");
first = false;
type = dt_parse(expr, e->name);
if (type != DT_INVALID)
escape_value(type, e->name);
fprintf(stderr, "%s.%s.%s[%u]=%s validates as %s with %s\n",
ptr->package, ptr->section, ptr->option, i++, e->name,
expr, type ? "true" : "false");
}
printf("; ");
}
else
{
if (!opt->v.string || !*opt->v.string)
{
export_value(DT_STRING, ptr->option, def);
return 0;
}
type = dt_parse(expr, opt->v.string);
export_value(type, ptr->option, opt->v.string);
fprintf(stderr, "%s.%s.%s=%s validates as %s with %s\n",
ptr->package, ptr->section, ptr->option, opt->v.string,
expr, type ? "true" : "false");
}
return type ? 0 : -1;
}
static int
validate_option(struct uci_context *ctx, char *package, char *section, char *option)
{
char *opt, *expr, *def;
struct uci_ptr ptr = { 0 };
if (!parse_tuple(option, &opt, &expr, &def))
{
fprintf(stderr, "%s is not a valid option\n", option);
return -1;
}
ptr.package = package;
ptr.section = section;
ptr.option = opt;
if (uci_lookup_ptr(ctx, &ptr, NULL, false) ||
!(ptr.flags & UCI_LOOKUP_COMPLETE) ||
(ptr.last->type != UCI_TYPE_OPTION))
{
export_value(DT_STRING, opt, def);
return 0;
}
return validate_value(&ptr, expr, def);
}
int
main(int argc, char **argv)
{
struct uci_context *ctx;
struct uci_package *package;
char *opt, *expr, *def;
int len = argc - 4;
enum dt_type rv;
int i, rc;
if (argc == 3) {
rv = dt_parse(argv[1], argv[2]);
fprintf(stderr, "%s - %s = %s\n", argv[1], argv[2], rv ? "true" : "false");
return rv ? 0 : 1;
} else if (argc < 5) {
print_usage(*argv);
return -1;
}
if (*argv[3] == '\0') {
printf("json_add_object; ");
printf("json_add_string \"package\" \"%s\"; ", argv[1]);
printf("json_add_string \"type\" \"%s\"; ", argv[2]);
printf("json_add_object \"data\"; ");
for (i = 0; i < len; i++) {
if (!parse_tuple(argv[4 + i], &opt, &expr, &def))
continue;
printf("json_add_string \"%s\" \"%s\"; ", opt, expr);
}
printf("json_close_object; ");
printf("json_close_object; ");
return 0;
}
ctx = uci_alloc_context();
if (!ctx)
return -1;
if (uci_load(ctx, argv[1], &package))
return -1;
rc = 0;
for (i = 0; i < len; i++) {
if (validate_option(ctx, argv[1], argv[3], argv[4 + i])) {
rc = -1;
}
}
return rc;
}

View File

@@ -0,0 +1,13 @@
#ifndef _VALIDATE_H__
#define _VALIDATE_H__
enum dt_type {
DT_INVALID,
DT_BOOL,
DT_NUMBER,
DT_STRING
};
enum dt_type dt_parse(const char *code, const char *value);
#endif

File diff suppressed because it is too large Load Diff