Changeset 85345 in vbox for trunk/src/VBox/Runtime/common/string

Timestamp:

Jul 15, 2020 8:03:03 AM (5 years ago)

Author:

vboxsync

Message:

IPRT: Follow-up fixes for r139114 (RTStrSplit + testcases).

File:

• trunk/src/VBox/Runtime/common/string/RTStrSplit.cpp (modified) (3 diffs)

trunk/src/VBox/Runtime/common/string/RTStrSplit.cpp

@@ -46 +46 @@
 
     /* Determine the number of paths in buffer first. */
-    size_t       cch    = cbStrings - 1;
-    char const  *pszTmp = pcszStrings;
+    size_t      cch     = cbStrings - 1;
+    char const *pcszTmp = pcszStrings;
+    const char *pcszEnd = RTStrEnd(pcszTmp, RTSTR_MAX);
+    char const *pcszNext;
     const size_t cchSep = strlen(pcszSeparator);
+          size_t cchNext;
     while (cch > 0)
     {
-        char const *pszNext = RTStrStr(pszTmp, pcszSeparator);
-        if (!pszNext)
+        pcszNext = RTStrStr(pcszTmp, pcszSeparator);
+        if (!pcszNext)
             break;
-        const size_t cchNext = pszNext - pszTmp;
+        cchNext = pcszNext - pcszTmp;
         if (cchNext + cchSep > cch)
             break;
-        pszTmp += cchNext + cchSep;
-        cch    -= cchNext + cchSep;
+        pcszNext += cchSep;
+        pcszTmp  += cchNext + cchSep;
+        cch      -= cchNext + cchSep;
         if (cchNext)
             ++cStrings;
     }
 
+    if (pcszTmp != pcszEnd) /* Do we need to take a trailing string without separator into account? */
+        cStrings++;
+
     if (!cStrings)
     {
-        *pcStrings = 0;
+        *ppapszStrings = NULL;
+        *pcStrings     = 0;
         return VINF_SUCCESS;
     }
@@ -76 +84 @@
 
     cch    = cbStrings - 1;
-    pszTmp = pcszStrings;
+    pcszTmp = pcszStrings;
 
-    for (size_t i = 0; i < 3;)
+    for (size_t i = 0; i < cStrings;)
     {
-        char const *pszNext = RTStrStr(pszTmp, pcszSeparator);
-        if (!pszNext)
-            break;
-        const size_t cchNext = pszNext - pszTmp;
-        if (cchNext + cchSep > cch)
-            break;
+        pcszNext = RTStrStr(pcszTmp, pcszSeparator);
+        if (!pcszNext)
+            pcszNext = pcszEnd;
+        cchNext = pcszNext - pcszTmp;
         if (cchNext)
         {
-            papszStrings[i] = RTStrDupN(pszTmp, cchNext);
+            papszStrings[i] = RTStrDupN(pcszTmp, cchNext);
             if (!papszStrings[i])
             {
@@ -96 +102 @@
             i++;
         }
-        pszTmp += cchNext + cchSep;
-        cch    -= cchNext + cchSep;
+        pcszTmp += cchNext + cchSep;
+        cch     -= cchNext + cchSep;
     }