add ubus system object

1 files changed, 154 insertions, 0 deletions

diff --git a/system.c b/system.c
new file mode 100644
index 0000000..c35f75a
--- /dev/null
+++ b/system.c
@@ -0,0 +1,154 @@
+/*
+ * 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>
+#include <sys/ioctl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+
+#include <unistd.h>
+
+#include <libubox/uloop.h>
+
+#include "procd.h"
+#include "watchdog.h"
+
+#define HOSTNAME_PATH	"/proc/sys/kernel/hostname"
+
+static struct blob_buf b;
+static char *board_name;
+static char *board_model;
+
+static int system_info(struct ubus_context *ctx, struct ubus_object *obj,
+			struct ubus_request_data *req, const char *method,
+			struct blob_attr *msg)
+{
+	struct timespec ts;
+	struct utsname uts;
+
+	blob_buf_init(&b, 0);
+	if (board_name && board_model) {
+		blobmsg_add_string(&b, "boardname", board_name);
+		blobmsg_add_string(&b, "boardmodel", board_model);
+	}
+	if (!uname(&uts)) {
+		blobmsg_add_string(&b, "hostname", uts.nodename);
+		blobmsg_add_string(&b, "machine", uts.machine);
+		blobmsg_add_string(&b, "kernel", uts.release);
+	}
+	if (!clock_gettime(CLOCK_MONOTONIC, &ts))
+		blobmsg_add_u32(&b, "uptime", ts.tv_sec);
+	ubus_send_reply(ctx, req, b.head);
+
+	return 0;
+}
+
+enum {
+	WDT_FREQUENCY,
+	WDT_TIMEOUT,
+	__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 },
+};
+
+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];
+
+	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);
+	}
+
+	blob_buf_init(&b, 0);
+	blobmsg_add_string(&b, "status", (watchdog_fd() >= 0) ? ("running") : ("offline"));
+	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;
+}
+
+static const struct ubus_method system_methods[] = {
+	UBUS_METHOD_NOARG("info", system_info),
+	UBUS_METHOD("watchdog", watchdog_set, watchdog_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),
+};
+
+static char* load_file_content(const char *file)
+{
+	char buf[32];
+	int fd, r;
+
+	fd = open(file, O_RDONLY);
+	if (!fd)
+		return NULL;
+	r = read(fd, buf, sizeof(buf) - 1);
+	close(fd);
+	if (r < 1)
+		return NULL;
+	if (buf[r - 1] == '\n')
+		buf[r - 1] = '\0';
+	else
+		buf[r] = '\0';
+
+	return strdup(buf);
+}
+
+void ubus_init_system(struct ubus_context *ctx)
+{
+	int ret;
+
+	if (!board_model)
+		board_model = load_file_content("/tmp/sysinfo/model");
+	if (!board_name);
+		board_name = load_file_content("/tmp/sysinfo/board_name");
+	ret = ubus_add_object(ctx, &system_object);
+	if (ret)
+		ERROR("Failed to add object: %s\n", ubus_strerror(ret));
+}