Code

Moved core/error to utils/error.
[sysdb.git] / src / backend / mk-livestatus.c
index a5126fb037e8c3e9a70412b989692b3bdb1e6c3e..bf45da6fdd35df6d3170ed4a4ebafbc1b513b49a 100644 (file)
@@ -28,7 +28,7 @@
 #include "sysdb.h"
 #include "core/plugin.h"
 #include "core/store.h"
-#include "utils/string.h"
+#include "utils/error.h"
 #include "utils/unixsock.h"
 
 #include "liboconfig/utils.h"
@@ -52,10 +52,8 @@ sdb_livestatus_get_host(sdb_unixsock_client_t __attribute__((unused)) *client,
                size_t n, sdb_data_t *data,
                sdb_object_t __attribute__((unused)) *user_data)
 {
-       char *hostname = NULL;
-       sdb_time_t timestamp = 0;
-
-       sdb_host_t host = SDB_HOST_INIT;
+       const char *hostname;
+       sdb_time_t timestamp;
 
        int status;
 
@@ -63,27 +61,22 @@ sdb_livestatus_get_host(sdb_unixsock_client_t __attribute__((unused)) *client,
        assert((data[0].type == SDB_TYPE_STRING)
                        && (data[1].type == SDB_TYPE_DATETIME));
 
-       hostname  = strdup(data[0].data.string);
+       hostname  = data[0].data.string;
        timestamp = data[1].data.datetime;
 
-       host.host_name = hostname;
-       host.host_last_update = timestamp;
-
-       status = sdb_store_host(&host);
+       status = sdb_store_host(hostname, timestamp);
 
        if (status < 0) {
-               fprintf(stderr, "MK Livestatus backend: Failed to store/update "
-                               "host '%s'.\n", hostname);
-               free(hostname);
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to "
+                               "store/update host '%s'.", hostname);
                return -1;
        }
        else if (status > 0) /* value too old */
                return 0;
 
-       fprintf(stderr, "MK Livestatus backend: Added/updated host '%s' "
-                       "(last update timestamp = %"PRIscTIME").\n",
+       sdb_log(SDB_LOG_DEBUG, "MK Livestatus backend: Added/updated "
+                       "host '%s' (last update timestamp = %"PRIscTIME").",
                        hostname, timestamp);
-       free(hostname);
        return 0;
 } /* sdb_livestatus_get_host */
 
@@ -92,12 +85,10 @@ sdb_livestatus_get_svc(sdb_unixsock_client_t __attribute__((unused)) *client,
                size_t n, sdb_data_t *data,
                sdb_object_t __attribute__((unused)) *user_data)
 {
-       char *hostname = NULL;
-       char *svcname = NULL;
+       const char *hostname = NULL;
+       const char *svcname = NULL;
        sdb_time_t timestamp = 0;
 
-       sdb_service_t svc = SDB_SVC_INIT;
-
        int status;
 
        assert(n == 3);
@@ -105,31 +96,23 @@ sdb_livestatus_get_svc(sdb_unixsock_client_t __attribute__((unused)) *client,
                        && (data[1].type == SDB_TYPE_STRING)
                        && (data[2].type == SDB_TYPE_DATETIME));
 
-       hostname  = strdup(data[0].data.string);
-       svcname   = strdup(data[1].data.string);
+       hostname  = data[0].data.string;
+       svcname   = data[1].data.string;
        timestamp = data[2].data.datetime;
 
-       svc.hostname = hostname;
-       svc.svc_name = svcname;
-       svc.svc_last_update = timestamp;
-
-       status = sdb_store_service(&svc);
+       status = sdb_store_service(hostname, svcname, timestamp);
 
        if (status < 0) {
-               fprintf(stderr, "MK Livestatus backend: Failed to store/update "
-                               "service '%s / %s'.\n", hostname, svcname);
-               free(hostname);
-               free(svcname);
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to "
+                               "store/update service '%s / %s'.", hostname, svcname);
                return -1;
        }
        else if (status > 0) /* value too old */
                return 0;
 
-       fprintf(stderr, "MK Livestatus backend: Added/updated service '%s / %s' "
-                       "(last update timestamp = %"PRIscTIME").\n",
+       sdb_log(SDB_LOG_DEBUG, "MK Livestatus backend: Added/updated "
+                       "service '%s / %s' (last update timestamp = %"PRIscTIME").",
                        hostname, svcname, timestamp);
-       free(hostname);
-       free(svcname);
        return 0;
 } /* sdb_livestatus_get_svc */
 
@@ -147,14 +130,14 @@ sdb_livestatus_init(sdb_object_t *user_data)
 
        client = SDB_OBJ_WRAPPER(user_data)->data;
        if (sdb_unixsock_client_connect(client)) {
-               fprintf(stderr, "MK Livestatus backend: "
-                               "Failed to connect to livestatus @ %s.\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: "
+                               "Failed to connect to livestatus @ %s.",
                                sdb_unixsock_client_path(client));
                return -1;
        }
 
-       fprintf(stderr, "MK Livestatus backend: Successfully "
-                       "connected to livestatus @ %s.\n",
+       sdb_log(SDB_LOG_INFO, "MK Livestatus backend: Successfully "
+                       "connected to livestatus @ %s.",
                        sdb_unixsock_client_path(client));
        return 0;
 } /* sdb_livestatus_init */
@@ -174,8 +157,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
        status = sdb_unixsock_client_send(client, "GET hosts\r\n"
                        "Columns: name last_check");
        if (status <= 0) {
-               fprintf(stderr, "MK Livestatus backend: Failed to send "
-                               "'GET hosts' command to livestatus @ %s.\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to send "
+                               "'GET hosts' command to livestatus @ %s.",
                                sdb_unixsock_client_path(client));
                return -1;
        }
@@ -185,8 +168,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
        if (sdb_unixsock_client_process_lines(client, sdb_livestatus_get_host,
                                /* user data */ NULL, /* -> EOF */ -1, /* delim */ ";",
                                /* column count */ 2, SDB_TYPE_STRING, SDB_TYPE_DATETIME)) {
-               fprintf(stderr, "MK Livestatus backend: Failed to read response "
-                               "from livestatus @ %s while reading hosts.\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to read "
+                               "response from livestatus @ %s while reading hosts.",
                                sdb_unixsock_client_path(client));
                return -1;
        }
@@ -194,8 +177,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
        if ((! sdb_unixsock_client_eof(client))
                        || sdb_unixsock_client_error(client)) {
                char errbuf[1024];
-               fprintf(stderr, "MK Livestatus backend: Failed to read host "
-                               "from livestatus @ %s: %s\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to read "
+                               "host from livestatus @ %s: %s",
                                sdb_unixsock_client_path(client),
                                sdb_strerror(errno, errbuf, sizeof(errbuf)));
                return -1;
@@ -204,8 +187,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
        status = sdb_unixsock_client_send(client, "GET services\r\n"
                        "Columns: host_name description last_check");
        if (status <= 0) {
-               fprintf(stderr, "MK Livestatus backend: Failed to send "
-                               "'GET services' command to livestatus @ %s.\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to send "
+                               "'GET services' command to livestatus @ %s.",
                                sdb_unixsock_client_path(client));
                return -1;
        }
@@ -216,8 +199,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
                                /* user data */ NULL, /* -> EOF */ -1, /* delim */ ";",
                                /* column count */ 3, SDB_TYPE_STRING, SDB_TYPE_STRING,
                                SDB_TYPE_DATETIME)) {
-               fprintf(stderr, "MK Livestatus backend: Failed to read response "
-                               "from livestatus @ %s while reading services.\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to read "
+                               "response from livestatus @ %s while reading services.",
                                sdb_unixsock_client_path(client));
                return -1;
        }
