[RFC PATCH v2 02/48] lib: Handle a special case with str_to_list()

[Sughosh Ganu]

From: Simon Glass <s...@chromium.org>

The current implementation can return an extra result at the end when
the string ends with a space. Fix this by adding a special case.

Signed-off-by: Simon Glass <s...@chromium.org>
Signed-off-by: Sughosh Ganu <sughosh.g...@linaro.org>
---
 lib/strto.c   | 4 +++-
 test/str_ut.c | 4 +---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/strto.c b/lib/strto.c
index 5157332d6c..f83ac67c66 100644
--- a/lib/strto.c
+++ b/lib/strto.c
@@ -236,12 +236,14 @@ const char **str_to_list(const char *instr)
                return NULL;
 
        /* count the number of space-separated strings */
-       for (count = *str != '\0', p = str; *p; p++) {
+       for (count = 0, p = str; *p; p++) {
                if (*p == ' ') {
                        count++;
                        *p = '\0';
                }
        }
+       if (p != str && p[-1])
+               count++;
 
        /* allocate the pointer array, allowing for a NULL terminator */
        ptr = calloc(count + 1, sizeof(char *));
diff --git a/test/str_ut.c b/test/str_ut.c
index 389779859a..96e048975d 100644
--- a/test/str_ut.c
+++ b/test/str_ut.c
@@ -342,9 +342,7 @@ static int test_str_to_list(struct unit_test_state *uts)
        ut_asserteq_str("space", ptr[3]);
        ut_assertnonnull(ptr[4]);
        ut_asserteq_str("", ptr[4]);
-       ut_assertnonnull(ptr[5]);
-       ut_asserteq_str("", ptr[5]);
-       ut_assertnull(ptr[6]);
+       ut_assertnull(ptr[5]);
        str_free_list(ptr);
        ut_assertok(ut_check_delta(start));
 
-- 
2.34.1