+/**
+ * collectd - src/check_uptime.c
+ * Copyright (C) 2007-2019 Florian Forster
+ * Copyright (C) 2019 Pavel V. Rochnyack
+ *
+ * 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; only version 2 of the License is applicable.
+ *
+ * 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.,
+ * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+ *
+ * Author:
+ * Florian octo Forster <octo at collectd.org>
+ * Pavel Rochnyak <pavel2000 ngs.ru>
+ **/
+
+#include "collectd.h"
+#include "plugin.h"
+#include "utils/avltree/avltree.h"
+#include "utils/common/common.h"
+#include "utils_cache.h"
+
+/* Types are registered only in `config` phase, so access is not protected by
+ * locks */
+c_avl_tree_t *types_tree = NULL;
+
+static int format_uptime(unsigned long uptime_sec, char *buf, size_t bufsize) {
+
+ unsigned int uptime_days = uptime_sec / 24 / 3600;
+ uptime_sec -= uptime_days * 24 * 3600;
+ unsigned int uptime_hours = uptime_sec / 3600;
+ uptime_sec -= uptime_hours * 3600;
+ unsigned int uptime_mins = uptime_sec / 60;
+ uptime_sec -= uptime_mins * 60;
+
+ int ret = 0;
+ if (uptime_days) {
+ ret += snprintf(buf + ret, bufsize - ret, " %u day(s)", uptime_days);
+ }
+ if (uptime_days || uptime_hours) {
+ ret += snprintf(buf + ret, bufsize - ret, " %u hour(s)", uptime_hours);
+ }
+ if (uptime_days || uptime_hours || uptime_mins) {
+ ret += snprintf(buf + ret, bufsize - ret, " %u min", uptime_mins);
+ }
+ ret += snprintf(buf + ret, bufsize - ret, " %lu sec.", uptime_sec);
+ return ret;
+}
+
+static int cu_notify(enum cache_event_type_e event_type, const value_list_t *vl,
+ gauge_t old_uptime, gauge_t new_uptime) {
+ notification_t n;
+ NOTIFICATION_INIT_VL(&n, vl);
+
+ int status;
+ char *buf = n.message;
+ size_t bufsize = sizeof(n.message);
+
+ n.time = vl->time;
+
+ const char *service = "Service";
+ if (strcmp(vl->plugin, "uptime") == 0)
+ service = "Host";
+
+ switch (event_type) {
+ case CE_VALUE_NEW:
+ n.severity = NOTIF_OKAY;
+ status = snprintf(buf, bufsize, "%s is running.", service);
+ buf += status;
+ bufsize -= status;
+ break;
+ case CE_VALUE_UPDATE:
+ n.severity = NOTIF_WARNING;
+ status = snprintf(buf, bufsize, "%s just restarted.", service);
+ buf += status;
+ bufsize -= status;
+ break;
+ case CE_VALUE_EXPIRED:
+ n.severity = NOTIF_FAILURE;
+ status = snprintf(buf, bufsize, "%s is unreachable.", service);
+ buf += status;
+ bufsize -= status;
+ break;
+ }
+
+ if (!isnan(old_uptime)) {
+ status = snprintf(buf, bufsize, " Uptime was:");
+ buf += status;
+ bufsize -= status;
+
+ status = format_uptime(old_uptime, buf, bufsize);
+ buf += status;
+ bufsize -= status;
+
+ plugin_notification_meta_add_double(&n, "LastValue", old_uptime);
+ }
+
+ if (!isnan(new_uptime)) {
+ status = snprintf(buf, bufsize, " Uptime now:");
+ buf += status;
+ bufsize -= status;
+
+ status = format_uptime(new_uptime, buf, bufsize);
+ buf += status;
+ bufsize -= status;
+
+ plugin_notification_meta_add_double(&n, "CurrentValue", new_uptime);
+ }
+
+ plugin_dispatch_notification(&n);
+
+ plugin_notification_meta_free(n.meta);
+ return 0;
+}
+
+static int cu_cache_event(cache_event_t *event,
+ __attribute__((unused)) user_data_t *ud) {
+ gauge_t values_history[2];
+
+ /* For CE_VALUE_EXPIRED */
+ int ret;
+ value_t *values;
+ size_t values_num;
+ gauge_t old_uptime = NAN;
+
+ switch (event->type) {
+ case CE_VALUE_NEW:
+ DEBUG("check_uptime: CE_VALUE_NEW, %s", event->value_list_name);
+ if (c_avl_get(types_tree, event->value_list->type, NULL) == 0) {
+ event->ret = 1;
+ assert(event->value_list->values_len > 0);
+ cu_notify(CE_VALUE_NEW, event->value_list, NAN /* old */,
+ event->value_list->values[0].gauge /* new */);
+ }
+ break;
+ case CE_VALUE_UPDATE:
+ DEBUG("check_uptime: CE_VALUE_UPDATE, %s", event->value_list_name);
+ if (uc_get_history_by_name(event->value_list_name, values_history, 2, 1)) {
+ ERROR("check_uptime plugin: Failed to get value history for %s.",
+ event->value_list_name);
+ } else {
+ if (!isnan(values_history[0]) && !isnan(values_history[1]) &&
+ values_history[0] < values_history[1]) {
+ cu_notify(CE_VALUE_UPDATE, event->value_list,
+ values_history[1] /* old */, values_history[0] /* new */);
+ }
+ }
+ break;
+ case CE_VALUE_EXPIRED:
+ DEBUG("check_uptime: CE_VALUE_EXPIRED, %s", event->value_list_name);
+ ret = uc_get_value_by_name(event->value_list_name, &values, &values_num);
+ if (ret == 0) {
+ old_uptime = values[0].gauge;
+ sfree(values);
+ }
+
+ cu_notify(CE_VALUE_EXPIRED, event->value_list, old_uptime, NAN /* new */);
+ break;
+ }
+ return 0;
+}
+
+static int cu_config(oconfig_item_t *ci) {
+ if (types_tree == NULL) {
+ types_tree = c_avl_create((int (*)(const void *, const void *))strcmp);
+ if (types_tree == NULL) {
+ ERROR("check_uptime plugin: c_avl_create failed.");
+ return -1;
+ }
+ }
+
+ for (int i = 0; i < ci->children_num; ++i) {
+ oconfig_item_t *child = ci->children + i;
+ if (strcasecmp("Type", child->key) == 0) {
+ if ((child->values_num != 1) ||
+ (child->values[0].type != OCONFIG_TYPE_STRING)) {
+ WARNING("check_uptime plugin: The `Type' option needs exactly one "
+ "string argument.");
+ return -1;
+ }
+ char *type = child->values[0].value.string;
+
+ if (c_avl_get(types_tree, type, NULL) == 0) {
+ ERROR("check_uptime plugin: Type `%s' already added.", type);
+ return -1;
+ }
+
+ char *type_copy = strdup(type);
+ if (type_copy == NULL) {
+ ERROR("check_uptime plugin: strdup failed.");
+ return -1;
+ }
+
+ int status = c_avl_insert(types_tree, type_copy, NULL);
+ if (status != 0) {
+ ERROR("check_uptime plugin: c_avl_insert failed.");
+ sfree(type_copy);
+ return -1;
+ }
+ } else
+ WARNING("check_uptime plugin: Ignore unknown config option `%s'.",
+ child->key);
+ }
+
+ return 0;
+}
+
+static int cu_init(void) {
+ if (types_tree == NULL) {
+ types_tree = c_avl_create((int (*)(const void *, const void *))strcmp);
+ if (types_tree == NULL) {
+ ERROR("check_uptime plugin: c_avl_create failed.");
+ return -1;
+ }
+ /* Default configuration */
+ char *type = strdup("uptime");
+ if (type == NULL) {
+ ERROR("check_uptime plugin: strdup failed.");
+ return -1;
+ }
+ int status = c_avl_insert(types_tree, type, NULL);
+ if (status != 0) {
+ ERROR("check_uptime plugin: c_avl_insert failed.");
+ sfree(type);
+ return -1;
+ }
+ }
+
+ int ret = 0;
+ char *type;
+ void *val;
+ c_avl_iterator_t *iter = c_avl_get_iterator(types_tree);
+ while (c_avl_iterator_next(iter, (void *)&type, (void *)&val) == 0) {
+ data_set_t const *ds = plugin_get_ds(type);
+ if (ds == NULL) {
+ ERROR("check_uptime plugin: Failed to look up type \"%s\".", type);
+ ret = -1;
+ continue;
+ }
+ if (ds->ds_num != 1) {
+ ERROR("check_uptime plugin: The type \"%s\" has %" PRIsz " data sources. "
+ "Only types with a single GAUGE data source are supported.",
+ ds->type, ds->ds_num);
+ ret = -1;
+ continue;
+ }
+ if (ds->ds[0].type != DS_TYPE_GAUGE) {
+ ERROR("check_uptime plugin: The type \"%s\" has wrong data source type. "
+ "Only types with a single GAUGE data source are supported.",
+ ds->type);
+ ret = -1;
+ continue;
+ }
+ }
+ c_avl_iterator_destroy(iter);
+
+ if (ret == 0)
+ plugin_register_cache_event("check_uptime", cu_cache_event, NULL);
+
+ return ret;
+}
+
+void module_register(void) {
+ plugin_register_complex_config("check_uptime", cu_config);
+ plugin_register_init("check_uptime", cu_init);
+}