@@ -225,8 +208,8 @@ sdb_livestatus_collect(sdb_object_t *user_data)
        if ((! sdb_unixsock_client_eof(client))
                        || sdb_unixsock_client_error(client)) {
                char errbuf[1024];
-               fprintf(stderr, "MK Livestatus backend: Failed to read services "
-                               "from livestatus @ %s: %s\n",
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to read "
+                               "services from livestatus @ %s: %s",
                                sdb_unixsock_client_path(client),
                                sdb_strerror(errno, errbuf, sizeof(errbuf)));
                return -1;
@@ -248,8 +231,8 @@ sdb_livestatus_config_instance(oconfig_item_t *ci)
        int i;
 
        if (oconfig_get_string(ci, &name)) {
-               fprintf(stderr, "MK Livestatus backend: Instance requires a single "
-                               "string argument\n\tUsage: <Instance NAME>\n");
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Instance requires "
+                               "a single string argument\n\tUsage: <Instance NAME>");
                return -1;
        }
 
@@ -259,34 +242,35 @@ sdb_livestatus_config_instance(oconfig_item_t *ci)
                if (! strcasecmp(child->key, "Socket"))
                        oconfig_get_string(child, &socket_path);
                else
-                       fprintf(stderr, "MK Livestatus backend: Ignoring unknown config "
-                                       "option '%s' inside <Instance %s>.\n",
+                       sdb_log(SDB_LOG_WARNING, "MK Livestatus backend: Ignoring "
+                                       "unknown config option '%s' inside <Instance %s>.",
                                        child->key, name);
        }
 
        if (! socket_path) {
-               fprintf(stderr, "MK Livestatus backend: Instance '%s' missing "
-                               "the 'Socket' option.\n", name);
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Instance '%s' "
+                               "missing the 'Socket' option.", name);
                return -1;
        }
 
-       snprintf(cb_name, sizeof(cb_name), "mk-livestatus-%s", name);
+       snprintf(cb_name, sizeof(cb_name), "mk-livestatus::%s", name);
        cb_name[sizeof(cb_name) - 1] = '\0';
 
        client = sdb_unixsock_client_create(socket_path);
        if (! client) {
                char errbuf[1024];
-               fprintf(stderr, "MK Livestatus backend: Failed to create unixsock "
-                               "client: %s\n", sdb_strerror(errno, errbuf, sizeof(errbuf)));
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to create "
+                               "unixsock client: %s",
+                               sdb_strerror(errno, errbuf, sizeof(errbuf)));
                return -1;
        }
 
-       user_data = sdb_object_create_wrapper(client,
+       user_data = sdb_object_create_wrapper("unixsock-client", client,
                        (void (*)(void *))sdb_unixsock_client_destroy);
        if (! user_data) {
                sdb_unixsock_client_destroy(client);
-               fprintf(stderr, "MK Livestatus backend: Failed to "
-                               "allocate sdb_object_t\n");
+               sdb_log(SDB_LOG_ERR, "MK Livestatus backend: Failed to "
+                               "allocate sdb_object_t");
                return -1;
        }
 
@@ -310,8 +294,8 @@ sdb_livestatus_config(oconfig_item_t *ci)
                if (! strcasecmp(child->key, "Instance"))
                        sdb_livestatus_config_instance(child);
                else
-                       fprintf(stderr, "MK Livestatus backend: Ignoring unknown config "
-                                       "option '%s'.\n", child->key);
+                       sdb_log(SDB_LOG_WARNING, "MK Livestatus backend: Ignoring "
+                                       "unknown config option '%s'.", child->key);
        }
        return 0;
 } /* sdb_livestatus_config */