From dd7a268854d23425d2cea6b3e896e21f2fbc9aba Mon Sep 17 00:00:00 2001
From: Sebastian Harl <sh@tokkee.org>
Date: Thu, 13 Dec 2012 13:57:38 +0100
Subject: [PATCH] MK Livestatus backend: Converted to use
 sc_unixsock_client_process_lines().

This removes the need to do the parsing of returned lines in the backend code.
---
 src/backend/mk-livestatus.c | 98 ++++++++++++-------------------------
 1 file changed, 32 insertions(+), 66 deletions(-)

diff --git a/src/backend/mk-livestatus.c b/src/backend/mk-livestatus.c
index f6b9b5f..9f99c46 100644
--- a/src/backend/mk-livestatus.c
+++ b/src/backend/mk-livestatus.c
@@ -48,46 +48,43 @@ SC_PLUGIN_MAGIC;
  */
 
 static int
-sc_livestatus_parse_line(char *line,
-		char **host, sc_time_t *timestamp)
+sc_livestatus_get_data(sc_unixsock_client_t __attribute__((unused)) *client,
+		size_t n, sc_data_t *data, sc_object_t __attribute__((unused)) *user_data)
 {
-	char *timestamp_str;
-	double timestamp_dbl;
+	char *hostname = NULL;
+	sc_time_t timestamp = 0;
 
-	char *hostname;
+	sc_host_t host = SC_HOST_INIT;
 
-	char *endptr = NULL;
+	int status;
 
-	hostname = line;
-	timestamp_str = strchr(hostname, ';');
+	assert(n == 2);
+	assert((data[0].type == SC_TYPE_STRING)
+			&& (data[1].type == SC_TYPE_DATETIME));
 
-	if (! timestamp_str) {
-		fprintf(stderr, "MK Livestatus backend: Failed to find timestamp "
-				"in 'GET hosts' output, line: %s\n", line);
-		return -1;
-	}
+	hostname  = strdup(data[0].data.string);
+	timestamp = data[1].data.datetime;
 
-	*timestamp_str = '\0';
-	++timestamp_str;
+	host.host_name = hostname;
+	host.host_last_update = timestamp;
 
-	errno = 0;
-	timestamp_dbl = strtod(timestamp_str, &endptr);
-	if (errno || (timestamp_str == endptr)) {
-		char errbuf[1024];
-		fprintf(stderr, "MK Livestatus backend: Failed to "
-				"parse timestamp (%s): %s\n", timestamp_str,
-				sc_strerror(errno, errbuf, sizeof(errbuf)));
+	status = sc_store_host(&host);
+
+	if (status < 0) {
+		fprintf(stderr, "MK Livestatus backend: Failed to store/update "
+				"host '%s'.\n", hostname);
+		free(hostname);
 		return -1;
 	}
-	if (endptr && (*endptr != '\0'))
-		fprintf(stderr, "MK Livestatus backend: Ignoring garbage "
-				"after number when parsing timestamp: %s.\n",
-				endptr);
+	else if (status > 0) /* value too old */
+		return 0;
 
-	*timestamp = DOUBLE_TO_SC_TIME(timestamp_dbl);
-	*host = hostname;
+	fprintf(stderr, "MK Livestatus backend: Added/updated host '%s' "
+			"(last update timestamp = %"PRIscTIME").\n",
+			hostname, timestamp);
+	free(hostname);
 	return 0;
-} /* sc_livestatus_parse_line */
+} /* sc_livestatus_get_data */
 
 /*
  * plugin API
@@ -138,43 +135,12 @@ sc_livestatus_collect(sc_object_t *user_data)
 
 	sc_unixsock_client_shutdown(client, SHUT_WR);
 
-	while (42) {
-		char  buffer[1024];
-		char *line;
-
-		char *hostname = NULL;
-		sc_time_t timestamp = 0;
-
-		sc_host_t host = SC_HOST_INIT;
-
-		sc_unixsock_client_clearerr(client);
-		line = sc_unixsock_client_recv(client, buffer, sizeof(buffer));
-
-		if (! line)
-			break;
-
-		if (sc_livestatus_parse_line(line, &hostname, &timestamp)) {
-			fprintf(stderr, "MK Livestatus backend: Failed to parse "
-					"hosts line: %s\n", line);
-			continue;
-		}
-
-		host.host_name = hostname;
-		host.host_last_update = timestamp;
-
-		status = sc_store_host(&host);
-
-		if (status < 0) {
-			fprintf(stderr, "MK Livestatus backend: Failed to store/update "
-					"host '%s'.\n", hostname);
-			continue;
-		}
-		else if (status > 0) /* value too old */
-			continue;
-
-		fprintf(stderr, "MK Livestatus backend: Added/updated host '%s' "
-				"(last update timestamp = %"PRIscTIME").\n",
-				hostname, timestamp);
+	if (sc_unixsock_client_process_lines(client, sc_livestatus_get_data,
+				/* user data */ NULL, /* -> EOF */ -1, /* delim */ ";",
+				/* column count */ 2, SC_TYPE_STRING, SC_TYPE_DATETIME)) {
+		fprintf(stderr, "MK Livestatus backend: Failed to read response "
+				"from livestatus @ %s.\n", sc_unixsock_client_path(client));
+		return -1;
 	}
 
 	if ((! sc_unixsock_client_eof(client))
-- 
2.30.2