123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434 |
- /*
- * 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/utsname.h>
- #ifdef linux
- #include <sys/sysinfo.h>
- #endif
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <fcntl.h>
- #include <signal.h>
- #include <unistd.h>
- #include <stdlib.h>
- #include <libubox/uloop.h>
- #include "procd.h"
- #include "watchdog.h"
- static struct blob_buf b;
- static int notify;
- static struct ubus_context *_ctx;
- int upgrade_running = 0;
- static int system_board(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- void *c;
- char line[256];
- char *key, *val, *next;
- struct utsname utsname;
- FILE *f;
- blob_buf_init(&b, 0);
- if (uname(&utsname) >= 0)
- {
- blobmsg_add_string(&b, "kernel", utsname.release);
- blobmsg_add_string(&b, "hostname", utsname.nodename);
- }
- if ((f = fopen("/proc/cpuinfo", "r")) != NULL)
- {
- while(fgets(line, sizeof(line), f))
- {
- key = strtok(line, "\t:");
- val = strtok(NULL, "\t\n");
- if (!key || !val)
- continue;
- if (!strcasecmp(key, "system type") ||
- !strcasecmp(key, "processor") ||
- !strcasecmp(key, "model name"))
- {
- strtoul(val + 2, &key, 0);
- if (key == (val + 2) || *key != 0)
- {
- blobmsg_add_string(&b, "system", val + 2);
- break;
- }
- }
- }
- fclose(f);
- }
- if ((f = fopen("/tmp/sysinfo/model", "r")) != NULL ||
- (f = fopen("/proc/device-tree/model", "r")) != NULL)
- {
- if (fgets(line, sizeof(line), f))
- {
- val = strtok(line, "\t\n");
- if (val)
- blobmsg_add_string(&b, "model", val);
- }
- fclose(f);
- }
- else if ((f = fopen("/proc/cpuinfo", "r")) != NULL)
- {
- while(fgets(line, sizeof(line), f))
- {
- key = strtok(line, "\t:");
- val = strtok(NULL, "\t\n");
- if (!key || !val)
- continue;
- if (!strcasecmp(key, "machine") ||
- !strcasecmp(key, "hardware"))
- {
- blobmsg_add_string(&b, "model", val + 2);
- break;
- }
- }
- fclose(f);
- }
- if ((f = fopen("/etc/openwrt_release", "r")) != NULL)
- {
- c = blobmsg_open_table(&b, "release");
- while (fgets(line, sizeof(line), f))
- {
- char *dest;
- char ch;
- key = line;
- val = strchr(line, '=');
- if (!val)
- continue;
- *(val++) = 0;
- if (!strcasecmp(key, "DISTRIB_ID"))
- key = "distribution";
- else if (!strcasecmp(key, "DISTRIB_RELEASE"))
- key = "version";
- else if (!strcasecmp(key, "DISTRIB_REVISION"))
- key = "revision";
- else if (!strcasecmp(key, "DISTRIB_CODENAME"))
- key = "codename";
- else if (!strcasecmp(key, "DISTRIB_TARGET"))
- key = "target";
- else if (!strcasecmp(key, "DISTRIB_DESCRIPTION"))
- key = "description";
- else
- continue;
- dest = blobmsg_alloc_string_buffer(&b, key, strlen(val));
- if (!dest) {
- ERROR("Failed to allocate blob.\n");
- continue;
- }
- while (val && (ch = *(val++)) != 0) {
- switch (ch) {
- case '\'':
- case '"':
- next = strchr(val, ch);
- if (next)
- *next = 0;
- strcpy(dest, val);
- if (next)
- val = next + 1;
- dest += strlen(dest);
- break;
- case '\\':
- *(dest++) = *(val++);
- break;
- }
- }
- blobmsg_add_string_buffer(&b);
- }
- blobmsg_close_array(&b, c);
- fclose(f);
- }
- ubus_send_reply(ctx, req, b.head);
- return UBUS_STATUS_OK;
- }
- static int system_info(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- time_t now;
- struct tm *tm;
- #ifdef linux
- struct sysinfo info;
- void *c;
- if (sysinfo(&info))
- return UBUS_STATUS_UNKNOWN_ERROR;
- #endif
- now = time(NULL);
- if (!(tm = localtime(&now)))
- return UBUS_STATUS_UNKNOWN_ERROR;
- blob_buf_init(&b, 0);
- blobmsg_add_u32(&b, "localtime", mktime(tm));
- #ifdef linux
- blobmsg_add_u32(&b, "uptime", info.uptime);
- c = blobmsg_open_array(&b, "load");
- blobmsg_add_u32(&b, NULL, info.loads[0]);
- blobmsg_add_u32(&b, NULL, info.loads[1]);
- blobmsg_add_u32(&b, NULL, info.loads[2]);
- blobmsg_close_array(&b, c);
- c = blobmsg_open_table(&b, "memory");
- blobmsg_add_u64(&b, "total", info.mem_unit * info.totalram);
- blobmsg_add_u64(&b, "free", info.mem_unit * info.freeram);
- blobmsg_add_u64(&b, "shared", info.mem_unit * info.sharedram);
- blobmsg_add_u64(&b, "buffered", info.mem_unit * info.bufferram);
- blobmsg_close_table(&b, c);
- c = blobmsg_open_table(&b, "swap");
- blobmsg_add_u64(&b, "total", info.mem_unit * info.totalswap);
- blobmsg_add_u64(&b, "free", info.mem_unit * info.freeswap);
- blobmsg_close_table(&b, c);
- #endif
- ubus_send_reply(ctx, req, b.head);
- return UBUS_STATUS_OK;
- }
- static int system_upgrade(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- upgrade_running = 1;
- return 0;
- }
- enum {
- WDT_FREQUENCY,
- WDT_TIMEOUT,
- WDT_STOP,
- __WDT_MAX
- };
- static const struct blobmsg_policy watchdog_policy[__WDT_MAX] = {
- [WDT_FREQUENCY] = { .name = "frequency", .type = BLOBMSG_TYPE_INT32 },
- [WDT_TIMEOUT] = { .name = "timeout", .type = BLOBMSG_TYPE_INT32 },
- [WDT_STOP] = { .name = "stop", .type = BLOBMSG_TYPE_BOOL },
- };
- static int watchdog_set(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- struct blob_attr *tb[__WDT_MAX];
- const char *status;
- if (!msg)
- return UBUS_STATUS_INVALID_ARGUMENT;
- blobmsg_parse(watchdog_policy, __WDT_MAX, tb, blob_data(msg), blob_len(msg));
- if (tb[WDT_FREQUENCY]) {
- unsigned int timeout = watchdog_timeout(0);
- unsigned int freq = blobmsg_get_u32(tb[WDT_FREQUENCY]);
- if (freq) {
- if (freq > timeout / 2)
- freq = timeout / 2;
- watchdog_frequency(freq);
- }
- }
- if (tb[WDT_TIMEOUT]) {
- unsigned int timeout = blobmsg_get_u32(tb[WDT_TIMEOUT]);
- unsigned int frequency = watchdog_frequency(0);
- if (timeout <= frequency)
- timeout = frequency * 2;
- watchdog_timeout(timeout);
- }
- if (tb[WDT_STOP])
- watchdog_set_stopped(blobmsg_get_bool(tb[WDT_STOP]));
- if (watchdog_fd() == NULL)
- status = "offline";
- else if (watchdog_get_stopped())
- status = "stopped";
- else
- status = "running";
- blob_buf_init(&b, 0);
- blobmsg_add_string(&b, "status", status);
- blobmsg_add_u32(&b, "timeout", watchdog_timeout(0));
- blobmsg_add_u32(&b, "frequency", watchdog_frequency(0));
- ubus_send_reply(ctx, req, b.head);
- return 0;
- }
- enum {
- SIGNAL_PID,
- SIGNAL_NUM,
- __SIGNAL_MAX
- };
- static const struct blobmsg_policy signal_policy[__SIGNAL_MAX] = {
- [SIGNAL_PID] = { .name = "pid", .type = BLOBMSG_TYPE_INT32 },
- [SIGNAL_NUM] = { .name = "signum", .type = BLOBMSG_TYPE_INT32 },
- };
- static int proc_signal(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- struct blob_attr *tb[__SIGNAL_MAX];
- if (!msg)
- return UBUS_STATUS_INVALID_ARGUMENT;
- blobmsg_parse(signal_policy, __SIGNAL_MAX, tb, blob_data(msg), blob_len(msg));
- if (!tb[SIGNAL_PID || !tb[SIGNAL_NUM]])
- return UBUS_STATUS_INVALID_ARGUMENT;
- kill(blobmsg_get_u32(tb[SIGNAL_PID]), blobmsg_get_u32(tb[SIGNAL_NUM]));
- return 0;
- }
- enum {
- NAND_PATH,
- __NAND_MAX
- };
- static const struct blobmsg_policy nand_policy[__NAND_MAX] = {
- [NAND_PATH] = { .name = "path", .type = BLOBMSG_TYPE_STRING },
- };
- static void
- procd_spawn_upgraded(char *path)
- {
- char *wdt_fd = watchdog_fd();
- char *argv[] = { "/tmp/upgraded", NULL, NULL};
- argv[1] = path;
- DEBUG(2, "Exec to upgraded now\n");
- if (wdt_fd) {
- watchdog_no_cloexec();
- setenv("WDTFD", wdt_fd, 1);
- }
- execvp(argv[0], argv);
- }
- static int nand_set(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- struct blob_attr *tb[__NAND_MAX];
- if (!msg)
- return UBUS_STATUS_INVALID_ARGUMENT;
- blobmsg_parse(nand_policy, __NAND_MAX, tb, blob_data(msg), blob_len(msg));
- if (!tb[NAND_PATH])
- return UBUS_STATUS_INVALID_ARGUMENT;
- procd_spawn_upgraded(blobmsg_get_string(tb[NAND_PATH]));
- fprintf(stderr, "Yikees, something went wrong. no /sbin/upgraded ?\n");
- return 0;
- }
- static void
- procd_subscribe_cb(struct ubus_context *ctx, struct ubus_object *obj)
- {
- notify = obj->has_subscribers;
- }
- static const struct ubus_method system_methods[] = {
- UBUS_METHOD_NOARG("board", system_board),
- UBUS_METHOD_NOARG("info", system_info),
- UBUS_METHOD_NOARG("upgrade", system_upgrade),
- UBUS_METHOD("watchdog", watchdog_set, watchdog_policy),
- UBUS_METHOD("signal", proc_signal, signal_policy),
- /* must remain at the end as it ia not always loaded */
- UBUS_METHOD("nandupgrade", nand_set, nand_policy),
- };
- static struct ubus_object_type system_object_type =
- UBUS_OBJECT_TYPE("system", system_methods);
- static struct ubus_object system_object = {
- .name = "system",
- .type = &system_object_type,
- .methods = system_methods,
- .n_methods = ARRAY_SIZE(system_methods),
- .subscribe_cb = procd_subscribe_cb,
- };
- void
- procd_bcast_event(char *event, struct blob_attr *msg)
- {
- int ret;
- if (!notify)
- return;
- ret = ubus_notify(_ctx, &system_object, event, msg, -1);
- if (ret)
- fprintf(stderr, "Failed to notify log: %s\n", ubus_strerror(ret));
- }
- void ubus_init_system(struct ubus_context *ctx)
- {
- struct stat s;
- int ret;
- if (stat("/sbin/upgraded", &s))
- system_object.n_methods -= 1;
- _ctx = ctx;
- ret = ubus_add_object(ctx, &system_object);
- if (ret)
- ERROR("Failed to add object: %s\n", ubus_strerror(ret));
- }
|