Add support for parsing NaN, INF, and -INF

2 files changed, 24 insertions, 4 deletions

diff --git a/src/string.c b/src/string.c
index 635588ae..488d0ae3 100644
--- a/src/string.c
+++ b/src/string.c
@@ -114,6 +114,19 @@ serd_strtod(const char* str, size_t* end)
 {
 	double result = 0.0;
 
+#define SET_END(index) if (end) { *end = index; }
+
+	if (!strcmp(str, "NaN")) {
+		SET_END(3);
+		return NAN;
+	} else if (!strcmp(str, "-INF")) {
+		SET_END(4);
+		return -INFINITY;
+	} else if (!strcmp(str, "INF")) {
+		SET_END(3);
+		return INFINITY;
+	}
+
 	// Point s at the first non-whitespace character
 	const char* s = str;
 	while (is_space(*s)) { ++s; }
@@ -146,10 +159,7 @@ serd_strtod(const char* str, size_t* end)
 		result *= pow(10, expt * expt_sign);
 	}
 
-	if (end) {
-		*end = s - str;
-	}
-
+	SET_END(s - str);
 	return result * sign;
 }
 
diff --git a/tests/serd_test.c b/tests/serd_test.c
index 85c9db27..5ab22846 100644
--- a/tests/serd_test.c
+++ b/tests/serd_test.c
@@ -179,6 +179,16 @@ test_string_to_double(void)
 		test_strtod(dbl, 1 / (double)MAX);
 	}
 
+	size_t end = 0;
+	assert(isnan(serd_strtod("NaN", &end)));
+	assert(end == 3);
+
+	assert(serd_strtod("INF", &end) == INFINITY);
+	assert(end == 3);
+
+	assert(serd_strtod("-INF", &end) == -INFINITY);
+	assert(end == 4);
+
 	const double expt_test_nums[] = {
 		2.0E18, -5e19, +8e20, 2e+24, -5e-5, 8e0, 9e-0, 2e+0
 	};