Move check_ntp's extract_value to utils_base.c.

This function can also be used to parse performance data strings which
could be useful in the future.

3 files changed, 145 insertions, 1 deletions

diff --git a/lib/tests/test_utils.c b/lib/tests/test_utils.c
index 86a17dc3..97343afa 100644
--- a/lib/tests/test_utils.c
+++ b/lib/tests/test_utils.c
@@ -29,7 +29,7 @@ main (int argc, char **argv)
 	thresholds *thresholds = NULL;
 	int	rc;
 
-	plan_tests(81);
+	plan_tests(81+23);
 
 	range = parse_range_string("6");
 	ok( range != NULL, "'6' is valid range");
@@ -172,5 +172,84 @@ main (int argc, char **argv)
 	test = np_escaped_string("everything");
 	ok( strcmp(test, "everything") == 0, "everything okay");
 
+	/* np_extract_value tests (23) */
+	test=np_extract_value("foo=bar, bar=foo, foobar=barfoo\n", "foo");
+	ok(test && !strcmp(test, "bar"), "1st test as expected");
+	free(test);
+
+	test=np_extract_value("foo=bar,bar=foo,foobar=barfoo\n", "bar");
+	ok(test && !strcmp(test, "foo"), "2nd test as expected");
+	free(test);
+
+	test=np_extract_value("foo=bar, bar=foo, foobar=barfoo\n", "foobar");
+	ok(test && !strcmp(test, "barfoo"), "3rd test as expected");
+	free(test);
+
+	test=np_extract_value("foo=bar\n", "foo");
+	ok(test && !strcmp(test, "bar"), "Single test as expected");
+	free(test);
+
+	test=np_extract_value("foo=bar, bar=foo, foobar=barfooi\n", "abcd");
+	ok(!test, "Key not found 1");
+
+	test=np_extract_value("foo=bar\n", "abcd");
+	ok(!test, "Key not found 2");
+
+	test=np_extract_value("foo=bar=foobar", "foo");
+	ok(test && !strcmp(test, "bar=foobar"), "Strange string 1");
+	free(test);
+
+	test=np_extract_value("foo", "foo");
+	ok(!test, "Malformed string 1");
+
+	test=np_extract_value("foo,", "foo");
+	ok(!test, "Malformed string 2");
+
+	test=np_extract_value("foo=", "foo");
+	ok(!test, "Malformed string 3");
+
+	test=np_extract_value("foo=,bar=foo", "foo");
+	ok(!test, "Malformed string 4");
+
+	test=np_extract_value(",foo", "foo");
+	ok(!test, "Malformed string 5");
+
+	test=np_extract_value("=foo", "foo");
+	ok(!test, "Malformed string 6");
+
+	test=np_extract_value("=foo,", "foo");
+	ok(!test, "Malformed string 7");
+
+	test=np_extract_value(",,,", "foo");
+	ok(!test, "Malformed string 8");
+
+	test=np_extract_value("===", "foo");
+	ok(!test, "Malformed string 9");
+
+	test=np_extract_value(",=,=,", "foo");
+	ok(!test, "Malformed string 10");
+
+	test=np_extract_value("=,=,=", "foo");
+	ok(!test, "Malformed string 11");
+
+	test=np_extract_value("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "foo");
+	ok(test && !strcmp(test, "bar"), "Random spaces and newlines 1");
+	free(test);
+
+	test=np_extract_value("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "bar");
+	ok(test && !strcmp(test, "foo"), "Random spaces and newlines 2");
+	free(test);
+
+	test=np_extract_value("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "foobar");
+	ok(test && !strcmp(test, "barfoo"), "Random spaces and newlines 3");
+	free(test);
+
+	test=np_extract_value("  foo=bar  ,\n bar\n \n= \n foo\n , foobar=barfoo  \n  ", "bar");
+	ok(test && !strcmp(test, "foo"), "Random spaces and newlines 4");
+	free(test);
+
+	test=np_extract_value("", "foo");
+	ok(!test, "Empty string return NULL");
+
 	return exit_status();
 }
diff --git a/lib/utils_base.c b/lib/utils_base.c
index d6437fc8..a34cc5cc 100644
--- a/lib/utils_base.c
+++ b/lib/utils_base.c
@@ -251,3 +251,60 @@ int np_warn_if_not_root(void) {
 	}
 	return status;
 }
+
+/*
+ * Extract the value from key/value pairs, or return NULL. The value returned
+ * can be free()ed.
+ * This function can be used to parse NTP control packet data and performance
+ * data strings.
+ */
+char *np_extract_value(const char *varlist, const char *name) {
+	char *tmp=NULL, *value=NULL;
+	int i;
+
+	while (1) {
+		/* Strip any leading space */
+		for (varlist; isspace(varlist[0]); varlist++);
+
+		if (strncmp(name, varlist, strlen(name)) == 0) {
+			varlist += strlen(name);
+			/* strip trailing spaces */
+			for (varlist; isspace(varlist[0]); varlist++);
+
+			if (varlist[0] == '=') {
+				/* We matched the key, go past the = sign */
+				varlist++;
+				/* strip leading spaces */
+				for (varlist; isspace(varlist[0]); varlist++);
+
+				if (tmp = index(varlist, ',')) {
+					/* Value is delimited by a comma */
+					if (tmp-varlist == 0) continue;
+					value = (char *)malloc(tmp-varlist+1);
+					strncpy(value, varlist, tmp-varlist);
+					value[tmp-varlist] = '\0';
+				} else {
+					/* Value is delimited by a \0 */
+					if (strlen(varlist) == 0) continue;
+					value = (char *)malloc(strlen(varlist) + 1);
+					strncpy(value, varlist, strlen(varlist));
+					value[strlen(varlist)] = '\0';
+				}
+				break;
+			}
+		}
+		if (tmp = index(varlist, ',')) {
+			/* More keys, keep going... */
+			varlist = tmp + 1;
+		} else {
+			/* We're done */
+			break;
+		}
+	}
+
+	/* Clean-up trailing spaces/newlines */
+	if (value) for (i=strlen(value)-1; isspace(value[i]); i--) value[i] = '\0';
+
+	return value;
+}
+
diff --git a/lib/utils_base.h b/lib/utils_base.h
index bda76595..c34f0445 100644
--- a/lib/utils_base.h
+++ b/lib/utils_base.h
@@ -50,4 +50,12 @@ int np_check_if_root(void);
  * code from the above function, in case it's helpful for testing */
 int np_warn_if_not_root(void);
 
+/*
+ * Extract the value from key/value pairs, or return NULL. The value returned
+ * can be free()ed.
+ * This function can be used to parse NTP control packet data and performance
+ * data strings.
+ */
+char *np_extract_value(const char*, const char*);
+
 #endif /* _UTILS_BASE_ */